ColossalAI/colossalai/fx/tracer
Super Daniel c41e59e5ad
[fx] allow native ckpt trace and codegen. (#2438)
2023-01-11 13:49:59 +08:00
..
bias_addition_patch
meta_patch
__init__.py
_meta_trace.py
_symbolic_trace.py [fx] allow native ckpt trace and codegen. (#2438) 2023-01-11 13:49:59 +08:00
_tracer_utils.py
experimental.py [fx] allow native ckpt trace and codegen. (#2438) 2023-01-11 13:49:59 +08:00
registry.py
tracer.py