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>
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);
});