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.
ColossalAI/colossalai/fx/tracer
digger yu 70c8cdecf4
[nfc] fix typo colossalai/cli fx kernel (#3847)
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
_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