diff options
author | Renato Arantes <renato.arantes@arm.com> | 2023-06-15 13:40:02 +0000 |
---|---|---|
committer | ramy.elgammal@arm.com <ramy.elgammal@arm.com> | 2023-06-27 18:28:19 +0100 |
commit | 901a61a13123efe8631ea58b420c7938854fc57e (patch) | |
tree | b17d32de4e197c51a2f66ba7cb14ad5ed861fef3 | |
parent | 3f70cd053573cb6140990ab619ead865f55f3139 (diff) | |
download | ComputeLibrary-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.cpp | 3 |
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); |