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. |
||
---|---|---|
.. | ||
test_aten.py | ||
test_backward.py | ||
test_meta_trace.py |