From 42d4c6f42182c9fd79566c0955f99cc82abd5144 Mon Sep 17 00:00:00 2001 From: Masahiro Masuda Date: Wed, 18 May 2022 09:53:57 +0900 Subject: [PATCH] update tuned factors for fp16 --- .../python/unittest/test_tir_schedule_tensorize_ldmatrix_mma.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/python/unittest/test_tir_schedule_tensorize_ldmatrix_mma.py b/tests/python/unittest/test_tir_schedule_tensorize_ldmatrix_mma.py index cbede480ebea4..d6b14bc1a8cfa 100644 --- a/tests/python/unittest/test_tir_schedule_tensorize_ldmatrix_mma.py +++ b/tests/python/unittest/test_tir_schedule_tensorize_ldmatrix_mma.py @@ -288,7 +288,7 @@ def index_map(i, j): k_inner = 16 in_dtype = "float16" out_dtype = "float16" - i_factors, j_factors, k_factors = [4, 8, 2, 4, 1], [1, 64, 2, 1, 2], [128, 2, 1] + i_factors, j_factors, k_factors = [16, 2, 1, 4, 2], [16, 2, 2, 1, 4], [128, 2, 1] timer = run_test( k_inner,