aot autograd: dont allow symint outputs to get tangents in the bw graph#96219
aot autograd: dont allow symint outputs to get tangents in the bw graph#96219bdhirsh wants to merge 3 commits intogh/bdhirsh/391/basefrom
Conversation
[ghstack-poisoned]
🔗 Helpful Links🧪 See artifacts and rendered test results at hud.pytorch.org/pr/96219
Note: Links to docs will display an error until the docs builds have been completed. ✅ No FailuresAs of commit e6858d6: This comment was automatically generated by Dr. CI and updates every 15 minutes. |
|
@pytorchbot merge |
Merge startedYour change will be merged once all checks pass (ETA 0-4 Hours). Learn more about merging in the wiki. Questions? Feedback? Please reach out to the PyTorch DevX Team |
Merge failedReason: 4 mandatory check(s) failed. The first few are:
Dig deeper by viewing the failures on hud |
… the bw graph" Previously, if dynamic shapes were turned on and we had a forward graph that returns a symint, then we would generate a backward graph that takes in a tangent input for that symint fwd output. This causes problems for downstream - inductor will see an input that it expects to be a symint, but it gets a `None` from autograd. Confirmed that this repro now passes: ``` benchmarks/dynamo/torchbench.py --devices cuda --inductor --dynamic-shapes --unspecialize-int --accuracy --training --only drq ``` [ghstack-poisoned]
|
@pytorchbot merge |
Merge startedYour change will be merged once all checks pass (ETA 0-4 Hours). Learn more about merging in the wiki. Questions? Feedback? Please reach out to the PyTorch DevX Team |
|
@pytorchbot merge |
|
The merge job was canceled. If you believe this is a mistake,then you can re trigger it through pytorch-bot. |
Merge startedYour change will be merged once all checks pass (ETA 0-4 Hours). Learn more about merging in the wiki. Questions? Feedback? Please reach out to the PyTorch DevX Team |
|
@pytorchbot merge |
|
The merge job was canceled. If you believe this is a mistake,then you can re trigger it through pytorch-bot. |
Merge startedYour change will be merged once all checks pass (ETA 0-4 Hours). Learn more about merging in the wiki. Questions? Feedback? Please reach out to the PyTorch DevX Team |
Merge failedReason: 2 mandatory check(s) failed. The first few are:
Dig deeper by viewing the failures on hud |
… the bw graph" Previously, if dynamic shapes were turned on and we had a forward graph that returns a symint, then we would generate a backward graph that takes in a tangent input for that symint fwd output. This causes problems for downstream - inductor will see an input that it expects to be a symint, but it gets a `None` from autograd. Confirmed that this repro now passes: ``` benchmarks/dynamo/torchbench.py --devices cuda --inductor --dynamic-shapes --unspecialize-int --accuracy --training --only drq ``` cc anijain2305 [ghstack-poisoned]
|
@pytorchbot merge |
Merge startedYour change will be merged once all checks pass (ETA 0-4 Hours). Learn more about merging in the wiki. Questions? Feedback? Please reach out to the PyTorch DevX Team |
This assert would have caught #96219 Signed-off-by: Edward Z. Yang <[email protected]> [ghstack-poisoned]
This assert would have caught #96219 Signed-off-by: Edward Z. Yang <[email protected]> Pull Request resolved: #96300 Approved by: https://github.com/bdhirsh
…ph (#96219) Previously, if dynamic shapes were turned on and we had a forward graph that returns a symint, then we would generate a backward graph that takes in a tangent input for that symint fwd output. This causes problems for downstream - inductor will see an input that it expects to be a symint, but it gets a `None` from autograd. Confirmed that this repro now passes: ``` benchmarks/dynamo/torchbench.py --devices cuda --inductor --dynamic-shapes --unspecialize-int --accuracy --training --only drq ``` Pull Request resolved: pytorch/pytorch#96219 Approved by: https://github.com/ezyang
…ph (#96219) Previously, if dynamic shapes were turned on and we had a forward graph that returns a symint, then we would generate a backward graph that takes in a tangent input for that symint fwd output. This causes problems for downstream - inductor will see an input that it expects to be a symint, but it gets a `None` from autograd. Confirmed that this repro now passes: ``` benchmarks/dynamo/torchbench.py --devices cuda --inductor --dynamic-shapes --unspecialize-int --accuracy --training --only drq ``` Pull Request resolved: pytorch/pytorch#96219 Approved by: https://github.com/ezyang
This assert would have caught pytorch/pytorch#96219 Signed-off-by: Edward Z. Yang <[email protected]> Pull Request resolved: pytorch/pytorch#96300 Approved by: https://github.com/bdhirsh
…ph (pytorch#96219) Previously, if dynamic shapes were turned on and we had a forward graph that returns a symint, then we would generate a backward graph that takes in a tangent input for that symint fwd output. This causes problems for downstream - inductor will see an input that it expects to be a symint, but it gets a `None` from autograd. Confirmed that this repro now passes: ``` benchmarks/dynamo/torchbench.py --devices cuda --inductor --dynamic-shapes --unspecialize-int --accuracy --training --only drq ``` Pull Request resolved: pytorch#96219 Approved by: https://github.com/ezyang
This assert would have caught pytorch#96219 Signed-off-by: Edward Z. Yang <[email protected]> Pull Request resolved: pytorch#96300 Approved by: https://github.com/bdhirsh
Previously, if dynamic shapes were turned on and we had a forward graph that returns a symint, then we would generate a backward graph that takes in a tangent input for that symint fwd output. This causes problems for downstream - inductor will see an input that it expects to be a symint, but it gets a
Nonefrom autograd.Confirmed that this repro now passes:
cc @anijain2305
Stack from ghstack (oldest at bottom):