diff options
| author | Georgios Pinitas <georgios.pinitas@arm.com> | 2018-07-27 12:20:32 +0100 |
|---|---|---|
| committer | Anthony Barbier <anthony.barbier@arm.com> | 2018-11-02 16:54:54 +0000 |
| commit | a4658ae6cac4694fe28df5837bc4f4c154ab7204 (patch) | |
| tree | 37cfc46f43c2953a71b4d4073ca6643b2356d999 /src/core/NEON/kernels/assembly | |
| parent | ead90b579a6d93af53e4e6e104c873b9dcc7ee25 (diff) | |
| download | ComputeLibrary-a4658ae6cac4694fe28df5837bc4f4c154ab7204.tar.gz | |
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>
Diffstat (limited to 'src/core/NEON/kernels/assembly')
| -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); }); |