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. |
||
---|---|---|
.. | ||
experimental | ||
__init__.py | ||
constants.py | ||
dataflow.py | ||
memory_utils.py | ||
opcount.py | ||
profiler.py | ||
shard_utils.py | ||
tensor.py |