[PyTorch] torch.compile for an OperationFuser group holding one operation - #3496

Draft

pggPL wants to merge 6 commits into

NVIDIA:mainfrom

pggPL:ops_fuser_compile_main

Commits on Sep 4, 2026

[PyTorch] Register an op's forward and backward without autograd glue
pggPL
committed
cb69309
[PyTorch] Compile an OperationFuser group holding one operation
pggPL
committed
419c4e2
[PyTorch] Accept an operation's declared forward kwargs under compile

pggPL
and
claude
committed
7cf976f
[PyTorch] Restrict compiled forward kwargs to tensors

pggPL
and
claude
committed
4f773fc
[PyTorch] Refine fusible custom-op integration
pggPL
committed
97e4764

Commits on Sep 8, 2026

[PyTorch] Register custom ops one at a time

pggPL
and
claude
committed
2281cd5