mirror of https://github.com/hpcaitech/ColossalAI
2cc1175c76
* [fx] tested the complete workflow for auto-parallel * polish code * polish code * polish code |
||
---|---|---|
.. | ||
__init__.py | ||
adding_split_node_pass.py | ||
meta_info_prop.py | ||
shard_1d_pass.py | ||
split_module.py | ||
utils.py |