aboutsummaryrefslogtreecommitdiff
diff options
context:
space:
mode:
authorMichalis Spyrou <michalis.spyrou@arm.com>2019-08-23 15:31:08 +0100
committerGian Marco Iodice <gianmarco.iodice@arm.com>2019-08-23 17:13:24 +0100
commit935ae8dd5d573161f1fe4d4a1f9f5fa80bf436fd (patch)
treeec7470263b5a8940f605eb22df1d0a2d07170368
parentbf193548b3afdee7db154c2fa5625e2194e51de2 (diff)
downloadComputeLibrary-935ae8dd5d573161f1fe4d4a1f9f5fa80bf436fd.tar.gz
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>
-rw-r--r--src/core/NEON/kernels/arm_gemm/gemm_hybrid_quantized.hpp2
1 files changed, 1 insertions, 1 deletions
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);
}