ColossalAI/colossalai/fx/tracer
Michelle ad285e1656
[NFC] polish colossalai/fx/tracer/_tracer_utils.py (#3323)
* [NFC] polish colossalai/engine/schedule/_pipeline_schedule.py code style

* [NFC] polish colossalai/fx/tracer/_tracer_utils.py  code style

---------

Co-authored-by: Qianran Ma <qianranm@luchentech.com>
2023-03-29 17:53:32 +08:00
..
bias_addition_patch [autoparallel] support linear function bias addition (#2104) 2022-12-09 10:31:36 +08:00
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 [NFC] polish colossalai/fx/tracer/_tracer_utils.py (#3323) 2023-03-29 17:53:32 +08:00
experimental.py [fx] allow native ckpt trace and codegen. (#2438) 2023-01-11 13:49:59 +08:00
registry.py [autoparallel] support addbmm computation (#2102) 2022-12-08 21:15:11 +08:00
tracer.py [autoparallel] record parameter attribute in colotracer (#2217) 2022-12-28 19:29:08 +08:00