ColossalAI/tests/test_fx/test_ckpt_solvers
Super Daniel e7383f578b
[fx] add rules to linearize computation graphs for searching. (#1461)
* [fx] polish ckpt_test.

* [fx] add rules to linearize computation graphs for searching.

* [fx] remove chen_sqrt for sake of simplicity

* [fx] fix inconsistencies.
2022-08-17 14:47:12 +08:00
..
test_ckpt_torchvision.py [fx] add rules to linearize computation graphs for searching. (#1461) 2022-08-17 14:47:12 +08:00