We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 1c9e7aa commit 413137aCopy full SHA for 413137a
numba_cuda/numba/cuda/dispatcher.py
@@ -1629,11 +1629,15 @@ def typeof_pyval(self, val):
1629
try:
1630
return typeof(val, Purpose.argument)
1631
except ValueError:
1632
- if cuda.is_cuda_array(val):
+ if (
1633
+ interface := getattr(val, "__cuda_array_interface__")
1634
+ ) is not None:
1635
# When typing, we don't need to synchronize on the array's
1636
# stream - this is done when the kernel is launched.
1637
+
1638
return typeof(
- cuda.as_cuda_array(val, sync=False), Purpose.argument
1639
+ cuda.from_cuda_array_interface(interface, sync=False),
1640
+ Purpose.argument,
1641
)
1642
else:
1643
raise
0 commit comments