Skip to content

Commit 42d4c6f

Browse files
committed
update tuned factors for fp16
1 parent 328d0aa commit 42d4c6f

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

tests/python/unittest/test_tir_schedule_tensorize_ldmatrix_mma.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -288,7 +288,7 @@ def index_map(i, j):
288288
k_inner = 16
289289
in_dtype = "float16"
290290
out_dtype = "float16"
291-
i_factors, j_factors, k_factors = [4, 8, 2, 4, 1], [1, 64, 2, 1, 2], [128, 2, 1]
291+
i_factors, j_factors, k_factors = [16, 2, 1, 4, 2], [16, 2, 2, 1, 4], [128, 2, 1]
292292

293293
timer = run_test(
294294
k_inner,

0 commit comments

Comments
 (0)