From 935ae8dd5d573161f1fe4d4a1f9f5fa80bf436fd Mon Sep 17 00:00:00 2001
From: Michalis Spyrou <michalis.spyrou@arm.com>
Date: Fri, 23 Aug 2019 15:31:08 +0100
Subject: COMPMID-2336: Update GEMM code

Fix incorrect stride

Change-Id: I5bb9660bd73c40d587fa869d2488b7ab5f5b1ea7
Signed-off-by: Michalis Spyrou <michalis.spyrou@arm.com>
Reviewed-on: https://review.mlplatform.org/c/1807
Tested-by: Arm Jenkins <bsgcomp@arm.com>
Reviewed-by: Gian Marco Iodice <gianmarco.iodice@arm.com>
---
 src/core/NEON/kernels/arm_gemm/gemm_hybrid_quantized.hpp | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/src/core/NEON/kernels/arm_gemm/gemm_hybrid_quantized.hpp b/src/core/NEON/kernels/arm_gemm/gemm_hybrid_quantized.hpp
index ab419e2216..1bd9126da9 100644
--- a/src/core/NEON/kernels/arm_gemm/gemm_hybrid_quantized.hpp
+++ b/src/core/NEON/kernels/arm_gemm/gemm_hybrid_quantized.hpp
@@ -209,7 +209,7 @@ public:
 #endif
                     strat.kernel(this->_Aptr + (multi * this->_A_multi_stride) + (batch * this->_A_batch_stride) + (m_start * this->_lda) + k0, this->_lda,
                                  b_panel,
-                                 result_buffer, this->_Nsize,
+                                 result_buffer, (nmax-n0),
                                  (k0 == 0) ? _beta : static_cast<Tr>(1),
                                  (m_end - m_start), (nmax - n0), kern_k);
                 }
-- 
cgit v1.2.1