mirror of https://github.com/hpcaitech/ColossalAI
5c2ef9fc76
* [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 |
||
---|---|---|
.. | ||
__init__.py | ||
comm_metainfo_pass.py | ||
constants.py | ||
meta_info_prop.py | ||
runtime_apply_pass.py | ||
runtime_preparation_pass.py |