ColossalAI/colossalai/fx/tracer
YuliangLiu0306 fea3cb661c
[autoparallel] support addmm in tracer and solver (#1961)
* [fx] patch addmm

* [autoparallel] support addmm in tracer and solver
2022-11-16 14:59:18 +08:00
..
bias_addition_patch [autoparallel] fix bias addition module (#1800) 2022-11-08 16:21:25 +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
_symbolic_trace.py [fx] add a symbolic_trace api. (#1812) 2022-11-08 13:59:20 +08:00
_tracer_utils.py
registry.py [fx] support module with bias addition (#1780) 2022-11-01 22:53:51 +08:00
tracer.py [autoparallel] add getattr handler (#1767) 2022-11-03 12:31:33 +08:00