Skip to content

Commit 702b707

Browse files
Added comments to the test file
1 parent e92d1f9 commit 702b707

File tree

1 file changed

+3
-0
lines changed

1 file changed

+3
-0
lines changed

dpctl/tests/test_usm_ndarray_reductions.py

+3
Original file line numberDiff line numberDiff line change
@@ -169,9 +169,12 @@ def test_search_reduction_kernels(arg_dtype):
169169
m = dpt.argmax(x)
170170
assert m == idx
171171

172+
# test case of strided input mapping to contig
173+
# implementation
172174
m = dpt.argmax(dpt.flip(x))
173175
assert m == x.size - 1 - idx
174176

177+
# test case of strided implementation
175178
y = dpt.ones(2 * x.size, dtype=arg_dtype, sycl_queue=q)
176179
y[::2] = x
177180
m = dpt.argmax(y)

0 commit comments

Comments
 (0)