diff options
Diffstat (limited to 'src/cpu/kernels/lut/list.h')
-rw-r--r-- | src/cpu/kernels/lut/list.h | 26 |
1 files changed, 18 insertions, 8 deletions
diff --git a/src/cpu/kernels/lut/list.h b/src/cpu/kernels/lut/list.h index da90346267..9acfe97728 100644 --- a/src/cpu/kernels/lut/list.h +++ b/src/cpu/kernels/lut/list.h @@ -1,5 +1,5 @@ /* - * Copyright (c) 2023 Arm Limited. + * Copyright (c) 2023-2024 Arm Limited. * * SPDX-License-Identifier: MIT * @@ -22,8 +22,8 @@ * SOFTWARE. */ -#ifndef SRC_CORE_NEON_KERNELS_LUT_LIST_H -#define SRC_CORE_NEON_KERNELS_LUT_LIST_H +#ifndef ACL_SRC_CPU_KERNELS_LUT_LIST_H +#define ACL_SRC_CPU_KERNELS_LUT_LIST_H #include <cstddef> #include <cstdint> @@ -34,17 +34,27 @@ namespace cpu { #ifdef __aarch64__ -#define DECLARE_LUT_KERNEL(func_name) \ +#define DECLARE_LUT_U8_KERNEL(func_name) \ void func_name(const uint8_t *table, size_t num_strings, size_t string_length, const uint8_t *const *input, \ uint8_t *const *output) -DECLARE_LUT_KERNEL(lut_u8_neon); -DECLARE_LUT_KERNEL(lut_u8_sve2); +DECLARE_LUT_U8_KERNEL(lut_u8_neon); +DECLARE_LUT_U8_KERNEL(lut_u8_sve2); + +#undef DECLARE_LUT_U8_KERNEL + +#define DECLARE_LUT_U16_KERNEL(func_name) \ + void func_name(const uint16_t *table, size_t num_strings, size_t string_length, const uint16_t *input, \ + uint16_t *output) + +DECLARE_LUT_U16_KERNEL(lut_u16_neon); +DECLARE_LUT_U16_KERNEL(lut_u16_sve); + +#undef DECLARE_LUT_U16_KERNEL -#undef DECLARE_LUT_KERNEL #endif // __aarch64__ } // namespace cpu } // namespace arm_compute -#endif // SRC_CORE_NEON_KERNELS_LUT_LIST_H +#endif // ACL_SRC_CPU_KERNELS_LUT_LIST_H |