Boyuan Yao 46c6cc79a9 [fx] Add common node in model linearize (#1542) ...

* [fx] Add common node into linearize

* [fx] Add common node to solver

2022-09-05 18:35:05 +08:00

..

__init__.py

[fx] Add activation checkpoint solver rotor (#1496)

2022-08-26 10:34:21 +08:00

ckpt_solver_chen.py

[fx] add profiler for fx nodes. (#1480)

2022-08-24 16:22:44 +08:00

ckpt_solver_rotor.py

[fx] Add common node in model linearize (#1542)

2022-09-05 18:35:05 +08:00

linearize.py

[fx] Add common node in model linearize (#1542)

2022-09-05 18:35:05 +08:00

utils.py

[fx] fix wrong variable name in solver rotor (#1502)

2022-08-26 15:47:08 +08:00