From 7f8caf770417574a874d53733718f1b7052456fe Mon Sep 17 00:00:00 2001 From: Michalis Spyrou Date: Thu, 13 May 2021 13:35:30 +0100 Subject: Fix integer overflow and null dereference Resolves: COMPMID-4527 Change-Id: If038d2477d8898d3ee307fe58301fb0b16b64c02 Signed-off-by: Michalis Spyrou Reviewed-on: https://review.mlplatform.org/c/ml/ComputeLibrary/+/5640 Reviewed-by: Michele Di Giorgio Comments-Addressed: Arm Jenkins Tested-by: Arm Jenkins --- src/core/cpu/kernels/CpuMulKernel.cpp | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) (limited to 'src/core/cpu/kernels/CpuMulKernel.cpp') diff --git a/src/core/cpu/kernels/CpuMulKernel.cpp b/src/core/cpu/kernels/CpuMulKernel.cpp index dabf656e6e..82ec322875 100644 --- a/src/core/cpu/kernels/CpuMulKernel.cpp +++ b/src/core/cpu/kernels/CpuMulKernel.cpp @@ -846,7 +846,7 @@ void mul_S32_S32_S32(const ITensor *src1, const ITensor *src2, ITensor *out, con } else { - uint64_t mask = (1u << n) - 1; + uint64_t mask = ((uint64_t)1u << n) - 1; tmp = (tmp + static_cast(mask)) >> n; } if(is_sat) @@ -909,7 +909,7 @@ void mul_S32_S32_S32(const ITensor *src1, const ITensor *src2, ITensor *out, con } else { - uint64_t mask = (1u << n) - 1; + uint64_t mask = ((uint64_t)1u << n) - 1; tmp = (tmp + static_cast(mask)) >> n; } if(is_sat) -- cgit v1.2.1