mirror of https://github.com/hpcaitech/ColossalAI
632753abbc
* add DAG to split_module * add comment * add test case for DAG * remove print 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 | ||