| commit | cbede286da8711031fb6fc56bb2e2c246a4c5455 | [log] [tgz] |
|---|---|---|
| author | Sang-Hoon Park <sang-hoon.park@arm.com> | Mon Oct 12 21:44:23 2020 +0100 |
| committer | TeresaARM <teresa.charlinreyes@arm.com> | Tue Oct 13 12:11:51 2020 +0000 |
| tree | a952431d502aa0f5c19ae4008323b5bc6cc2d127 | |
| parent | 141c31a532efb20698466aaabbecc92639f05b0d [diff] [blame] |
COMPMID-3828: fix unsigned overflow in quantized reduce mean Change-Id: I9d3122b4858137d422548d1d417eb04a27ae9c7b Signed-off-by: Sang-Hoon Park <sang-hoon.park@arm.com> Reviewed-on: https://review.mlplatform.org/c/ml/ComputeLibrary/+/4143 Reviewed-by: Michele Di Giorgio <michele.digiorgio@arm.com> Reviewed-by: TeresaARM <teresa.charlinreyes@arm.com> Comments-Addressed: Arm Jenkins <bsgcomp@arm.com> Tested-by: Arm Jenkins <bsgcomp@arm.com>
diff --git a/src/core/NEON/kernels/NEReductionOperationKernel.cpp b/src/core/NEON/kernels/NEReductionOperationKernel.cpp index 9eebb42..9af7f2a 100644 --- a/src/core/NEON/kernels/NEReductionOperationKernel.cpp +++ b/src/core/NEON/kernels/NEReductionOperationKernel.cpp
@@ -1459,7 +1459,7 @@ case ReductionOperation::MEAN_SUM: { int32_t res = static_cast<int32_t>(res_value); - res /= in_info.dimension(axis); + res /= static_cast<int32_t>(in_info.dimension(axis)); *reinterpret_cast<T *>(output.ptr() + x) = utils::cast::saturate_cast<T>(res); break; }