Skip to content

[PyTorch] Split TE ops op_forward into op_forward and setup_context#2877

Closed
pggPL wants to merge 6 commits into
NVIDIA:mainfrom
pggPL:torch_compile_sequential_split_into_setup_ctx_and_forward
Closed

[PyTorch] Split TE ops op_forward into op_forward and setup_context#2877
pggPL wants to merge 6 commits into
NVIDIA:mainfrom
pggPL:torch_compile_sequential_split_into_setup_ctx_and_forward

Require paired split API methods and widen tensors_to_save types

6202dd0
Select commit
Loading
Failed to load commit list.
DCO / DCO succeeded Apr 16, 2026 in 1s

DCO

All commits are signed off!

View more details on DCO