You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
ColossalAI/colossalai/auto_parallel
YuliangLiu0306 3b1b91eaf4
[autoparallel] record parameter attribute in colotracer (#2217)
2 years ago
..
checkpoint [autoparallel] user-friendly API for CheckpointSolver. (#1879) 2 years ago
meta_profiler [autoparallel] Attach input, buffer and output tensor to MetaInfo class (#2162) 2 years ago
passes [autoparallel] record parameter attribute in colotracer (#2217) 2 years ago
pipeline_shard [autoparallel] init new folder structure (#1696) 2 years ago
tensor_shard [autoparallel] new metainfoprop based on metainfo class (#2179) 2 years ago
__init__.py