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 8ded9c3 commit 3fe2718Copy full SHA for 3fe2718
dpctl/tests/test_usm_ndarray_indexing.py
@@ -1345,3 +1345,15 @@ def test_nonzero_arg_validation():
1345
dpt.nonzero(list())
1346
with pytest.raises(ValueError):
1347
dpt.nonzero(dpt.asarray(1))
1348
+
1349
1350
+def test_nonzero_dtype():
1351
+ "See gh-1322"
1352
+ get_queue_or_skip()
1353
+ x = dpt.ones((3, 4))
1354
+ idx, idy = dpt.nonzero(x)
1355
+ # create array using device's
1356
+ # default integral data type
1357
+ ref = dpt.arange(8)
1358
+ assert idx.dtype == ref.dtype
1359
+ assert idy.dtype == ref.dtype
0 commit comments