Closed njroussel closed 6 months ago
Fixes to the DLPack interfaces:
from_dlpack
dlcapsule
I've also added a new set of roundtrip conversion tests with torch, jax, numpy and tensorflow.
torch
jax
numpy
tensorflow
Perfect, LTGM 👍
Fixes to the DLPack interfaces:
from_dlpack
expected to directly receive thedlcapsule
rather than the Python objectI've also added a new set of roundtrip conversion tests with
torch
,jax
,numpy
andtensorflow
.