ColossalAI/tests/test_auto_parallel
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
..
test_tensor_shard [autoparallel] shard param and buffer as expected (#1753) 2022-10-21 15:45:13 +08:00
__init__.py [autoparallel] handled illegal sharding strategy (#1728) 2022-10-19 12:53:06 +08:00