diff options
author | SiCong Li <sicong.li@arm.com> | 2023-05-17 13:46:13 +0100 |
---|---|---|
committer | SiCong Li <sicong.li@arm.com> | 2023-05-17 16:21:31 +0100 |
commit | 3f70cd053573cb6140990ab619ead865f55f3139 (patch) | |
tree | 6ed9cf0222c4b0ab1def1c9276bef41767accd36 /src/cpu/kernels/activation | |
parent | 81ca48606cea2220a83ae8d736d3935bcc17f854 (diff) | |
download | ComputeLibrary-branches/arm_compute_23_05.tar.gz |
Move lut kernel to sve2 categoryv23.05branches/arm_compute_23_05
This specific Lut kernel uses sve2 instructions
Resolves: COMPMID-6268
Signed-off-by: SiCong Li <sicong.li@arm.com>
Change-Id: I44fa3812e96fa79b3d1e1e3a31d587581f59f0e1
Reviewed-on: https://review.mlplatform.org/c/ml/ComputeLibrary/+/9675
Reviewed-by: Viet-Hoa Do <viet-hoa.do@arm.com>
Reviewed-by: Jakub Sujak <jakub.sujak@arm.com>
Comments-Addressed: Arm Jenkins <bsgcomp@arm.com>
Benchmark: Arm Jenkins <bsgcomp@arm.com>
Tested-by: Arm Jenkins <bsgcomp@arm.com>
Diffstat (limited to 'src/cpu/kernels/activation')
-rw-r--r-- | src/cpu/kernels/activation/generic/sve2/lut.cpp (renamed from src/cpu/kernels/activation/generic/sve/lut.cpp) | 4 | ||||
-rw-r--r-- | src/cpu/kernels/activation/list.h | 4 |
2 files changed, 4 insertions, 4 deletions
diff --git a/src/cpu/kernels/activation/generic/sve/lut.cpp b/src/cpu/kernels/activation/generic/sve2/lut.cpp index b4042662b9..2e5975744b 100644 --- a/src/cpu/kernels/activation/generic/sve/lut.cpp +++ b/src/cpu/kernels/activation/generic/sve2/lut.cpp @@ -30,7 +30,7 @@ namespace arm_compute namespace cpu { #ifdef __aarch64__ -void sve_q8_activation_lut(const ITensor *src, ITensor *dst, const ActivationLayerInfo &act_info, const Window &window) +void sve2_q8_activation_lut(const ITensor *src, ITensor *dst, const ActivationLayerInfo &act_info, const Window &window) { ARM_COMPUTE_ERROR_ON(!ActivationLayerInfo::is_lut_supported(act_info.activation(), src->info()->data_type())); const auto window_end_x = window.x().end(); @@ -42,7 +42,7 @@ void sve_q8_activation_lut(const ITensor *src, ITensor *dst, const ActivationLay { const auto input_ptr = input.ptr(); auto output_ptr = output.ptr(); - lut_u8_sve(act_info.lut().data(), 1u, window_end_x, &input_ptr, &output_ptr); + lut_u8_sve2(act_info.lut().data(), 1u, window_end_x, &input_ptr, &output_ptr); }, input, output); } diff --git a/src/cpu/kernels/activation/list.h b/src/cpu/kernels/activation/list.h index c2149b38ff..6550ddfeca 100644 --- a/src/cpu/kernels/activation/list.h +++ b/src/cpu/kernels/activation/list.h @@ -1,5 +1,5 @@ /* - * Copyright (c) 2020-2022 Arm Limited. + * Copyright (c) 2020-2023 Arm Limited. * * SPDX-License-Identifier: MIT * @@ -34,7 +34,7 @@ namespace cpu #ifdef __aarch64__ DECLARE_ACTIVATION_KERNEL(neon_q8_activation_lut); #endif // __aarch64__ -DECLARE_ACTIVATION_KERNEL(sve_q8_activation_lut); +DECLARE_ACTIVATION_KERNEL(sve2_q8_activation_lut); DECLARE_ACTIVATION_KERNEL(neon_qasymm8_activation); DECLARE_ACTIVATION_KERNEL(sve2_qasymm8_activation); DECLARE_ACTIVATION_KERNEL(neon_qasymm8_signed_activation); |