ColossalAI/tests/test_auto_parallel/test_tensor_shard
YuliangLiu0306 d2fc067231
[autoparallel] fix param hook issue in transform pass (#1755)
2022-10-24 13:13:38 +08:00
..
test_deprecated [hotfix] autoparallel unit test (#1752) 2022-10-20 19:51:38 +08:00
test_node_handler [autoparallel] added addbmm handler (#1751) 2022-10-21 18:55:48 +08:00
__init__.py [autoparallel] handled illegal sharding strategy (#1728) 2022-10-19 12:53:06 +08:00
test_broadcast.py [autoparallel] refactored the autoparallel module for organization (#1706) 2022-10-14 13:27:00 +08:00
test_liveness_analysis.py [autoparallel] refactored the autoparallel module for organization (#1706) 2022-10-14 13:27:00 +08:00
test_resnet_block_runtime.py [autoparallel] fix param hook issue in transform pass (#1755) 2022-10-24 13:13:38 +08:00
test_shape_consistency_pass.py [autoparallel] add sequential order to communication actions (#1735) 2022-10-20 18:48:18 +08:00
test_solver_with_resnet_v2.py [autoparallel] refactored the autoparallel module for organization (#1706) 2022-10-14 13:27:00 +08:00