mirror of https://github.com/hpcaitech/ColossalAI
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. |
||
---|---|---|
.. | ||
__init__.py | ||
activation_checkpoint_codegen.py |