ColossalAI/tests/test_auto_parallel
YuliangLiu0306 7f72eb0510
[autoparallel]add embedding handler (#2089)
* [autoparallel] add embedding handler

* fix bugs
2022-12-07 09:41:46 +08:00
..
test_tensor_shard [autoparallel]add embedding handler (#2089) 2022-12-07 09:41:46 +08:00
__init__.py [autoparallel] handled illegal sharding strategy (#1728) 2022-10-19 12:53:06 +08:00