File tree Expand file tree Collapse file tree 1 file changed +22
-0
lines changed Expand file tree Collapse file tree 1 file changed +22
-0
lines changed Original file line number Diff line number Diff line change @@ -321,3 +321,25 @@ def test_set_functions_compute_follows_data():
321
321
assert ind .sycl_queue == q
322
322
assert inv_ind .sycl_queue == q
323
323
assert uc .sycl_queue == q
324
+
325
+
326
+ def test_gh_1738 ():
327
+ get_queue_or_skip ()
328
+
329
+ ones = dpt .ones (10 , dtype = "i8" )
330
+ iota = dpt .arange (10 , dtype = "i8" )
331
+
332
+ assert ones .device == iota .device
333
+
334
+ dpt_info = dpt .__array_namespace_info__ ()
335
+ ind_dt = dpt_info .default_dtypes (device = ones .device )["indexing" ]
336
+
337
+ dt = dpt .unique_inverse (ones ).inverse_indices .dtype
338
+ assert dt == ind_dt
339
+ dt = dpt .unique_all (ones ).inverse_indices .dtype
340
+ assert dt == ind_dt
341
+
342
+ dt = dpt .unique_inverse (iota ).inverse_indices .dtype
343
+ assert dt == ind_dt
344
+ dt = dpt .unique_all (iota ).inverse_indices .dtype
345
+ assert dt == ind_dt
You can’t perform that action at this time.
0 commit comments