xla icon indicating copy to clipboard operation
xla copied to clipboard

Published 20 hours ago • verified publisher icon pytorch

DLPack: add test using PyTorch DLPack functions.

Open ysiraichi opened this issue 4 months ago • 0 comments

Taking over: #7213

This PR introduces a test using PyTorch's to_dlpack and from_dlpack functions. In order to use the former, PyTorch's tensor __dlpack__ and __dlpack_device__ methods had to be modified. The actual changes are in https://github.com/pytorch/pytorch/pull/138470.

ysiraichi avatar Oct 21 '24 15:10 ysiraichi