google/jax

Support non-trivial strides when importing DLPack tensors

hawkinsp opened this issue · 3 comments

hawkinsp commented
Support non-trivial strides when importing DLPack tensors
jakevdp commented

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

lucidrains commented

i'm interested in this!

mattjj commented

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