You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.

Boyuan Yao a7cda6f57d [fx] Add offload codegen (#1598) ... * [fx] add input activation offload to codegen * [fx] modify unit test * [fx] remove two skips in torch11 * [fx] use all_input_nodes instead of _input_nodes		2 years ago
..
test_activation_checkpoint_codegen.py	[fx] Add offload codegen (#1598)	2 years ago
test_nested_activation_checkpoint_codegen.py	[fx] Add nested checkpoint in activation checkpoint codegen (#1585)	2 years ago
test_offload_codegen.py	[fx] Add offload codegen (#1598)	2 years ago