Boyuan Yao 5c2ef9fc76 [autoparallel] modify comm nodes' memory cost in construct chain (#2263) ...

* [autoparallel] align the data_ptr with the old version of auto activation checkpoint pipeline

* [autoparallel] using fwd_time and bwd_time instead of fwd_flop and bwd_flop

* [autoparallel] specifycomm nodes' memory cost in construct chain

2023-01-03 11:38:48 +08:00

..

checkpoint

[autoparallel] modify comm nodes' memory cost in construct chain (#2263)

2023-01-03 11:38:48 +08:00

meta_profiler

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

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

passes

[autoparallel] modify comm nodes' memory cost in construct chain (#2263)

2023-01-03 11:38:48 +08:00

pipeline_shard

[autoparallel] init new folder structure (#1696)

2022-10-13 14:18:55 +08:00

tensor_shard

[autockpt] make it work. (#2257)

2023-01-02 23:37:45 +08:00

__init__.py

[autoparallel] standardize the code structure (#1469)

2022-08-19 15:51:54 +08:00