You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
ColossalAI/colossalai/auto_parallel/checkpoint
Boyuan Yao ac3739930d
[autoparallel] modify construct chain in rotor solver (#2254)
2 years ago
..
__init__.py [autoparallel] move ckpt solvers to autoparallel folder / refactor code (#1764) 2 years ago
build_c_ext.py [autoparallel] refactor and add rotorc. (#1789) 2 years ago
ckpt_solver_base.py [autoparallel] user-friendly API for CheckpointSolver. (#1879) 2 years ago
ckpt_solver_chen.py [autoparallel] move ckpt solvers to autoparallel folder / refactor code (#1764) 2 years ago
ckpt_solver_rotor.c [autoparallel] refactor and add rotorc. (#1789) 2 years ago
ckpt_solver_rotor.py [autoparallel] modify construct chain in rotor solver (#2254) 2 years ago
operation.py [autoparallel] refactor and add rotorc. (#1789) 2 years ago