ColossalAI/colossalai/fx/passes/algorithms
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
..
__init__.py [fx] Add pofo solver (#1608) 2022-09-20 11:20:48 +08:00
build_c_ext.py [autoparallel] add rotor C version (#1658) 2022-10-03 17:13:30 +08:00
ckpt_solver_chen.py [fx/profiler] assigned UUID to each unrecorded tensor/ improved performance on GPT-2 (#1679) 2022-10-11 11:03:35 +08:00
ckpt_solver_pofo.py [fx/meta/rpc] move _meta_registration.py to fx folder / register fx functions with compatibility checks / remove color debug (#1710) 2022-10-18 10:44:23 +08:00
ckpt_solver_rotor.py [autoparallel] fix C version rotor inconsistency (#1691) 2022-10-12 15:21:58 +08:00
dynamic_programs.c [autoparallel] fix C version rotor inconsistency (#1691) 2022-10-12 15:21:58 +08:00
linearize.py [fx/tuning] tune performance on rotor with meta info. (#1599) 2022-09-15 14:46:36 +08:00
operation.py [autoparallel] Add pofo sequence annotation (#1637) 2022-09-24 01:52:57 +08:00