From 1d8936db60c91b0dfc6c4c83f545300bea648cee Mon Sep 17 00:00:00 2001 From: Pablo Marquez Tello Date: Wed, 15 May 2024 10:57:14 +0100 Subject: Fix nightly build error * Resolves COMPMID-7059 Change-Id: If77e579199720b7234298d2dc844d88c05989bf9 Signed-off-by: Pablo Marquez Tello Reviewed-on: https://review.mlplatform.org/c/ml/ComputeLibrary/+/11556 Benchmark: Arm Jenkins Tested-by: Arm Jenkins Reviewed-by: Viet-Hoa Do Comments-Addressed: Arm Jenkins --- src/cpu/operators/internal/CpuGemmAssemblyDispatch.cpp | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) (limited to 'src') diff --git a/src/cpu/operators/internal/CpuGemmAssemblyDispatch.cpp b/src/cpu/operators/internal/CpuGemmAssemblyDispatch.cpp index cc21ccbaa1..a4c856bb8f 100644 --- a/src/cpu/operators/internal/CpuGemmAssemblyDispatch.cpp +++ b/src/cpu/operators/internal/CpuGemmAssemblyDispatch.cpp @@ -965,13 +965,13 @@ Status CpuGemmAssemblyDispatch::has_opt_impl(arm_compute::WeightFormat &expected } #endif /* defined(ARM_COMPUTE_ENABLE_BF16) */ -#if defined(ARM_COMPUTE_ENABLE_FP16) +#if defined(ENABLE_FP16_KERNELS) case DataType::F16: ARM_COMPUTE_RETURN_ERROR_ON_MSG( !(arm_gemm::has_opt_gemm(arm_gemm_expected_wf, args, {})), "We could not find an optimized kernel for F16 input and F16 output"); break; -#endif /* ARM_COMPUTE_ENABLE_FP16 */ +#endif /* ENABLE_FP16_KERNELS */ default: ARM_COMPUTE_RETURN_ERROR_ON_MSG(true, "Usupported type. Could not find a kernel"); break; @@ -1104,11 +1104,11 @@ void CpuGemmAssemblyDispatch::configure( } break; #endif /* defined(ARM_COMPUTE_ENABLE_BF16) */ -#ifdef ARM_COMPUTE_ENABLE_FP16 +#ifdef ENABLE_FP16_KERNELS case DataType::F16: create_arm_gemm(_arm_gemm, a, b, c, d, act, info); break; -#endif /* ARM_COMPUTE_ENABLE_FP16 */ +#endif /* ENABLE_FP16_KERNELS */ default: break; } -- cgit v1.2.1