Boyuan Yao b231430bcb [fx] Fix wrong index in annotation and minimal flops in ckpt solver (#1521) ... * [fx] fix wrong variable name in solver rotor * [fx] fix wrong variable name in solver rotor * [fx] fix the discretize bug * [fx] fix the first op in activation checkpoint codegen * [fx] fix some bugs of ckpt solver * [fx] modify test_ckpt_torchvision * [fx] set sequence to __sequence__ attr of GraphModule * [fx] docstring modification * [fx] remove performance test		2022-08-31 18:10:48 +08:00
..
__init__.py	…
ckpt_solver_chen.py	…
ckpt_solver_rotor.py	[fx] Fix wrong index in annotation and minimal flops in ckpt solver (#1521)	2022-08-31 18:10:48 +08:00
linearize.py	…
utils.py	…