ColossalAI/tests/test_auto_parallel
Frank Lee 638a07a7f9
[test] fixed gemini plugin test (#3411)
* [test] fixed gemini plugin test

* polish code

* polish code
2023-04-03 17:12:22 +08:00
..
test_ckpt_solvers [hotfix] skip auto checkpointing tests (#3029) 2023-03-07 15:50:00 +08:00
test_offload [test] fixed gemini plugin test (#3411) 2023-04-03 17:12:22 +08:00
test_pass [autoparallel] find repeat blocks (#2854) 2023-02-23 17:28:19 +08:00
test_tensor_shard [autoparallel] adapt autoparallel with new analyzer (#3261) 2023-03-30 17:47:24 +08:00
__init__.py [autoparallel] handled illegal sharding strategy (#1728) 2022-10-19 12:53:06 +08:00