aboutsummaryrefslogtreecommitdiff
diff options
context:
space:
mode:
authorRenato Arantes <renato.arantes@arm.com>2023-06-15 13:40:02 +0000
committerramy.elgammal@arm.com <ramy.elgammal@arm.com>2023-06-27 18:28:19 +0100
commit901a61a13123efe8631ea58b420c7938854fc57e (patch)
treeb17d32de4e197c51a2f66ba7cb14ad5ed861fef3
parent3f70cd053573cb6140990ab619ead865f55f3139 (diff)
downloadComputeLibrary-901a61a13123efe8631ea58b420c7938854fc57e.tar.gz
Address the issues with the ACL coverage pipeline failures related to matmul.
Signed-off-by: Renato Arantes <renato.arantes@arm.com> Change-Id: I98de659d1289c930e366727d4799f0dacc8121ab Reviewed-on: https://review.mlplatform.org/c/ml/ComputeLibrary/+/9782 Benchmark: Arm Jenkins <bsgcomp@arm.com> Tested-by: Arm Jenkins <bsgcomp@arm.com> Reviewed-by: SiCong Li <sicong.li@arm.com> Reviewed-by: Jakub Sujak <jakub.sujak@arm.com> Comments-Addressed: Arm Jenkins <bsgcomp@arm.com>
-rw-r--r--src/cpu/operators/CpuGemm.cpp3
1 files changed, 2 insertions, 1 deletions
diff --git a/src/cpu/operators/CpuGemm.cpp b/src/cpu/operators/CpuGemm.cpp
index 7411d763b4..34b845928d 100644
--- a/src/cpu/operators/CpuGemm.cpp
+++ b/src/cpu/operators/CpuGemm.cpp
@@ -173,7 +173,8 @@ Status CpuGemm::validate(const ITensorInfo *a, const ITensorInfo *b, const ITens
}
const int block_by = arm_compute::block_by(gemm_info.weight_format());
- if(block_by > 1)
+ // test if im2col has changed the dimensions that are needed for padding
+ if(a->dimension(0) != b->dimension(1) && block_by > 1)
{
// have to verify bias
const size_t dim0_sz = a->dimension(0);