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 | ||