diff options
| author | Pablo Marquez Tello <pablo.tello@arm.com> | 2023-10-06 13:49:44 +0100 |
|---|---|---|
| committer | Pablo Marquez Tello <pablo.tello@arm.com> | 2023-10-10 08:08:35 +0000 |
| commit | 6777359cd5b7aa322723d881e9ada426668491fa (patch) | |
| tree | 46ae20dd7d0dd572bacf7d3095808f2868914fe8 /src/cpu/kernels/CpuSubKernel.cpp | |
| parent | 1434155d8d937c5a07af67439f8241064f841ad8 (diff) | |
| download | ComputeLibrary-6777359cd5b7aa322723d881e9ada426668491fa.tar.gz | |
CpuSubKernel changes to enable fp16 in armv8a multi_isa builds
* FP16 kernels must be instantiated in fp16.cpp.
* Partially resolves MLCE-1102
Change-Id: I497fe0ba6e84493a5072c3e80bbba7ecd5de8095
Signed-off-by: Pablo Marquez Tello <pablo.tello@arm.com>
Reviewed-on: https://review.mlplatform.org/c/ml/ComputeLibrary/+/10448
Tested-by: Arm Jenkins <bsgcomp@arm.com>
Reviewed-by: Viet-Hoa Do <viet-hoa.do@arm.com>
Comments-Addressed: Arm Jenkins <bsgcomp@arm.com>
Benchmark: Arm Jenkins <bsgcomp@arm.com>
Diffstat (limited to 'src/cpu/kernels/CpuSubKernel.cpp')
| -rw-r--r-- | src/cpu/kernels/CpuSubKernel.cpp | 5 |
1 files changed, 3 insertions, 2 deletions
diff --git a/src/cpu/kernels/CpuSubKernel.cpp b/src/cpu/kernels/CpuSubKernel.cpp index 2b2c6f2e92..c8706ff651 100644 --- a/src/cpu/kernels/CpuSubKernel.cpp +++ b/src/cpu/kernels/CpuSubKernel.cpp @@ -1,5 +1,5 @@ /* - * Copyright (c) 2021-2022 Arm Limited. + * Copyright (c) 2021-2023 Arm Limited. * * SPDX-License-Identifier: MIT * @@ -31,6 +31,7 @@ #include "src/core/helpers/AutoConfiguration.h" #include "src/core/helpers/WindowHelpers.h" #include "src/cpu/kernels/add/generic/neon/impl.h" +#include "src/cpu/kernels/sub/neon/impl.h" #include "src/cpu/kernels/sub/neon/list.h" #if defined(ENABLE_FP32_KERNELS) @@ -57,7 +58,7 @@ static const std::vector<CpuSubKernel::SubKernel> available_kernels = { REGISTER_FP32_NEON(arm_compute::cpu::sub_same_neon<float>)}, {"neon_fp16_sub", [](const CpuSubKernelDataTypeISASelectorData &data) { return (data.dt == DataType::F16) && data.isa.fp16; }, - REGISTER_FP16_NEON(arm_compute::cpu::sub_same_neon<float16_t>)}, + REGISTER_FP16_NEON(arm_compute::cpu::sub_same_neon_fp16)}, {"neon_u8_sub", [](const CpuSubKernelDataTypeISASelectorData &data) { return (data.dt == DataType::U8); }, REGISTER_INTEGER_NEON(arm_compute::cpu::sub_same_neon<uint8_t>)}, {"neon_s16_sub", [](const CpuSubKernelDataTypeISASelectorData &data) { return (data.dt == DataType::S16); }, |