ColossalAI/colossalai/auto_parallel/checkpoint
Super Daniel cc55ff0aa4
[autoparallel] user-friendly API for CheckpointSolver. (#1879)
Merge for SC tutorial
2022-11-10 20:59:28 +08:00
..
__init__.py
build_c_ext.py [autoparallel] refactor and add rotorc. (#1789) 2022-11-03 12:32:51 +08:00
ckpt_solver_base.py [autoparallel] user-friendly API for CheckpointSolver. (#1879) 2022-11-10 20:59:28 +08:00
ckpt_solver_chen.py
ckpt_solver_rotor.c [autoparallel] refactor and add rotorc. (#1789) 2022-11-03 12:32:51 +08:00
ckpt_solver_rotor.py [autoparallel] user-friendly API for CheckpointSolver. (#1879) 2022-11-10 20:59:28 +08:00
operation.py [autoparallel] refactor and add rotorc. (#1789) 2022-11-03 12:32:51 +08:00