From ec0a057af6e6c5147f493a9c305f92c9db6ecaf8 Mon Sep 17 00:00:00 2001 From: Gunes Bayir Date: Wed, 13 Dec 2023 09:38:07 +0000 Subject: Revert "Fix nightly bug caused by wrong validation in Gemm mmul kernel" This reverts commit 270576a9fbeeda5210483931388e62f9a1059dd9. Signed-off-by: Gunes Bayir Change-Id: Ia4e965156af46a9afd78819e90fd2a033a97fc2b Reviewed-on: https://review.mlplatform.org/c/ml/ComputeLibrary/+/10888 Reviewed-by: Jakub Sujak Benchmark: Arm Jenkins Tested-by: Arm Jenkins --- .../kernels/ClGemmMatrixMultiplyReshapedOnlyRhsMMULKernel.cpp | 11 ++++++++++- 1 file changed, 10 insertions(+), 1 deletion(-) diff --git a/src/gpu/cl/kernels/ClGemmMatrixMultiplyReshapedOnlyRhsMMULKernel.cpp b/src/gpu/cl/kernels/ClGemmMatrixMultiplyReshapedOnlyRhsMMULKernel.cpp index d67c595449..9a2a4890f3 100644 --- a/src/gpu/cl/kernels/ClGemmMatrixMultiplyReshapedOnlyRhsMMULKernel.cpp +++ b/src/gpu/cl/kernels/ClGemmMatrixMultiplyReshapedOnlyRhsMMULKernel.cpp @@ -99,7 +99,16 @@ Status validate_arguments(const ITensorInfo *src0, ARM_COMPUTE_UNUSED(k); ARM_COMPUTE_RETURN_ERROR_ON(src0->dimension(0) != k); - ARM_COMPUTE_RETURN_ERROR_ON(src0->dimension(1) != m); + + // Validate the reinterpreted-as-3D-case + if (gemm_info.depth_output_gemm3d != 0) + { + ARM_COMPUTE_RETURN_ERROR_ON(src0->dimension(1) * src0->dimension(2) != m); + } + else + { + ARM_COMPUTE_RETURN_ERROR_ON(src0->dimension(1) != m); + } // Validate the gemm-batched case if (src1->num_dimensions() > 2) -- cgit v1.2.1