google/jax

Support non-trivial strides when importing DLPack tensors

hawkinsp opened this issue · 3 comments

Support non-trivial strides when importing DLPack tensors

Related bug: #7657 The bug was closed due to a suitable workaround being proposed, but I think the underlying issue is still there.

i'm interested in this!

Just so we have a test case: running the code in this gist without the call to .contiguous() results in an error.