ColossalAI/colossalai/fx/codegen
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
activation_checkpoint_codegen.py [fx] Modify solver linearize and add corresponding test (#1531) 2022-09-02 10:24:41 +08:00