Skip to content
This repository was archived by the owner on Mar 28, 2023. It is now read-only.

Commit 5467e1c

Browse files
authored
[SYCL][Matrix] Fix issues in bf16 test that were causing failed execution
Remove old joint matrix syntax declaration
1 parent 8d98508 commit 5467e1c

File tree

1 file changed

+7
-11
lines changed

1 file changed

+7
-11
lines changed

SYCL/Matrix/joint_matrix_bf16.cpp

Lines changed: 7 additions & 11 deletions
Original file line numberDiff line numberDiff line change
@@ -70,18 +70,14 @@ void matrix_multiply(big_matrix<T1, NUM_ROWS_C, NUM_COLS_C> &C,
7070
const auto sg_startx = global_idx - spmd_item.get_local_id(0);
7171
const auto sg_starty = global_idy - spmd_item.get_local_id(1);
7272

73-
ext::oneapi::sub_group sg = spmd_item.get_sub_group();
74-
joint_matrix<ext::oneapi::sub_group, unsigned short, TM, TK> sub_a(
75-
sg);
73+
sub_group sg = spmd_item.get_sub_group();
74+
joint_matrix<unsigned short, TM, TK> sub_a(sg);
7675
// For B, since current implementation does not support non-packed
77-
// layout, users need to specify the updated VNNI sizes along with
78-
// the packed_b layout. By default, the layout is row_major and size
79-
// is (TK, TN).
80-
joint_matrix<ext::oneapi::sub_group, unsigned short, TK, TN,
81-
matrix_layout::packed_b>
82-
sub_b(sg);
83-
joint_matrix<ext::oneapi::sub_group, float, TM, TN> sub_c(sg);
84-
76+
// layout, users need to specify the packed_b layout.
77+
// By default, the layout is row_major
78+
joint_matrix<unsigned short, TK, TN, matrix_layout::packed_b> sub_b(sg);
79+
joint_matrix<float, TM, TN> sub_c(sg);
80+
8581
joint_matrix_load(sg, sub_c,
8682
accC.get_pointer() + (sg_startx * TM) * N +
8783
sg_starty / SG_SZ * TN,

0 commit comments

Comments
 (0)