Skip to content

GitLab

Switch branch/tag
Find file BlameHistoryPermalink
  • Boyuan Yao's avatar
    [fx] Add offload codegen (#1598) · a7cda6f5
    Boyuan Yao authored 
    * [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
    a7cda6f5
activation_checkpoint_codegen.py 39 KB