Skip to content

Commit 3c460af

Browse files
author
mmoadeli
committed
Add decoration type for call to get_multi_ptr function.
1 parent 02bec23 commit 3c460af

File tree

3 files changed

+32
-14
lines changed

3 files changed

+32
-14
lines changed

sycl/test-e2e/Matrix/joint_matrix_hip_apply.hpp

Lines changed: 14 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -55,19 +55,27 @@ void hip_matrix_apply() {
5555
joint_matrix<sub_group, InType, use::a, M, K, layout::col_major>
5656
sub_a{};
5757

58-
joint_matrix_load(sg, sub_a, accA.template get_multi_ptr(), K);
59-
joint_matrix_load(sg, sub_b, accB.template get_multi_ptr(), N);
60-
joint_matrix_load(sg, sub_c, accC.template get_multi_ptr(), N,
61-
layout::row_major);
58+
joint_matrix_load(
59+
sg, sub_a,
60+
accA.template get_multi_ptr<access::decorated::yes>(), K);
61+
joint_matrix_load(
62+
sg, sub_b,
63+
accB.template get_multi_ptr<access::decorated::yes>(), N);
64+
joint_matrix_load(
65+
sg, sub_c,
66+
accC.template get_multi_ptr<access::decorated::yes>(), N,
67+
layout::row_major);
6268

6369
joint_matrix_apply(sg, sub_a, [=](InType v) { return v * 2; });
6470
joint_matrix_apply(sg, sub_b, [=](InType v) { return v * 3; });
6571
joint_matrix_apply(sg, sub_c, [=](OutType v) { return v * 4; });
6672

6773
sub_c = joint_matrix_mad(sg, sub_a, sub_b, sub_c);
6874

69-
joint_matrix_store(sg, sub_c, accD.template get_multi_ptr(), N,
70-
layout::row_major);
75+
joint_matrix_store(
76+
sg, sub_c,
77+
accD.template get_multi_ptr<access::decorated::yes>(), N,
78+
layout::row_major);
7179
});
7280
})
7381
.wait();

sycl/test-e2e/Matrix/joint_matrix_hip_fill.hpp

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -61,8 +61,10 @@ void hip_matrix_fill() {
6161

6262
sub_c = joint_matrix_mad(sg, sub_a, sub_b, sub_c);
6363

64-
joint_matrix_store(sg, sub_c, accD.template get_multi_ptr(), N,
65-
layout::row_major);
64+
joint_matrix_store(
65+
sg, sub_c,
66+
accD.template get_multi_ptr<access::decorated::yes>(), N,
67+
layout::row_major);
6668
});
6769
})
6870
.wait();

sycl/test-e2e/Matrix/joint_matrix_hip_mfma.hpp

Lines changed: 14 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -71,15 +71,23 @@ void hip_matrix_mfma() {
7171
joint_matrix<sub_group, InType, use::a, M, K, layout::col_major>
7272
sub_a{};
7373

74-
joint_matrix_load(sg, sub_a, accA.template get_multi_ptr(), K);
75-
joint_matrix_load(sg, sub_b, accB.template get_multi_ptr(), N);
76-
joint_matrix_load(sg, sub_c, accC.template get_multi_ptr(), N,
77-
layout::row_major);
74+
joint_matrix_load(
75+
sg, sub_a,
76+
accA.template get_multi_ptr<access::decorated::yes>(), K);
77+
joint_matrix_load(
78+
sg, sub_b,
79+
accB.template get_multi_ptr<access::decorated::yes>(), N);
80+
joint_matrix_load(
81+
sg, sub_c,
82+
accC.template get_multi_ptr<access::decorated::yes>(), N,
83+
layout::row_major);
7884

7985
sub_c = joint_matrix_mad(sg, sub_a, sub_b, sub_c);
8086

81-
joint_matrix_store(sg, sub_c, accD.template get_multi_ptr(), N,
82-
OutLayout);
87+
joint_matrix_store(
88+
sg, sub_c,
89+
accD.template get_multi_ptr<access::decorated::yes>(), N,
90+
OutLayout);
8391
});
8492
})
8593
.wait();

0 commit comments

Comments
 (0)