Boyuan Yao 49ccf8b5f8 [fx] Improve linearize and rotor solver (#1586) ... * [fx] add nested activation_checkpoint codegen * undo algorithms commits * solver * undo some commits * [fx] torch11 add nested activation checkpoint codegen * remove some imports * [fx] add some comments in activation codegen * [fx] codegen instance error fix * [fx] imporve linearize and rotor solver * [fx] some comments and format modification		2022-09-13 14:50:04 +08:00
..
__init__.py	[fx] Add activation checkpoint solver rotor (#1496)	2022-08-26 10:34:21 +08:00
ckpt_solver_chen.py	[fx] provide a stable but not accurate enough version of profiler. (#1547)	2022-09-07 11:21:04 +08:00
ckpt_solver_rotor.py	[fx] Improve linearize and rotor solver (#1586)	2022-09-13 14:50:04 +08:00
linearize.py	[fx] Improve linearize and rotor solver (#1586)	2022-09-13 14:50:04 +08:00
utils.py	[fx] fix wrong variable name in solver rotor (#1502)	2022-08-26 15:47:08 +08:00