From af56d520b486079e5f773a530582a0a710f7f376 Mon Sep 17 00:00:00 2001 From: Georgios Pinitas Date: Wed, 1 Jul 2020 12:35:30 +0100 Subject: COMPMID-3324: Fix per-channel quantization on N blocking Direct the column to start from in the quantized code Signed-off-by: Georgios Pinitas Change-Id: I8231e0b541c6b1b76becf349a1d6ddf973ade9e2 Signed-off-by: Georgios Pinitas Reviewed-on: https://review.mlplatform.org/c/ml/ComputeLibrary/+/3488 Tested-by: Arm Jenkins Reviewed-by: Michele Di Giorgio Comments-Addressed: Arm Jenkins --- src/core/NEON/kernels/arm_gemm/quantize_wrapper.hpp | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) (limited to 'src/core/NEON/kernels/arm_gemm/quantize_wrapper.hpp') diff --git a/src/core/NEON/kernels/arm_gemm/quantize_wrapper.hpp b/src/core/NEON/kernels/arm_gemm/quantize_wrapper.hpp index 18f030fec0..995716575a 100644 --- a/src/core/NEON/kernels/arm_gemm/quantize_wrapper.hpp +++ b/src/core/NEON/kernels/arm_gemm/quantize_wrapper.hpp @@ -115,7 +115,7 @@ private: _args._Nsize, this->_Cptr + (multi * this->_C_multi_stride) + (batch * this->_C_batch_stride) + (first_row * this->_ldc), this->_ldc, _row_sums + (multi * _args._nbatches * _args._Msize) + (batch * _args._Msize) + first_row, - _col_sums + (multi * _args._Nsize)); + _col_sums + (multi * _args._Nsize), 0); } } } -- cgit v1.2.1