New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom

[fx] Add pofo solver #1608

Merged

FrankLeeeee merged 38 commits into hpcaitech:main from Cypher30:feature/add_ckpt_offload_solver

Sep 20, 2022

colossalai/fx/passes/algorithms/__init__.py

-Original file line number
+Diff line change
@@ -1,3 +1,4 @@
     from .ckpt_solver_chen import chen_greedy
     from .linearize import linearize
     from .ckpt_solver_rotor import solver_rotor
+    from .ckpt_solver_pofo import solver_pofo