Skip to content

Commit 5d8f573

Browse files
committed
[SYCL][E2E][Joint Matrix] Add k=32 for bfloat16 tests
1 parent e68ebeb commit 5d8f573

File tree

3 files changed

+5
-0
lines changed

3 files changed

+5
-0
lines changed

sycl/test-e2e/Matrix/element_wise_all_ops_impl.hpp

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -230,7 +230,9 @@ int main() {
230230
// This combination is not currently supported for sub group size = 32 in IGC
231231
#if (!defined(SG_SZ) || SG_SZ != 32)
232232
test_ewops_ab<bfloat16, 32, 16, use::a, layout::row_major, 1>();
233+
test_ewops_ab<bfloat16, 32, 32, use::a, layout::row_major, 1>();
233234
test_ewops_ab<bfloat16, 16, 64, use::b, layout::ext_intel_packed, 2>();
235+
test_ewops_ab<bfloat16, 32, 64, use::b, layout::ext_intel_packed, 2>();
234236
test_ewops_c<float, 1, 64>();
235237
test_ewops_c<float, 32, 64>();
236238
#endif

sycl/test-e2e/Matrix/element_wise_ops_impl.hpp

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -134,6 +134,7 @@ int main() {
134134
passed &= test<bfloat16, float, 16, 16, 16, 2, class pvc_bf16_16x16x16>();
135135
passed &= test<bfloat16, float, 1, 64, 16, 2, class pvc_bf16_1x64x16>();
136136
passed &= test<bfloat16, float, 32, 64, 16, 2, class pvc_bf16_32x64x16>();
137+
passed &= test<bfloat16, float, 32, 64, 32, 2, class pvc_bf16_32x64x32>();
137138
#endif
138139
break;
139140
}

sycl/test-e2e/Matrix/joint_matrix_rowmajorA_rowmajorB_impl.hpp

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -129,6 +129,8 @@ int main() {
129129
float>();
130130
res += gemm_row_major<32, 64, 16, class bf16_32x64x16, bfloat16,
131131
bfloat16, float>();
132+
res += gemm_row_major<32, 64, 32, class bf16_32x64x32, bfloat16,
133+
bfloat16, float>();
132134
}
133135
break;
134136
}

0 commit comments

Comments
 (0)