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

intel · Nov 9, 2024 · 5d8f573 · 5d8f573
1 parent e68ebeb
commit 5d8f573
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 0 deletions.
diff --git a/sycl/test-e2e/Matrix/element_wise_all_ops_impl.hpp b/sycl/test-e2e/Matrix/element_wise_all_ops_impl.hpp
@@ -230,7 +230,9 @@ int main() {
 // This combination is not currently supported for sub group size = 32 in IGC
 #if (!defined(SG_SZ) || SG_SZ != 32)
         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

diff --git a/sycl/test-e2e/Matrix/element_wise_ops_impl.hpp b/sycl/test-e2e/Matrix/element_wise_ops_impl.hpp
@@ -134,6 +134,7 @@ int main() {
       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, 32, 64, 16, 2, class pvc_bf16_32x64x16>();
+      passed &= test<bfloat16, float, 32, 64, 32, 2, class pvc_bf16_32x64x32>();
 #endif
       break;
     }

diff --git a/sycl/test-e2e/Matrix/joint_matrix_rowmajorA_rowmajorB_impl.hpp b/sycl/test-e2e/Matrix/joint_matrix_rowmajorA_rowmajorB_impl.hpp
@@ -129,6 +129,8 @@ int main() {
                               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;
     }