github/ColossalAI
1
0
Fork 0
mirror of https://github.com/hpcaitech/ColossalAI.git synced 2026-07-11 00:56:38 +00:00
Code Issues Packages Projects Releases Wiki Activity
Files
26a37b5cd5df55382e7ef9799cfa852edcf6fef2
ColossalAI/tests/test_fx/test_ckpt_solvers
History
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