mirror of https://github.com/hpcaitech/ColossalAI
fea3cb661c
* [fx] patch addmm * [autoparallel] support addmm in tracer and solver |
||
---|---|---|
.. | ||
bias_addition_patch | ||
meta_patch | ||
__init__.py | ||
_meta_trace.py | ||
_symbolic_trace.py | ||
_tracer_utils.py | ||
registry.py | ||
tracer.py |