ColossalAI/colossalai/fx/tracer
Super Daniel 2edbef13cc
[fx] add more meta_registry for MetaTensor execution. (#2000)
* [sc] add examples for auto checkpoint.

* merge upstream

* [fx] add more meta_registry for MetaTensor execution.
2022-11-23 10:55:46 +08:00
..
bias_addition_patch
meta_patch [autoparallel] support addmm in tracer and solver (#1961) 2022-11-16 14:59:18 +08:00
__init__.py
_meta_trace.py
_symbolic_trace.py [fx] add more meta_registry for MetaTensor execution. (#2000) 2022-11-23 10:55:46 +08:00
_tracer_utils.py
registry.py
tracer.py