Boyuan Yao 49ccf8b5f8 [fx] Improve linearize and rotor solver (#1586) ... * [fx] add nested activation_checkpoint codegen * undo algorithms commits * solver * undo some commits * [fx] torch11 add nested activation checkpoint codegen * remove some imports * [fx] add some comments in activation codegen * [fx] codegen instance error fix * [fx] imporve linearize and rotor solver * [fx] some comments and format modification		2022-09-13 14:50:04 +08:00
..
algorithms	[fx] Improve linearize and rotor solver (#1586)	2022-09-13 14:50:04 +08:00
__init__.py	[fx]add autoparallel passes (#1121)	2022-06-15 16:36:46 +08:00
adding_split_node_pass.py	[fx] update split module pass and add customized policy (#1373)	2022-07-27 13:40:54 +08:00
meta_info_prop.py	[fx] provide a stable but not accurate enough version of profiler. (#1547)	2022-09-07 11:21:04 +08:00
passes_for_gpt2_test.py	[hotfix] fix some bugs during gpt2 testing (#1379)	2022-07-28 17:21:07 +08:00
shard_1d_pass.py	[Doc] add more doc for ColoTensor. (#1458)	2022-08-16 10:38:41 +08:00
split_module.py	[fx] fixed compatiblity issue with torch 1.10 (#1331)	2022-07-18 11:41:27 +08:00
utils.py	[autoparallel] added liveness analysis (#1516)	2022-08-30 15:54:37 +08:00