commit a4658ae6cac4694fe28df5837bc4f4c154ab7204
author Georgios Pinitas <georgios.pinitas@arm.com> Fri Jul 27 12:20:32 2018 +0100
committer Anthony Barbier <anthony.barbier@arm.com> Fri Nov 02 16:54:54 2018 +0000
tree 37cfc46f43c2953a71b4d4073ca6643b2356d999
parent ead90b579a6d93af53e4e6e104c873b9dcc7ee25

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 <anthony.barbier@arm.com>
Tested-by: Jenkins <bsgcomp@arm.com>

src/core/NEON/kernels/assembly/NEGEMMNativeWrapperKernel.cpp

diff --git a/src/core/NEON/kernels/assembly/NEGEMMNativeWrapperKernel.cpp b/src/core/NEON/kernels/assembly/NEGEMMNativeWrapperKernel.cpp
index 38eb40a..ea6a06c 100644
--- a/src/core/NEON/kernels/assembly/NEGEMMNativeWrapperKernel.cpp
+++ b/src/core/NEON/kernels/assembly/NEGEMMNativeWrapperKernel.cpp
@@ -98,8 +98,9 @@
         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);
     });