ColossalAI/colossalai/auto_parallel
YuliangLiu0306 550f8f8905
[autoparallel] integrate_gpt_related_tests (#2134)
* [autoparallel] integrate_gpt_related_tests

* polish code

* polish code

* add GPT2Model into runtime test
2022-12-23 12:36:59 +08:00
..
checkpoint [autoparallel] user-friendly API for CheckpointSolver. (#1879) 2022-11-10 20:59:28 +08:00
meta_profiler [autoparallel] use metainfo in handler (#2149) 2022-12-20 10:31:22 +08:00
passes [autoparallel] integrate_gpt_related_tests (#2134) 2022-12-23 12:36:59 +08:00
pipeline_shard [autoparallel] init new folder structure (#1696) 2022-10-13 14:18:55 +08:00
tensor_shard [autoparallel] memory estimation for shape consistency (#2144) 2022-12-21 10:39:37 +08:00
__init__.py [autoparallel] standardize the code structure (#1469) 2022-08-19 15:51:54 +08:00