diff options
| author | Mohammed Suhail Munshi <MohammedSuhail.Munshi@arm.com> | 2023-02-03 11:30:52 +0000 |
|---|---|---|
| committer | Jakub Sujak <jakub.sujak@arm.com> | 2023-02-03 14:23:56 +0000 |
| commit | 797ba3f7cdee6acf961747f8f42bc35748d276c2 (patch) | |
| tree | 4e5cd633b601853538e0ed4f3c13f8717592dbe6 | |
| parent | e5dafd39b726a55cd42827222a89513746254508 (diff) | |
| download | ComputeLibrary-797ba3f7cdee6acf961747f8f42bc35748d276c2.tar.gz | |
Fix armv7a failing GEMMConvolutionLayer tests
Resolves: [COMPMID-5854]
Signed-off-by: Mohammed Suhail Munshi <MohammedSuhail.Munshi@arm.com>
Change-Id: Ib0228409be5e816acca7e123f2660eb01a79e38f
Reviewed-on: https://review.mlplatform.org/c/ml/ComputeLibrary/+/9078
Benchmark: Arm Jenkins <bsgcomp@arm.com>
Tested-by: Arm Jenkins <bsgcomp@arm.com>
Reviewed-by: Gunes Bayir <gunes.bayir@arm.com>
Comments-Addressed: Arm Jenkins <bsgcomp@arm.com>
| -rw-r--r-- | src/cpu/kernels/CpuGemmLowpOffsetContributionOutputStageKernel.cpp | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/src/cpu/kernels/CpuGemmLowpOffsetContributionOutputStageKernel.cpp b/src/cpu/kernels/CpuGemmLowpOffsetContributionOutputStageKernel.cpp index c69af55f57..190487eced 100644 --- a/src/cpu/kernels/CpuGemmLowpOffsetContributionOutputStageKernel.cpp +++ b/src/cpu/kernels/CpuGemmLowpOffsetContributionOutputStageKernel.cpp @@ -839,7 +839,7 @@ Status validate_arguments(const ITensorInfo *mm_result, const ITensorInfo *vecto } // Check Tensor Rank of vector_sum_row - ARM_COMPUTE_RETURN_ERROR_ON(vector_sum_row->num_dimensions() > 2); + ARM_COMPUTE_RETURN_ERROR_ON(vector_sum_row->num_dimensions() > 3); } if(output->total_size() != 0) |