ColossalAI/tests/test_fx/test_meta/test_aten.py at e99edfcb51df48dec17498c60bbbd06baa293c22

mirror of https://github.com/hpcaitech/ColossalAI.git synced 2025-12-05 07:34:25 +00:00

Files

Super Daniel 393f594051 [fx/meta/rpc] move _meta_registration.py to fx folder / register fx functions with compatibility checks / remove color debug (#1710 )

* [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.

2022-10-18 10:44:23 +08:00

3.6 KiB

Raw Blame History

View Raw

3.6 KiB Raw Blame History

3.6 KiB

Raw Blame History