aboutsummaryrefslogtreecommitdiff
path: root/src/cpu/kernels/CpuScaleKernel.cpp
diff options
context:
space:
mode:
authorDana Zlotnik <dana.zlotnik@arm.com>2022-02-21 13:12:41 +0200
committerDana Zlotnik <dana.zlotnik@arm.com>2022-03-01 11:19:23 +0000
commita538ae583c8816f69d05b98c62a9d3092f88f798 (patch)
tree33ffb4611e9d660223b50d805ac60babbaba4ebb /src/cpu/kernels/CpuScaleKernel.cpp
parentee9050089e391e598cd58e05bc7a07597a6d1db0 (diff)
downloadComputeLibrary-a538ae583c8816f69d05b98c62a9d3092f88f798.tar.gz
Multi ISA Technical Debt
* Update json struct meet multi-ISA updates * Add impl.cpp in kernels where we only have impl.h Resolves COMPMID-5173 Change-Id: I5da3c4b016a5d0115c4ba46cbfefde7bce518ac1 Signed-off-by: Dana Zlotnik <dana.zlotnik@arm.com> Reviewed-on: https://review.mlplatform.org/c/ml/ComputeLibrary/+/7191 Tested-by: Arm Jenkins <bsgcomp@arm.com> Reviewed-by: Giorgio Arena <giorgio.arena@arm.com> Comments-Addressed: Arm Jenkins <bsgcomp@arm.com>
Diffstat (limited to 'src/cpu/kernels/CpuScaleKernel.cpp')
-rw-r--r--src/cpu/kernels/CpuScaleKernel.cpp8
1 files changed, 1 insertions, 7 deletions
diff --git a/src/cpu/kernels/CpuScaleKernel.cpp b/src/cpu/kernels/CpuScaleKernel.cpp
index 60564a97dd..e230dfa938 100644
--- a/src/cpu/kernels/CpuScaleKernel.cpp
+++ b/src/cpu/kernels/CpuScaleKernel.cpp
@@ -50,10 +50,9 @@ namespace
{
static const std::vector<CpuScaleKernel::ScaleKernel> available_kernels =
{
-#if defined(ARM_COMPUTE_ENABLE_SVE)
{
"sve_fp16_scale",
- [](const DataTypeISASelectorData & data) { return data.dt == DataType::F16 && data.isa.sve; },
+ [](const DataTypeISASelectorData & data) { return data.dt == DataType::F16 && data.isa.sve && data.isa.fp16; },
REGISTER_FP16_SVE(arm_compute::cpu::fp16_sve_scale)
},
{
@@ -81,15 +80,11 @@ static const std::vector<CpuScaleKernel::ScaleKernel> available_kernels =
[](const DataTypeISASelectorData & data) { return data.dt == DataType::S16 && data.isa.sve; },
REGISTER_INTEGER_SVE(arm_compute::cpu::s16_sve_scale)
},
-#endif /* defined(ARM_COMPUTE_ENABLE_SVE) */
-#if defined(ARM_COMPUTE_ENABLE_NEON)
-#if defined(__ARM_FEATURE_FP16_VECTOR_ARITHMETIC)
{
"neon_fp16_scale",
[](const DataTypeISASelectorData & data) { return data.dt == DataType::F16 && data.isa.fp16; },
REGISTER_FP16_NEON(arm_compute::cpu::common_neon_scale<float16_t>)
},
-#endif /* !defined(__ARM_FEATURE_FP16_VECTOR_ARITHMETIC) */
{
"neon_fp32_scale",
[](const DataTypeISASelectorData & data) { return data.dt == DataType::F32; },
@@ -115,7 +110,6 @@ static const std::vector<CpuScaleKernel::ScaleKernel> available_kernels =
[](const DataTypeISASelectorData & data) { return data.dt == DataType::S16; },
REGISTER_INTEGER_NEON(arm_compute::cpu::s16_neon_scale)
},
-#endif /* defined(ARM_COMPUTE_ENABLE_NEON) */
};
Status validate_arguments(const ITensorInfo *src, const ITensorInfo *dx, const ITensorInfo *dy,