mirror of https://github.com/hpcaitech/ColossalAI
b904748210
* [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 * [autoparallel] fix wrong runtime apply calculation * [autoparallel] fix wrong runtime apply calculation * [autoparallel] fix wrong runtime apply calculation * [autoparallel] bypass metainfo when available and modify BCAST_FUNC_OP |
||
|---|---|---|
| .. | ||
| meta_registry | ||
| __init__.py | ||
| constants.py | ||
| metainfo.py | ||
| registry.py | ||