diff options
| author | Pablo Marquez Tello <pablo.tello@arm.com> | 2024-05-15 10:57:14 +0100 |
|---|---|---|
| committer | Pablo Marquez Tello <pablo.tello@arm.com> | 2024-05-15 12:09:09 +0000 |
| commit | 1d8936db60c91b0dfc6c4c83f545300bea648cee (patch) | |
| tree | 1faed2b580e829e67b36146ba4857834e81fb45c | |
| parent | b4b61a649deef5374fa070b79562ae83d7f64f23 (diff) | |
| download | ComputeLibrary-1d8936db60c91b0dfc6c4c83f545300bea648cee.tar.gz | |
Fix nightly build error
* Resolves COMPMID-7059
Change-Id: If77e579199720b7234298d2dc844d88c05989bf9
Signed-off-by: Pablo Marquez Tello <pablo.tello@arm.com>
Reviewed-on: https://review.mlplatform.org/c/ml/ComputeLibrary/+/11556
Benchmark: Arm Jenkins <bsgcomp@arm.com>
Tested-by: Arm Jenkins <bsgcomp@arm.com>
Reviewed-by: Viet-Hoa Do <viet-hoa.do@arm.com>
Comments-Addressed: Arm Jenkins <bsgcomp@arm.com>
| -rw-r--r-- | src/cpu/operators/internal/CpuGemmAssemblyDispatch.cpp | 8 |
1 files changed, 4 insertions, 4 deletions
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<float16_t, float16_t, arm_gemm::Nothing>(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<float16_t, float16_t>(_arm_gemm, a, b, c, d, act, info); break; -#endif /* ARM_COMPUTE_ENABLE_FP16 */ +#endif /* ENABLE_FP16_KERNELS */ default: break; } |