diff options
author | Georgios Pinitas <georgios.pinitas@arm.com> | 2020-07-01 12:35:30 +0100 |
---|---|---|
committer | Georgios Pinitas <georgios.pinitas@arm.com> | 2020-07-02 18:12:47 +0000 |
commit | af56d520b486079e5f773a530582a0a710f7f376 (patch) | |
tree | fa9c1041211da6a4a98170f56f19eb53b3763c40 /src/core/NEON/kernels/arm_gemm/gemm_hybrid_quantized.hpp | |
parent | d355653e3e967667a7d380c9c2ae2016207a6046 (diff) | |
download | ComputeLibrary-af56d520b486079e5f773a530582a0a710f7f376.tar.gz |
COMPMID-3324: Fix per-channel quantization on N blocking
Direct the column to start from in the quantized code
Signed-off-by: Georgios Pinitas <georgios.pinitas@arm.com>
Change-Id: I8231e0b541c6b1b76becf349a1d6ddf973ade9e2
Signed-off-by: Georgios Pinitas <georgios.pinitas@arm.com>
Reviewed-on: https://review.mlplatform.org/c/ml/ComputeLibrary/+/3488
Tested-by: Arm Jenkins <bsgcomp@arm.com>
Reviewed-by: Michele Di Giorgio <michele.digiorgio@arm.com>
Comments-Addressed: Arm Jenkins <bsgcomp@arm.com>
Diffstat (limited to 'src/core/NEON/kernels/arm_gemm/gemm_hybrid_quantized.hpp')
-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 d9b1a71ea8..2b936d0b8f 100644 --- a/src/core/NEON/kernels/arm_gemm/gemm_hybrid_quantized.hpp +++ b/src/core/NEON/kernels/arm_gemm/gemm_hybrid_quantized.hpp @@ -228,7 +228,7 @@ public: requantize_block_32(_qp, (nmax - n0), (m_end - m_start), result_buffer, (nmax - n0), this->_Cptr + (multi * this->_C_multi_stride) + (batch * this->_C_batch_stride) + (m_start * this->_ldc) + n0, this->_ldc, - local_row_sums, col_bias + (multi * _Nsize) + n0); + local_row_sums, col_bias + (multi * _Nsize) + n0, n0); } } while (p.next_dim0()); } |