ColossalAI/colossalai/fx/passes/algorithms
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