Skip to content

Commit e46997b

Browse files
committed
wi_data constructor made private.
Signed-off-by: JackAKirk <[email protected]>
1 parent a5ebf2a commit e46997b

File tree

1 file changed

+10
-7
lines changed

1 file changed

+10
-7
lines changed

sycl/include/sycl/ext/oneapi/matrix/matrix-tensorcore.hpp

Lines changed: 10 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -17,11 +17,20 @@ enum class matrix_use { a, b, accumulator };
1717

1818
enum class matrix_layout { row_major, col_major, packed_a, packed_b };
1919

20+
template <typename T, matrix_use Use, size_t Rows = sycl::dynamic_extent,
21+
size_t Cols = sycl::dynamic_extent,
22+
matrix_layout Layout = matrix_layout::row_major,
23+
typename Group = sycl::sub_group, typename Cond = void>
24+
struct joint_matrix;
25+
2026
template <typename type, size_t size> class wi_data {
2127
marray<type, size> &data;
28+
wi_data(marray<type, size> &wi_data) : data(wi_data){};
29+
template <typename T, matrix_use Use, size_t Rows, size_t Cols,
30+
matrix_layout Layout, typename Group, typename Cond>
31+
friend struct joint_matrix;
2232

2333
public:
24-
wi_data(marray<type, size> &wi_data) : data(wi_data){};
2534
size_t length() {
2635
#if defined(__SYCL_DEVICE_ONLY__) && defined(__NVPTX__)
2736
return data.size();
@@ -41,12 +50,6 @@ template <typename type, size_t size> class wi_data {
4150
};
4251
};
4352

44-
template <typename T, matrix_use Use, size_t Rows = sycl::dynamic_extent,
45-
size_t Cols = sycl::dynamic_extent,
46-
matrix_layout Layout = matrix_layout::row_major,
47-
typename Group = sycl::sub_group, typename Cond = void>
48-
struct joint_matrix;
49-
5053
#define __SYCL_JOINT_MATRIX_OVERLOAD_ARR(type, use, M, N, size) \
5154
template <matrix_layout Layout> \
5255
struct joint_matrix< \

0 commit comments

Comments
 (0)