[SYCL][E2E][Joint Matrix] Add k=32 for bfloat16 tests

Author: YuriPlyakhin

sycl/test-e2e/Matrix/Inputs/element_wise_all_ops_impl.hpp

@@ -229,8 +229,11 @@ int main() {
         test_ewops_c<float, 16, 16>();
 // This combination is not currently supported for sub group size = 32 in IGC
 #if (!defined(SG_SZ) || SG_SZ != 32)
+        test_ewops_ab<bfloat16, 1, 32, use::a, layout::row_major, 1>();
         test_ewops_ab<bfloat16, 32, 16, use::a, layout::row_major, 1>();
+        test_ewops_ab<bfloat16, 32, 32, use::a, layout::row_major, 1>();
         test_ewops_ab<bfloat16, 16, 64, use::b, layout::ext_intel_packed, 2>();
+        test_ewops_ab<bfloat16, 32, 64, use::b, layout::ext_intel_packed, 2>();
         test_ewops_c<float, 1, 64>();
         test_ewops_c<float, 32, 64>();
 #endif

sycl/test-e2e/Matrix/Inputs/element_wise_ops_impl.hpp

@@ -133,7 +133,9 @@ int main() {
       // IGC
       passed &= test<bfloat16, float, 16, 16, 16, 2, class pvc_bf16_16x16x16>();
       passed &= test<bfloat16, float, 1, 64, 16, 2, class pvc_bf16_1x64x16>();
+      passed &= test<bfloat16, float, 1, 64, 32, 2, class pvc_bf16_1x64x32>();
       passed &= test<bfloat16, float, 32, 64, 16, 2, class pvc_bf16_32x64x16>();
+      passed &= test<bfloat16, float, 32, 64, 32, 2, class pvc_bf16_32x64x32>();
 #endif
       break;
     }

sycl/test-e2e/Matrix/Inputs/joint_matrix_bf16_fill_k_cache_impl.hpp

@@ -483,6 +483,11 @@ int main(
            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*/ 32, MCache2, NCache2, KCache2>(matrix_size);
+      test<bfloat16, float, VnniFactor, /*TM*/ 32, /*TN*/ 64, /*TK*/ 32,
+           MCache1, NCache1, /*KCache1*/ 32, MCache2, NCache2, KCache2>(
+          matrix_size);
 #endif
       break;
     }

sycl/test-e2e/Matrix/Inputs/joint_matrix_rowmajorA_rowmajorB_impl.hpp

@@ -127,8 +127,12 @@ int main() {
                               bfloat16, float>();
         res += gemm_row_major<1, 64, 16, class bf16_1x64x16, bfloat16, bfloat16,
                               float>();
+        res += gemm_row_major<1, 64, 32, class bf16_1x64x32, bfloat16, bfloat16,
+                              float>();
         res += gemm_row_major<32, 64, 16, class bf16_32x64x16, bfloat16,
                               bfloat16, float>();
+        res += gemm_row_major<32, 64, 32, class bf16_32x64x32, bfloat16,
+                              bfloat16, float>();
       }
       break;
     }