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 [fx] added activation checkpoint codegen support for torch < 1.12 (#1359) 2022-07-25 23:35:31 +08:00
activation_checkpoint_codegen.py [fx] Modify solver linearize and add corresponding test (#1531) 2022-09-02 10:24:41 +08:00