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 16:12:39 +0000 |
| commit | 3e183d98b28ff7c91bbb7166b6f4684cd0e6bd2e (patch) | |
| tree | ac5d36dc5f78e4ffe668021a73decba32737607a /src | |
| parent | 35ea9a7521b4b858a6bcc33a244b6e7f94b6e3a3 (diff) | |
| download | ComputeLibrary-3e183d98b28ff7c91bbb7166b6f4684cd0e6bd2e.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); } |