ColossalAI/colossalai/fx/tracer
digger yu 70c8cdecf4
[nfc] fix typo colossalai/cli fx kernel (#3847)
* fix typo colossalai/autochunk auto_parallel amp

* fix typo colossalai/auto_parallel nn utils etc.

* fix typo colossalai/auto_parallel autochunk fx/passes  etc.

* fix typo docs/

* change placememt_policy to placement_policy in docs/ and examples/

* fix typo colossalai/ applications/

* fix typo colossalai/cli fx kernel
2023-06-02 15:02:45 +08:00
..
bias_addition_patch [nfc] fix typo colossalai/cli fx kernel (#3847) 2023-06-02 15:02:45 +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 [nfc] fix typo colossalai/cli fx kernel (#3847) 2023-06-02 15:02:45 +08:00
registry.py [autoparallel] support addbmm computation (#2102) 2022-12-08 21:15:11 +08:00
tracer.py [nfc] fix typo colossalai/cli fx kernel (#3847) 2023-06-02 15:02:45 +08:00