Skip to content

Commit 58da416

Browse files
committed
radix sort triage test now parametrizes for identical sizes to topk_1d_largest test
1 parent ff81365 commit 58da416

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

dpctl/tests/test_usm_ndarray_top_k.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -96,9 +96,9 @@ def test_topk_1d_smallest(dtype, n):
9696

9797
# triage failing top k radix implementation on CPU
9898
# replicates from Python behavior of radix sort topk implementation
99-
def test_topk_largest_1d_radix_i1_255():
99+
@pytest.mark.parametrize("n", [33, 255, 511, 1021, 8193])
100+
def test_topk_largest_1d_radix_i1_255(n):
100101
get_queue_or_skip()
101-
n = 255
102102
dt = "i1"
103103

104104
o = dpt.ones(n, dtype=dt)

0 commit comments

Comments
 (0)