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 bf22a28 commit e70891bCopy full SHA for e70891b
dpctl/tests/test_usm_ndarray_indexing.py
@@ -22,6 +22,7 @@
22
23
import dpctl
24
import dpctl.tensor as dpt
25
+import dpctl.tensor._tensor_impl as ti
26
from dpctl.utils import ExecutionPlacementError
27
28
_all_dtypes = [
@@ -1353,7 +1354,7 @@ def test_nonzero_dtype():
1353
1354
x = dpt.ones((3, 4))
1355
idx, idy = dpt.nonzero(x)
1356
# create array using device's
- # default integral data type
1357
- ref = dpt.arange(8)
1358
- assert idx.dtype == ref.dtype
1359
- assert idy.dtype == ref.dtype
+ # default index data type
+ index_dt = dpt.dtype(ti.default_device_index_type(x.sycl_queue))
+ assert idx.dtype == index_dt
1360
+ assert idy.dtype == index_dt
0 commit comments