Skip to content

Commit 0eac618

Browse files
authored
[SYCL][Matrix] Correct Prefetch instruction usage (#12623)
Signed-off-by: Sidorov, Dmitry <[email protected]>
1 parent 96073b9 commit 0eac618

File tree

2 files changed

+8
-6
lines changed

2 files changed

+8
-6
lines changed

sycl/include/CL/__spirv/spirv_ops.hpp

+5-4
Original file line numberDiff line numberDiff line change
@@ -174,10 +174,11 @@ extern __DPCPP_SYCL_EXTERNAL __spv::__spirv_JointMatrixINTEL<T, R, C, L, S, U> *
174174
__spirv_VectorInsertDynamic(__spv::__spirv_JointMatrixINTEL<T, R, C, L, S, U> *,
175175
Ts val, size_t i);
176176

177-
template <typename T, std::size_t NumRows, std::size_t NumCols>
178-
extern __DPCPP_SYCL_EXTERNAL void __spirv_JointMatrixPrefetchINTEL(
179-
T *Ptr, std::size_t coordX, std::size_t coordY, unsigned int CacheLevel,
180-
__spv::MatrixLayout Layout, std::size_t Stride);
177+
template <typename T>
178+
extern __DPCPP_SYCL_EXTERNAL void __spirv_CooperativeMatrixPrefetchINTEL(
179+
T *Ptr, std::size_t coordX, std::size_t coordY, std::size_t NumRows,
180+
std::size_t NumCols, unsigned int CacheLevel, __spv::MatrixLayout Layout,
181+
std::size_t Stride);
181182

182183
#ifndef __SPIRV_BUILTIN_DECLARATIONS__
183184
#error \

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

+3-2
Original file line numberDiff line numberDiff line change
@@ -524,8 +524,9 @@ joint_matrix_prefetch(Group sg, T *Ptr, size_t stride,
524524
// Will be removed once SPIRV implementation also uses offsetpointer
525525
size_t coordX = 0;
526526
size_t coordY = 0;
527-
__spirv_JointMatrixPrefetchINTEL<T, NumRows, NumCols>(
528-
Ptr, coordX, coordY, detail::PropertyMetaInfo<decltype(prop)>::value,
527+
__spirv_CooperativeMatrixPrefetchINTEL<T>(
528+
Ptr, coordX, coordY, NumRows, NumCols,
529+
detail::PropertyMetaInfo<decltype(prop)>::value,
529530
sycl::detail::joint_matrix_layout_to_spv(Layout), stride);
530531
#endif // defined(__NVPTX__)
531532
#else

0 commit comments

Comments
 (0)