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. |
||
---|---|---|
.. | ||
meta_patch | ||
__init__.py | ||
_meta_trace.py | ||
_tracer_utils.py | ||
tracer.py |