dpt.nonzero should use default index type

oleksandr-pavlyk · oleksandr-pavlyk · commit bf22a284dadc · 2023-08-11T08:20:34.000-05:00
Closes gh-1335. The issue was caused by nonzero using default integral data type, not default index data type.
diff --git a/dpctl/tensor/_copy_utils.py b/dpctl/tensor/_copy_utils.py
@@ -586,7 +586,7 @@ def _nonzero_impl(ary):
         mask_nelems, dtype=cumsum_dt, sycl_queue=exec_q, order="C"
     )
     mask_count = ti.mask_positions(ary, cumsum, sycl_queue=exec_q)
-    indexes_dt = ti.default_device_int_type(exec_q.sycl_device)
+    indexes_dt = ti.default_device_index_type(exec_q.sycl_device)
     indexes = dpt.empty(
         (ary.ndim, mask_count),
         dtype=indexes_dt,

Original file line number	Diff line number	Diff line change
`@@ -586,7 +586,7 @@ def _nonzero_impl(ary):`
`586`	`586`	`mask_nelems, dtype=cumsum_dt, sycl_queue=exec_q, order="C"`
`587`	`587`	`)`
`588`	`588`	`mask_count = ti.mask_positions(ary, cumsum, sycl_queue=exec_q)`
`589`		`- indexes_dt = ti.default_device_int_type(exec_q.sycl_device)`
	`589`	`+ indexes_dt = ti.default_device_index_type(exec_q.sycl_device)`
`590`	`590`	`indexes = dpt.empty(`
`591`	`591`	`(ary.ndim, mask_count),`
`592`	`592`	`dtype=indexes_dt,`