fixed cache

intel · Nov 22, 2024 · bd4c85c · bd4c85c
1 parent 5f7e2dd
commit bd4c85c
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/sycl/test-e2e/Matrix/joint_matrix_bf16_fill_k_cache_impl.hpp b/sycl/test-e2e/Matrix/joint_matrix_bf16_fill_k_cache_impl.hpp
@@ -483,10 +483,11 @@ size_t matrix_size = -1;
            MCache1, NCache1, KCache1, MCache2, NCache2, KCache2>(matrix_size);
       test<bfloat16, float, VnniFactor, /*TM*/ 32, /*TN*/ 64, /*TK*/ 16,
            MCache1, NCache1, KCache1, MCache2, NCache2, KCache2>(matrix_size);
-      test<bfloat16, float, VnniFactor, /*TM*/ 1, /*TN*/ 64, /*TK*/ 32,
-           MCache1, NCache1, KCache1, MCache2, NCache2, KCache2>(matrix_size);
+      test<bfloat16, float, VnniFactor, /*TM*/ 1, /*TN*/ 64, /*TK*/ 32, MCache1,
+           NCache1, /*KCache1*/ 32, MCache2, NCache2, KCache2>(matrix_size);
       test<bfloat16, float, VnniFactor, /*TM*/ 32, /*TN*/ 64, /*TK*/ 32,
-           MCache1, NCache1, KCache1, MCache2, NCache2, KCache2>(matrix_size);
+           MCache1, NCache1, /*KCache1*/ 32, MCache2, NCache2, KCache2>(
+          matrix_size);
 #endif
       break;
     }