digger yu 32f81f14d4 [NFC] fix typo colossalai/amp auto_parallel autochunk (#3756)

2023-05-19 13:50:00 +08:00

..

meta_registry

[NFC] fix typo colossalai/amp auto_parallel autochunk (#3756)

2023-05-19 13:50:00 +08:00

__init__.py

[autoparallel]integrate auto parallel feature with new tracer (#3408)

2023-04-04 17:40:45 +08:00

constants.py

[autoparallel] align the data_ptr with the old version of auto activation checkpoint pipeline (#2261)

2023-01-03 10:30:15 +08:00

registry.py

[fx] Add linear metainfo class for auto parallel (#1783)

2022-11-04 10:55:09 +08:00

shard_metainfo.py

[autoparallel]integrate auto parallel feature with new tracer (#3408)

2023-04-04 17:40:45 +08:00