ColossalAI/tests/test_auto_parallel
Frank Lee f3f19a5c47
[autoparallel] added matmul handler (#1763)
* [autoparallel] added matmul handler

* polish code
2022-11-01 15:14:53 +08:00
..
test_tensor_shard [autoparallel] added matmul handler (#1763) 2022-11-01 15:14:53 +08:00
__init__.py [autoparallel] handled illegal sharding strategy (#1728) 2022-10-19 12:53:06 +08:00