diff options
author | Adnan AlSinan <adnan.alsinan@arm.com> | 2022-09-21 13:20:45 +0100 |
---|---|---|
committer | Adnan AlSinan <adnan.alsinan@arm.com> | 2022-10-03 15:06:16 +0000 |
commit | 304dfdba67958f5987d88ad0ce538399c3e50bc8 (patch) | |
tree | ce3d3146c6be54a04e8096452b051100943fb664 /src/cpu/kernels/gemm_matrix_mul/generic/neon | |
parent | cf07d118ad23f495a63546f65b0581bc8846156c (diff) | |
download | ComputeLibrary-304dfdba67958f5987d88ad0ce538399c3e50bc8.tar.gz |
Fix Batch Matmul nightly failure
Resolves COMPMID-5601
Signed-off-by: Adnan AlSinan <adnan.alsinan@arm.com>
Change-Id: I1baf92c4751d784d017c0b2f7de1fc09e42ce69c
Reviewed-on: https://review.mlplatform.org/c/ml/ComputeLibrary/+/8309
Benchmark: Arm Jenkins <bsgcomp@arm.com>
Tested-by: Arm Jenkins <bsgcomp@arm.com>
Reviewed-by: Gian Marco Iodice <gianmarco.iodice@arm.com>
Comments-Addressed: Arm Jenkins <bsgcomp@arm.com>
Diffstat (limited to 'src/cpu/kernels/gemm_matrix_mul/generic/neon')
-rw-r--r-- | src/cpu/kernels/gemm_matrix_mul/generic/neon/impl.cpp | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/src/cpu/kernels/gemm_matrix_mul/generic/neon/impl.cpp b/src/cpu/kernels/gemm_matrix_mul/generic/neon/impl.cpp index 37c296fe5e..300dc3ffc7 100644 --- a/src/cpu/kernels/gemm_matrix_mul/generic/neon/impl.cpp +++ b/src/cpu/kernels/gemm_matrix_mul/generic/neon/impl.cpp @@ -856,7 +856,7 @@ void matrix_matrix_multiply_f16(const ITensor *lhs, const ITensor *rhs, ITensor } // Set step_x and step_y for matrix B. Scale by a factor of 8 the X range as the input transposed matrix A has 8 times less the cols of the dst matrix win_b.set(Window::DimX, Window::Dimension(window.x().start() / 8, window.x().end() / 8, in_b_stride)); - win_b.set(Window::DimY, Window::Dimension(0, 1, 0)); + win_b.set(Window::DimY, Window::Dimension(0, 0, 0)); Iterator ina(lhs, win_a); Iterator inb(rhs, win_b); |