From a4658ae6cac4694fe28df5837bc4f4c154ab7204 Mon Sep 17 00:00:00 2001 From: Georgios Pinitas Date: Fri, 27 Jul 2018 12:20:32 +0100 Subject: COMPMID-1429: NEGEMMNativeWrapperKernel fails when ran for Winograd Change-Id: Ief9b717fe2bcf626660109ec491f8882d0ef06d7 Reviewed-on: https://eu-gerrit-1.euhpc.arm.com/141658 Reviewed-by: Anthony Barbier Tested-by: Jenkins --- src/core/NEON/kernels/assembly/NEGEMMNativeWrapperKernel.cpp | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) (limited to 'src/core/NEON/kernels') 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::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); }); -- cgit v1.2.1