pytorch
de0e0300 - Add DLPack support for XPU backend by mapping to kDLOneAPI in DLPack (#82867)

Commit
2 years ago
Add DLPack support for XPU backend by mapping to kDLOneAPI in DLPack (#82867) ## 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. Pull Request resolved: https://github.com/pytorch/pytorch/pull/82867 Approved by: https://github.com/kit1980
Author
Committer
Parents
Loading