mirror of
https://github.com/hpcaitech/ColossalAI.git
synced 2026-04-27 02:03:13 +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
|