DLPack: add test using PyTorch DLPack functions. #8294
Add a test to convert xla gpu tensor to cuda tensor by using the prot…
3b1602d9
add torch pin
a0de56a4
Make test use PyTorch's `to_dlpack`.
2bdcd83a
Update torch_pin.
4a86adfd
JackCaoG
approved these changes
on 2024-10-21
qihqi
merged
dc20b2d9
into master 1 year ago
qihqi
deleted the ysiraichi/dlpack-tensor-test branch 1 year ago
Assignees
No one assigned
Login to write a write a comment.
Login via GitHub