3 Commits (ae71036cd2210b6e60805357f4bd059674e316bc)

Author	SHA1	Message	Date
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
Boyuan Yao	56159049e8	[fx] Modify solver linearize and add corresponding test (#1531) ... * [fx] modify solver linearize and add test * [fx] add torch11 test of linearize but skip it * [fx] remove some unused imports	2022-09-02 10:24:41 +08:00
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