3 Commits (07f5a4e054e183f4604734b174fda5a84c5cd652)

Author	SHA1	Message	Date
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
Super Daniel	e7383f578b	[fx] add rules to linearize computation graphs for searching. (#1461) ... * [fx] polish ckpt_test. * [fx] add rules to linearize computation graphs for searching. * [fx] remove chen_sqrt for sake of simplicity * [fx] fix inconsistencies.	2022-08-17 14:47:12 +08:00
Super Daniel	3b26516c69	[fx] add vanilla activation checkpoint search with test on resnet and densenet (#1433) ... * [fx] activation checkpointing using Chen strategies. * [fx] add test for ckpt_solver_chen * [fx] add vanilla activation checkpoint search with test on resnet and densenet * [fx] add vanilla activation checkpoint search with test on resnet and densenet * [fx] add a namespace code for solver_chen.	2022-08-11 15:46:39 +08:00