You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.

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		2 years ago
..
bias_addition_patch	[nfc] fix typo colossalai/cli fx kernel (#3847)	2 years ago
meta_patch	[autoparallel] support addmm in tracer and solver (#1961)	2 years ago
__init__.py	[fx] add a symbolic_trace api. (#1812)	2 years ago
_meta_trace.py	[fx] refactor memory utils and extend shard utils. (#1754)	2 years ago
_symbolic_trace.py	[fx] allow native ckpt trace and codegen. (#2438)	2 years ago
_tracer_utils.py	[NFC] polish colossalai/fx/tracer/_tracer_utils.py (#3323)	2 years ago
experimental.py	[nfc] fix typo colossalai/cli fx kernel (#3847)	2 years ago
registry.py	[autoparallel] support addbmm computation (#2102)	2 years ago
tracer.py	[nfc] fix typo colossalai/cli fx kernel (#3847)	2 years ago