diff options
-rw-r--r-- | src/core/NEON/kernels/assembly/NEGEMMNativeWrapperKernel.cpp | 3 |
1 files changed, 2 insertions, 1 deletions
diff --git a/src/core/NEON/kernels/assembly/NEGEMMNativeWrapperKernel.cpp b/src/core/NEON/kernels/assembly/NEGEMMNativeWrapperKernel.cpp index 38eb40a06e..ea6a06cada 100644 --- a/src/core/NEON/kernels/assembly/NEGEMMNativeWrapperKernel.cpp +++ b/src/core/NEON/kernels/assembly/NEGEMMNativeWrapperKernel.cpp @@ -98,8 +98,9 @@ void NEGEMMNativeWrapperKernel<To, Tr>::run_internal(const Window &window, const const unsigned int multi = id.z(); const unsigned int ymax = std::min(y0 + strategy::out_height(), m_end); + // TODO(COMPMID-1424) : Agree on gemm IO layouts strat.kernel(a(0, y0, batch, multi), a.stride(Window::DimY), - b(0, 0, 0, multi), b.stride(Window::DimY), + b(0, 0, multi), b.stride(Window::DimY), c(0, y0, batch, multi), c.stride(Window::DimY), _beta, (ymax - y0), _params.N, _params.K); }); |