mirror of https://github.com/hpcaitech/ColossalAI
44ea461890
* use Topo class to rewrite DAG * polish code * polish code * polish code * add comment * add else to unended if Co-authored-by: Ziyue Jiang <ziyue.jiang@gmail.com> |
||
|---|---|---|
| .. | ||
| algorithms | ||
| experimental | ||
| __init__.py | ||
| adding_split_node_pass.py | ||
| concrete_info_prop.py | ||
| meta_info_prop.py | ||
| passes_for_gpt2_test.py | ||
| shard_1d_pass.py | ||
| split_module.py | ||
| utils.py | ||