mirror of https://github.com/hpcaitech/ColossalAI
393f594051
* [fx] move meta registration * [fx] fix tests. * [fx] fix test. * [fx] fix. * [meta] refactor meta registration.py. * [fx] add compatibility descriptions. * [fx] polish import. * [fx] add a decorator. * [fx] fix tests. * [fx] remove print. * [fx] edit raise error. * [fx] edit raise error. * [fx] add type hint. * [fx] fix import in experimental. * [rpc] remove color debug. * [meta] fix naming. |
||
---|---|---|
.. | ||
__init__.py | ||
activation_function.py | ||
arithmetic.py | ||
embedding.py | ||
linear.py | ||
normalization.py | ||
pooling.py | ||
python_ops.py | ||
torch_ops.py |