diff --git a/sycl/include/sycl/__spirv/spirv_ops.hpp b/sycl/include/sycl/__spirv/spirv_ops.hpp index 4f68506aaca03..7eacbeb6a529f 100644 --- a/sycl/include/sycl/__spirv/spirv_ops.hpp +++ b/sycl/include/sycl/__spirv/spirv_ops.hpp @@ -107,12 +107,11 @@ extern __DPCPP_SYCL_EXTERNAL __spv::__spirv_CooperativeMatrixKHR * __spirv_CompositeConstruct(const T v); -// TODO: replace with __spirv_CooperativeMatrixGetElementCoordINTEL when ready template extern __DPCPP_SYCL_EXTERNAL __ocl_vec_t -__spirv_JointMatrixGetElementCoordINTEL( +__spirv_CooperativeMatrixGetElementCoordINTEL( __spv::__spirv_CooperativeMatrixKHR *, size_t i); // AccessChain followed by load/store serves to extract/insert and element diff --git a/sycl/include/sycl/ext/oneapi/matrix/matrix-intel.hpp b/sycl/include/sycl/ext/oneapi/matrix/matrix-intel.hpp index 7a0fa9c0e6a79..48a3947ea8d10 100644 --- a/sycl/include/sycl/ext/oneapi/matrix/matrix-intel.hpp +++ b/sycl/include/sycl/ext/oneapi/matrix/matrix-intel.hpp @@ -116,7 +116,7 @@ class wi_element { inline __SYCL_ALWAYS_INLINE std::tuple get_coord() { #if defined(__SYCL_DEVICE_ONLY__) __ocl_vec_t coord = - __spirv_JointMatrixGetElementCoordINTEL(M.spvm, idx); + __spirv_CooperativeMatrixGetElementCoordINTEL(M.spvm, idx); const size_t row = coord[0]; const size_t col = coord[1]; return std::make_tuple(row, col); @@ -239,7 +239,7 @@ class wi_element get_coord() { #if defined(__SYCL_DEVICE_ONLY__) __ocl_vec_t coord = - __spirv_JointMatrixGetElementCoordINTEL(M.spvm, idx); + __spirv_CooperativeMatrixGetElementCoordINTEL(M.spvm, idx); const uint32_t row = coord[0]; const uint32_t col = coord[1]; return std::make_tuple(row, col);