Skip to content

Commit ab52704

Browse files
committed
Add packedB layout to default bfloat16 test
1 parent ed81d7f commit ab52704

File tree

1 file changed

+6
-0
lines changed

1 file changed

+6
-0
lines changed

sycl/test-e2e/Matrix/joint_matrix_bfloat16.cpp

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -30,18 +30,24 @@ int main() {
3030
if (combinations[i].nsize == 0) { // Intel AMX
3131
test<bfloat16, float, float, /*TM*/ 16, /*TN*/ 16, /*TK*/ 16,
3232
layout::row_major, 1>();
33+
test<bfloat16, float, float, /*TM*/ 16, /*TN*/ 16, /*TK*/ 16,
34+
layout::ext_intel_packed, 2>();
3335
break;
3436
}
3537

3638
if (combinations[i].nsize == 16) { // architecture::intel_gpu_pvc
3739
test<bfloat16, float, float, /*TM*/ 8, /*TN*/ 16, /*TK*/ 16,
3840
layout::row_major, 1>();
41+
test<bfloat16, float, float, /*TM*/ 8, /*TN*/ 16, /*TK*/ 16,
42+
layout::ext_intel_packed, 2>();
3943
break;
4044
}
4145

4246
if (combinations[i].nsize == 8) { // architecture::intel_gpu_dg2*
4347
test<bfloat16, float, float, /*TM*/ 8, /*TN*/ 8, /*TK*/ 16,
4448
layout::row_major, 1>();
49+
test<bfloat16, float, float, /*TM*/ 8, /*TN*/ 8, /*TK*/ 16,
50+
layout::ext_intel_packed, 2>();
4551
break;
4652
}
4753
}

0 commit comments

Comments
 (0)