github
/
ColossalAI
mirror of https://github.com/hpcaitech/ColossalAI
1
0
Fork 0
Code Issues Projects Releases Wiki Activity
ColossalAI/colossalai/fx/passes/algorithms/__init__.py

2 lines
43 B
Python
Raw Normal View History

[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 06:47:12 +00:00
from .ckpt_solver_chen import chen_greedy