ColossalAI

Author	SHA1	Message	Date
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
Boyuan Yao	ac3a453a50	[fx] fix the discretize bug (#1506 ) * [fx] fix wrong variable name in solver rotor * [fx] fix wrong variable name in solver rotor * code modification * [fx] fix the discretize bug	2022-08-26 17:15:52 +08:00
Boyuan Yao	31fffd3fc5	[fx] fix wrong variable name in solver rotor (#1502 ) * [fx] fix wrong variable name in solver rotor * [fx] fix wrong variable name in solver rotor * code modification	2022-08-26 15:47:08 +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