ColossalAI/colossalai/fx/tracer/meta_patch/patched_function
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
..
__init__.py [fx] support module with bias addition (#1780) 2022-11-01 22:53:51 +08:00
activation_function.py [fx] support module with bias addition (#1780) 2022-11-01 22:53:51 +08:00
arithmetic.py [autoparallel] support addmm in tracer and solver (#1961) 2022-11-16 14:59:18 +08:00
convolution.py [fx] support module with bias addition (#1780) 2022-11-01 22:53:51 +08:00
embedding.py [fx] support module with bias addition (#1780) 2022-11-01 22:53:51 +08:00
normalization.py [fx] support module with bias addition (#1780) 2022-11-01 22:53:51 +08:00
python_ops.py [fx] support module with bias addition (#1780) 2022-11-01 22:53:51 +08:00
torch_ops.py [fx] support module with bias addition (#1780) 2022-11-01 22:53:51 +08:00