xla
DLPack: add test using PyTorch DLPack functions.
#8294
Merged

DLPack: add test using PyTorch DLPack functions. #8294

qihqi merged 4 commits into master from ysiraichi/dlpack-tensor-test
ysiraichi
vanbasten23 Add a test to convert xla gpu tensor to cuda tensor by using the prot…
3b1602d9
vanbasten23 add torch pin
a0de56a4
ysiraichi Make test use PyTorch's `to_dlpack`.
2bdcd83a
ysiraichi Update torch_pin.
4a86adfd
ysiraichi ysiraichi requested a review from JackCaoG JackCaoG 1 year ago
ysiraichi ysiraichi removed review request from JackCaoG JackCaoG 1 year ago
ysiraichi ysiraichi requested a review from vanbasten23 vanbasten23 1 year ago
ysiraichi ysiraichi requested a review from JackCaoG JackCaoG 1 year ago
JackCaoG
JackCaoG approved these changes on 2024-10-21
vanbasten23
vanbasten23 approved these changes on 2024-10-21
qihqi qihqi merged dc20b2d9 into master 1 year ago
qihqi qihqi deleted the ysiraichi/dlpack-tensor-test branch 1 year ago
ysiraichi
ysiraichi
ysiraichi commented on 2024-11-01

Login to write a write a comment.

Login via GitHub

Reviewers
Assignees
No one assigned
Labels
Milestone