aboutsummaryrefslogtreecommitdiff
path: root/src/core/cpu/kernels/CpuMulKernel.cpp
diff options
context:
space:
mode:
authorMichalis Spyrou <michalis.spyrou@arm.com>2021-05-13 13:35:30 +0100
committerMichalis Spyrou <michalis.spyrou@arm.com>2021-05-13 15:00:24 +0000
commit7f8caf770417574a874d53733718f1b7052456fe (patch)
tree55dd025efb62a0b8692af7f40dea81870a5b7f99 /src/core/cpu/kernels/CpuMulKernel.cpp
parente57e11d66b8f1e49eff180bc65f877bf88c4c4cf (diff)
downloadComputeLibrary-7f8caf770417574a874d53733718f1b7052456fe.tar.gz
Fix integer overflow and null dereference
Resolves: COMPMID-4527 Change-Id: If038d2477d8898d3ee307fe58301fb0b16b64c02 Signed-off-by: Michalis Spyrou <michalis.spyrou@arm.com> Reviewed-on: https://review.mlplatform.org/c/ml/ComputeLibrary/+/5640 Reviewed-by: Michele Di Giorgio <michele.digiorgio@arm.com> Comments-Addressed: Arm Jenkins <bsgcomp@arm.com> Tested-by: Arm Jenkins <bsgcomp@arm.com>
Diffstat (limited to 'src/core/cpu/kernels/CpuMulKernel.cpp')
-rw-r--r--src/core/cpu/kernels/CpuMulKernel.cpp4
1 files changed, 2 insertions, 2 deletions
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<int64_t>(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<int64_t>(mask)) >> n;
}
if(is_sat)