mirror of https://github.com/hpcaitech/ColossalAI
0584654c79
* [fx] change memory.py to memory_utils.py. * [fx] add shard utils. * [fx] fix import. * [fx] check code style. * [fx] add comment. * [autoparallel] first move. * [fx] add time computations. |
||
---|---|---|
.. | ||
__init__.py | ||
build_c_ext.py | ||
ckpt_solver_chen.py | ||
ckpt_solver_pofo.py | ||
ckpt_solver_rotor.py | ||
dynamic_programs.c | ||
linearize.py | ||
operation.py |