mirror of
https://github.com/hpcaitech/ColossalAI.git
synced 2025-06-18 19:58:17 +00:00
1e88811c7a
* [autoparallel] first move. * [autoparallel] add solver rotor. * [autoparallel] add ckpt solvers. * [autoparallel] modify codegen. * [fx] fix annotation in test. * [fx] remove check. * [autoparallel] polish docstring. * [fx] refactor MetaTensor.
4 lines
155 B
Python
4 lines
155 B
Python
from .ckpt_solver_base import CheckpointSolverBase
|
|
from .ckpt_solver_chen import CheckpointSolverChen
|
|
from .ckpt_solver_rotor import CheckpointSolverRotor
|