[ET Device Support] TensorImpl carries device info#18431
[ET Device Support] TensorImpl carries device info#18431pytorchbot wants to merge 1 commit intomainfrom
Conversation
Pull Request resolved: #17534 This diff extends `TensorImpl` to carry device information, enabling the runtime tensor to track which device its data resides on (CPU, CUDA, etc.). This is a prerequisite for parsing device info from the schema and allocating device memory. ghstack-source-id: 356251710 @exported-using-ghexport Differential Revision: [D93635655](https://our.internmc.facebook.com/intern/diff/D93635655/)
🔗 Helpful Links🧪 See artifacts and rendered test results at hud.pytorch.org/pr/pytorch/executorch/18431
Note: Links to docs will display an error until the docs builds have been completed. ❌ 1 Cancelled Job, 2 Unrelated FailuresAs of commit bb590bb with merge base 60d57e5 ( CANCELLED JOB - The following job was cancelled. Please retry:
BROKEN TRUNK - The following jobs failed but were present on the merge base:👉 Rebase onto the `viable/strict` branch to avoid these failures
This comment was automatically generated by Dr. CI and updates every 15 minutes. |
This PR needs a
|
This PR was created by the merge bot to help merge the original PR into the main branch.
ghstack PR number: #17534 by @Gasoonjia
^ Please use this as the source of truth for the PR details, comments, and reviews
ghstack PR base: https://github.com/pytorch/executorch/tree/gh/gasoonjia/123/base
ghstack PR head: https://github.com/pytorch/executorch/tree/gh/gasoonjia/123/head
Merge bot PR base: https://github.com/pytorch/executorch/tree/main
Merge bot PR head: https://github.com/pytorch/executorch/tree/gh/gasoonjia/123/orig
Differential Revision: D93635655
@diff-train-skip-merge