ColossalAI/colossalai/auto_parallel
Boyuan Yao c8c79102f0
[autoparallel] patch torch.flatten metainfo for autoparallel (#2247)
* [autoparallel] patch torch.flatten
2023-01-02 15:51:03 +08:00
..
checkpoint [autoparallel] user-friendly API for CheckpointSolver. (#1879) 2022-11-10 20:59:28 +08:00
meta_profiler [autoparallel] patch torch.flatten metainfo for autoparallel (#2247) 2023-01-02 15:51:03 +08:00
passes [autoparallel] fix construct meta info. (#2245) 2022-12-30 19:56:44 +08:00
pipeline_shard [autoparallel] init new folder structure (#1696) 2022-10-13 14:18:55 +08:00
tensor_shard [autoparallel] autoparallel initialize (#2238) 2022-12-31 01:02:14 +08:00
__init__.py [autoparallel] standardize the code structure (#1469) 2022-08-19 15:51:54 +08:00