Skip to content

Commit 895ad60

Browse files
committed
Removed dlpack extraction leading to forced legacy path
1 parent f2d2d71 commit 895ad60

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

jax/_src/numpy/lax_numpy.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -2438,7 +2438,7 @@ def fromiter(*args, **kwargs):
24382438
""")
24392439
def from_dlpack(x: Any) -> Array:
24402440
from jax.dlpack import from_dlpack # pylint: disable=g-import-not-at-top
2441-
return from_dlpack(x.__dlpack__())
2441+
return from_dlpack(x)
24422442

24432443
@util.implements(np.fromfunction)
24442444
def fromfunction(function: Callable[..., Array], shape: Any,

0 commit comments

Comments
 (0)