ColossalAI/tests/test_tensor
YuliangLiu0306 980ed21723
[autoparallel] shard param and buffer as expected (#1753)
* [autoparallel] shard param and buffer as expected

* fix unit test issue
2022-10-21 15:45:13 +08:00
..
common_utils
core
model
test_colo_checkpoint_tools.py
test_comm_spec_apply.py [autoparallel] shard param and buffer as expected (#1753) 2022-10-21 15:45:13 +08:00
test_context.py
test_parameter.py
test_shape_consistency.py [autoparallel] update CommSpec (#1667) 2022-09-29 11:20:59 +08:00
test_shape_consistency_apply.py [autoparallel] shard param and buffer as expected (#1753) 2022-10-21 15:45:13 +08:00
test_sharded_linear.py [autoparallel] handled illegal sharding strategy (#1728) 2022-10-19 12:53:06 +08:00
test_sharding_spec.py [autoparallel] handled illegal sharding strategy (#1728) 2022-10-19 12:53:06 +08:00
test_tp_with_zero.py [zero] add chunk init function for users (#1729) 2022-10-18 16:31:22 +08:00