mirror of https://github.com/hpcaitech/ColossalAI
2edbef13cc
* [sc] add examples for auto checkpoint. * merge upstream * [fx] add more meta_registry for MetaTensor execution. |
||
---|---|---|
.. | ||
experimental | ||
__init__.py | ||
constants.py | ||
dataflow.py | ||
memory_utils.py | ||
opcount.py | ||
profiler.py | ||
shard_utils.py | ||
tensor.py |