pytorch
53c9bc8c - Add DLPack support for XPU backend by mapping to kDLOneAPI in DLPack … (#94968)

Commit
2 years ago
Add DLPack support for XPU backend by mapping to kDLOneAPI in DLPack … (#94968) # Motivation The DLPack device type kDLOneAPI stands for the Unified Shared Memory allocated on a oneAPI device. The corresponding Pytorch backend type is XPU. Support to export/import the Pytorch XPU tensor as a DLPack tensor of kDLOneAPI device. # Solution 1. Update the DLPack protocol to v0.7. 2. Add the XPU hooks to map the Aten device and DLPack device with the address value and device information. # Additional Context Reopen (#82867) Pull Request resolved: https://github.com/pytorch/pytorch/pull/94968 Approved by: https://github.com/kit1980
Author
Committer
Parents
Loading