ColossalAI/colossalai/fx/passes/algorithms
Boyuan Yao 56159049e8
[fx] Modify solver linearize and add corresponding test (#1531)
* [fx] modify solver linearize and add test

* [fx] add torch11 test of linearize but skip it

* [fx] remove some unused imports
2022-09-02 10:24:41 +08:00
..
__init__.py
ckpt_solver_chen.py
ckpt_solver_rotor.py [fx] Modify solver linearize and add corresponding test (#1531) 2022-09-02 10:24:41 +08:00
linearize.py [fx] Modify solver linearize and add corresponding test (#1531) 2022-09-02 10:24:41 +08:00
utils.py