ColossalAI/tests/test_auto_parallel
YuliangLiu0306 536560ccc0
[autoparallel] implement softmax handler (#2132)
2022-12-14 16:09:53 +08:00
..
test_tensor_shard [autoparallel] implement softmax handler (#2132) 2022-12-14 16:09:53 +08:00
__init__.py [autoparallel] handled illegal sharding strategy (#1728) 2022-10-19 12:53:06 +08:00