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 | ||