ColossalAI/tests/test_auto_parallel
YuliangLiu0306 35e6b9ec82
[autoparallel] adapt handlers with attention block (#1990)
* [autoparallel] adapt handlers with attention block

* polish
2022-11-21 10:44:11 +08:00
..
test_tensor_shard [autoparallel] adapt handlers with attention block (#1990) 2022-11-21 10:44:11 +08:00
__init__.py [autoparallel] handled illegal sharding strategy (#1728) 2022-10-19 12:53:06 +08:00