mirror of
https://github.com/hpcaitech/ColossalAI.git
synced 2026-05-02 08:37:36 +00:00
de1e716dc4
* [fx] fix defining ckpt functions inside forward * [fx] Modify activation checkpoint codegen and add ColoGraphModule * [fx] some modification * some modifications * some modifications * some modifications * some modifications * some code modifications * [automatic_parallel] ckpt solver rotor * [fx] add ckpt_solver_rotor * [fx] modification * code refactor * code refactor
3 lines
75 B
Python
3 lines
75 B
Python
from .tracer import ColoTracer
|
|
from .graph_module import ColoGraphModule
|