mirror of https://github.com/hpcaitech/ColossalAI
fea3cb661c
* [fx] patch addmm * [autoparallel] support addmm in tracer and solver |
||
---|---|---|
.. | ||
__init__.py | ||
activation_function.py | ||
arithmetic.py | ||
convolution.py | ||
embedding.py | ||
normalization.py | ||
python_ops.py | ||
torch_ops.py |