ColossalAI/tests/test_fx/test_ckpt_solvers
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
..
test_ckpt_torchvision.py [fx] Fix wrong index in annotation and minimal flops in ckpt solver (#1521) 2022-08-31 18:10:48 +08:00
test_linearize.py [fx] Modify solver linearize and add corresponding test (#1531) 2022-09-02 10:24:41 +08:00