diff options
| author | Georgios Pinitas <georgios.pinitas@arm.com> | 2021-02-23 00:00:42 +0000 |
|---|---|---|
| committer | Michele Di Giorgio <michele.digiorgio@arm.com> | 2021-02-23 22:29:07 +0000 |
| commit | 261df74754f0feb632f1b604091da927aacb6f7c (patch) | |
| tree | 5c7fe84d270dc2d5536ed344d05c6c38f1016a57 /src/core/CL/gemm/CLGEMMHelpers.cpp | |
| parent | 2ae08600d6266f10a13f0ced3039d94b3c1e8dbe (diff) | |
| download | ComputeLibrary-261df74754f0feb632f1b604091da927aacb6f7c.tar.gz | |
Avoid division with zero on CLGEMM lhs/rhs configuration
Resolves COMPMID-4285
Signed-off-by: Georgios Pinitas <georgios.pinitas@arm.com>
Change-Id: I414e3ba352ed86345d6355d9ae366ea92e641842
Reviewed-on: https://review.mlplatform.org/c/ml/ComputeLibrary/+/5153
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/CL/gemm/CLGEMMHelpers.cpp')
| -rw-r--r-- | src/core/CL/gemm/CLGEMMHelpers.cpp | 3 |
1 files changed, 2 insertions, 1 deletions
diff --git a/src/core/CL/gemm/CLGEMMHelpers.cpp b/src/core/CL/gemm/CLGEMMHelpers.cpp index d60626b158..5abe224a3f 100644 --- a/src/core/CL/gemm/CLGEMMHelpers.cpp +++ b/src/core/CL/gemm/CLGEMMHelpers.cpp @@ -1,5 +1,5 @@ /* - * Copyright (c) 2019-2020 Arm Limited. + * Copyright (c) 2019-2021 Arm Limited. * * SPDX-License-Identifier: MIT * @@ -40,6 +40,7 @@ using namespace arm_compute::misc::shape_calculator; std::pair<GEMMLHSMatrixInfo, GEMMRHSMatrixInfo> configure_lhs_rhs_info(unsigned int m, unsigned int n, unsigned int m0, unsigned int n0, unsigned int k0, unsigned int v0, unsigned int h0, bool lhs_interleave, bool rhs_interleave, bool lhs_transpose, bool rhs_transpose, bool export_to_cl_image) { + ARM_COMPUTE_ERROR_ON(m0 == 0 || n0 == 0); v0 = std::max(std::min(static_cast<int>(m / m0), static_cast<int>(v0)), static_cast<int>(1)); h0 = std::max(std::min(static_cast<int>(n / n0), static_cast<int>(h0)), static_cast<int>(1)); |