tree dfb50906b74ba16f38259cf9e58a4fc24f7576d6
parent 2ad011ca73f68185783ec9afeb730615769a3fca
author Simon Fan <xmfan@meta.com> 1723188760 -0700
committer PyTorch MergeBot <pytorchmergebot@users.noreply.github.com> 1723287944 +0000

[compiled autograd][cpp node] No recompiles from saved int scalars (#132771)

Addresses https://github.com/pytorch/pytorch/issues/130170 for int scalars saved from forward pass of a custom c++ autograd function

Currently, any non-tensors passed in ctx->saved_data is specialized on by compiled autograd. To stop specializing on int values, we lift the ints. We also require user code to use IValue::toSymInt instead of IValue::toInt in order to swap the SymInt to proxy during compiled autograd tracing

Pull Request resolved: https://github.com/pytorch/pytorch/pull/132771
Approved by: https://github.com/jansel