mirror of https://github.com/hpcaitech/ColossalAI
ab38aebace
* [autoparallel] hook node meta on graph nodes for checkpoint solver * [autoparallel] polish code * [autoparallel] restore some node handlers * colossalai/auto_parallel/passes/meta_info_prop.py * [autoparallel] remove some unused import * [autoparallel] hook bwd_mem_out |
||
|---|---|---|
| .. | ||
| meta_registry | ||
| __init__.py | ||
| constants.py | ||
| metainfo.py | ||
| registry.py | ||