ColossalAI/colossalai/fx/passes/experimental
YuliangLiu0306 451cd72dea
[autoparallel] adapt runtime passes (#1703)
* [autoparallel] adapt runtime passes v2

* polish code
2022-10-14 10:14:07 +08:00
..
adding_shape_consistency_pass.py [fx] PoC of runtime shape consistency application (#1607) 2022-09-20 14:00:04 +08:00
adding_shape_consistency_pass_v2.py [autoparallel] adapt runtime passes (#1703) 2022-10-14 10:14:07 +08:00