Support torch.cuda.*Tensor in Dynamo #156107
Conversation
Summary: This PR adds support for torch.cuda.FloatTensor and friends in Dynamo. These are indeed legacy APIs, but that doesn't stop us from adding support for them in torch.compile. Test Plan: - new test [ghstack-poisoned]
🔗 Helpful Links🧪 See artifacts and rendered test results at hud.pytorch.org/pr/156107
Note: Links to docs will display an error until the docs builds have been completed. ⏳ No Failures, 57 PendingAs of commit 8a07988 with merge base bf7e290 ( This comment was automatically generated by Dr. CI and updates every 15 minutes. |
Summary: This PR adds support for torch.cuda.FloatTensor and friends in Dynamo. These are indeed legacy APIs, but that doesn't stop us from adding support for them in torch.compile. I add support for these in the same way that we support torch.Tensor: these APIs can be safely put into the Dynamo graph. Test Plan: - new test [ghstack-poisoned]
Summary: This PR adds support for torch.cuda.FloatTensor and friends in Dynamo. These are indeed legacy APIs, but that doesn't stop us from adding support for them in torch.compile. I add support for these in the same way that we support torch.Tensor: these APIs can be safely put into the Dynamo graph. Fixes #130722 Test Plan: - new test [ghstack-poisoned]
Summary: This PR adds support for torch.cuda.FloatTensor and friends in Dynamo. These are indeed legacy APIs, but that doesn't stop us from adding support for them in torch.compile. I add support for these in the same way that we support torch.Tensor: these APIs can be safely put into the Dynamo graph. Fixes #130722 Test Plan: - new test ghstack-source-id: 3dfdf21 Pull Request resolved: #156107
zou3519
added
release notes: dynamo
ciflow/trunk
|
@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: 1 jobs have failed, first few of them are: trunk / macos-py3-arm64 / build Details for Dev Infra teamRaised by workflow job |
|
@pytorchbot merge -f "unrelated failure" |
|
Can't merge closed PR #156107 |
|
@pytorchbot merge -f "unrelated failure" |
Merge startedYour change will be merged immediately since you used the force (-f) flag, bypassing any CI checks (ETA: 1-5 minutes). Please use Learn more about merging in the wiki. Questions? Feedback? Please reach out to the PyTorch DevX Team |
3 participants
Stack from ghstack (oldest at bottom):
Summary:
This PR adds support for torch.cuda.FloatTensor and friends in Dynamo.
These are indeed legacy APIs, but that doesn't stop us from adding
support for them in torch.compile.
I add support for these in the same way that we support torch.Tensor:
these APIs can be safely put into the Dynamo graph.
Fixes #130722
Test Plan:
cc @voznesenskym @penguinwu @EikanWang @jgong5 @Guobing-Chen @XiaobingSuper @zhuhaozhe @blzheng @wenzhe-nrv @jiayisunx @chenyang78 @kadeng @chauhang @amjames