mirror of https://github.com/hpcaitech/ColossalAI
2edbef13cc
* [sc] add examples for auto checkpoint. * merge upstream * [fx] add more meta_registry for MetaTensor execution. |
||
---|---|---|
.. | ||
bias_addition_patch | ||
meta_patch | ||
__init__.py | ||
_meta_trace.py | ||
_symbolic_trace.py | ||
_tracer_utils.py | ||
registry.py | ||
tracer.py |