Boyuan Yao de1e716dc4 [fx] Add activation checkpoint solver rotor (#1496) ... * [fx] fix defining ckpt functions inside forward * [fx] Modify activation checkpoint codegen and add ColoGraphModule * [fx] some modification * some modifications * some modifications * some modifications * some modifications * some code modifications * [automatic_parallel] ckpt solver rotor * [fx] add ckpt_solver_rotor * [fx] modification * code refactor * code refactor		2022-08-26 10:34:21 +08:00
..
codegen	[fx] Fix ckpt functions' definitions in forward (#1476)	2022-08-22 16:59:54 +08:00
passes	[fx] Add activation checkpoint solver rotor (#1496)	2022-08-26 10:34:21 +08:00
profiler	[fx] add more op patches for profiler and error message for unsupported ops. (#1495)	2022-08-25 23:11:13 +08:00
tracer	[autoparallel] add cost graph class (#1481)	2022-08-25 17:19:59 +08:00
__init__.py	[fx] Add activation checkpoint solver rotor (#1496)	2022-08-26 10:34:21 +08:00
graph_module.py	[fx] Fix ckpt functions' definitions in forward (#1476)	2022-08-22 16:59:54 +08:00
proxy.py	[doc] update rst and docstring (#1351)	2022-07-21 15:54:53 +08:00