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

[autoparallel] support addmm in tracer and solver (#1961)

2022-11-16 14:59:18 +08:00

__init__.py

[fx] add a symbolic_trace api. (#1812)

2022-11-08 13:59:20 +08:00

_meta_trace.py

[fx] refactor memory utils and extend shard utils. (#1754)

2022-10-26 14:24:41 +08:00

_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