diff options
author | Michalis Spyrou <michalis.spyrou@arm.com> | 2019-08-23 15:31:08 +0100 |
---|---|---|
committer | Gian Marco Iodice <gianmarco.iodice@arm.com> | 2019-08-23 17:13:24 +0100 |
commit | 935ae8dd5d573161f1fe4d4a1f9f5fa80bf436fd (patch) | |
tree | ec7470263b5a8940f605eb22df1d0a2d07170368 /src | |
parent | bf193548b3afdee7db154c2fa5625e2194e51de2 (diff) | |
download | ComputeLibrary-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>
Diffstat (limited to 'src')
-rw-r--r-- | src/core/NEON/kernels/arm_gemm/gemm_hybrid_quantized.hpp | 2 |
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); } |