Gitiles
Code Review Sign In
review.mlplatform.org / ml / ComputeLibrary / 5c002ec70aa20569d44a3e4c5bbcf53135364e7b / . / src / core / common / Registrars.h
blob: 65f6c7093d44324e6680d1af35589046d16c7fe8 [file] [log] [blame]
Georgios Pinitasff4fca02020-10-02 21:00:00 +0100[diff] [blame]1/*
Michalis Spyroua3c9a3b2020-12-08 21:02:16 +0000[diff] [blame]2 * Copyright (c) 2020-2021 Arm Limited.
Georgios Pinitasff4fca02020-10-02 21:00:00 +0100[diff] [blame]3 *
4 * SPDX-License-Identifier: MIT
5 *
6 * Permission is hereby granted, free of charge, to any person obtaining a copy
7 * of this software and associated documentation files (the "Software"), to
8 * deal in the Software without restriction, including without limitation the
9 * rights to use, copy, modify, merge, publish, distribute, sublicense, and/or
10 * sell copies of the Software, and to permit persons to whom the Software is
11 * furnished to do so, subject to the following conditions:
12 *
13 * The above copyright notice and this permission notice shall be included in all
14 * copies or substantial portions of the Software.
15 *
16 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
17 * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
18 * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
19 * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
20 * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
21 * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
22 * SOFTWARE.
23 */
24#ifndef SRC_CORE_COMMON_REGISTRARS_H
25#define SRC_CORE_COMMON_REGISTRARS_H
26
Michalis Spyrouaa51a5b2020-11-22 00:49:42 +0000[diff] [blame]27#if defined(ENABLE_FP16_KERNELS)
28
Michalis Spyrou20fca522021-06-07 14:23:57 +0100[diff] [blame]29#if defined(ARM_COMPUTE_ENABLE_SVE)
Michalis Spyrouaa51a5b2020-11-22 00:49:42 +0000[diff] [blame]30#define REGISTER_FP16_SVE(func_name) &(func_name)
Michalis Spyrou20fca522021-06-07 14:23:57 +0100[diff] [blame]31#else /* !defined(ARM_COMPUTE_ENABLE_SVE) */
Michalis Spyrouaa51a5b2020-11-22 00:49:42 +0000[diff] [blame]32#define REGISTER_FP16_SVE(func_name) nullptr
Michalis Spyrou20fca522021-06-07 14:23:57 +0100[diff] [blame]33#endif /* defined(ARM_COMPUTE_ENABLE_SVE) */
Michalis Spyrouaa51a5b2020-11-22 00:49:42 +0000[diff] [blame]34
Michalis Spyrou20fca522021-06-07 14:23:57 +0100[diff] [blame]35#if defined(ARM_COMPUTE_ENABLE_NEON) && defined(__ARM_FEATURE_FP16_VECTOR_ARITHMETIC)
Georgios Pinitasff4fca02020-10-02 21:00:00 +0100[diff] [blame]36#define REGISTER_FP16_NEON(func_name) &(func_name)
Michalis Spyrou20fca522021-06-07 14:23:57 +0100[diff] [blame]37#else /* !defined(ARM_COMPUTE_ENABLE_NEON) */
Georgios Pinitasff4fca02020-10-02 21:00:00 +0100[diff] [blame]38#define REGISTER_FP16_NEON(func_name) nullptr
Michalis Spyrou20fca522021-06-07 14:23:57 +0100[diff] [blame]39#endif /* defined(ARM_COMPUTE_ENABLE_NEON) && defined(__ARM_FEATURE_FP16_VECTOR_ARITHMETIC) */
Michalis Spyrouaa51a5b2020-11-22 00:49:42 +0000[diff] [blame]40
41#else /* !defined(ENABLE_FP16_KERNELS) */
42#define REGISTER_FP16_NEON(func_name) nullptr
43#define REGISTER_FP16_SVE(func_name) nullptr
Georgios Pinitasff4fca02020-10-02 21:00:00 +0100[diff] [blame]44#endif /* defined(__ARM_FEATURE_FP16_VECTOR_ARITHMETIC) && defined(ENABLE_FP16_KERNELS) */
45
46#if defined(ENABLE_FP32_KERNELS)
Georgios Pinitasbdcdc392021-04-22 16:42:03 +0100[diff] [blame]47
Michalis Spyrou20fca522021-06-07 14:23:57 +0100[diff] [blame]48#if defined(ARM_COMPUTE_ENABLE_SVE)
Michalis Spyrouaa51a5b2020-11-22 00:49:42 +0000[diff] [blame]49#define REGISTER_FP32_SVE(func_name) &(func_name)
Michalis Spyrou20fca522021-06-07 14:23:57 +0100[diff] [blame]50#else /* !defined(ARM_COMPUTE_ENABLE_SVE) */
Georgios Pinitasbdcdc392021-04-22 16:42:03 +0100[diff] [blame]51#define REGISTER_FP32_SVE(func_name) nullptr
Michalis Spyrou20fca522021-06-07 14:23:57 +0100[diff] [blame]52#endif /* defined(ARM_COMPUTE_ENABLE_SVE) */
Georgios Pinitasbdcdc392021-04-22 16:42:03 +0100[diff] [blame]53
Michalis Spyrou20fca522021-06-07 14:23:57 +0100[diff] [blame]54#if defined(ARM_COMPUTE_ENABLE_NEON)
Georgios Pinitasff4fca02020-10-02 21:00:00 +0100[diff] [blame]55#define REGISTER_FP32_NEON(func_name) &(func_name)
Michalis Spyrou20fca522021-06-07 14:23:57 +0100[diff] [blame]56#else /* !defined(ARM_COMPUTE_ENABLE_NEON) */
Georgios Pinitasbdcdc392021-04-22 16:42:03 +0100[diff] [blame]57#define REGISTER_FP32_NEON(func_name) nullptr
Michalis Spyrou20fca522021-06-07 14:23:57 +0100[diff] [blame]58#endif /* defined(ARM_COMPUTE_ENABLE_NEON) */
Georgios Pinitasbdcdc392021-04-22 16:42:03 +0100[diff] [blame]59
Georgios Pinitasff4fca02020-10-02 21:00:00 +0100[diff] [blame]60#else /* defined(ENABLE_FP32_KERNELS) */
61#define REGISTER_FP32_NEON(func_name) nullptr
Michalis Spyrouaa51a5b2020-11-22 00:49:42 +0000[diff] [blame]62#define REGISTER_FP32_SVE(func_name) nullptr
Georgios Pinitasff4fca02020-10-02 21:00:00 +0100[diff] [blame]63#endif /* defined(ENABLE_FP32_KERNELS) */
64
Michalis Spyrouc4d45552020-10-19 12:41:30 +0100[diff] [blame]65#if defined(ENABLE_QASYMM8_SIGNED_KERNELS)
Georgios Pinitasbdcdc392021-04-22 16:42:03 +0100[diff] [blame]66
Michalis Spyrouc4d45552020-10-19 12:41:30 +0100[diff] [blame]67#define REGISTER_QASYMM8_SIGNED_NEON(func_name) &(func_name)
Georgios Pinitasbdcdc392021-04-22 16:42:03 +0100[diff] [blame]68
Michalis Spyrou20fca522021-06-07 14:23:57 +0100[diff] [blame]69#if defined(ARM_COMPUTE_ENABLE_SVE)
Georgios Pinitasbdcdc392021-04-22 16:42:03 +0100[diff] [blame]70#define REGISTER_QASYMM8_SIGNED_SVE(func_name) &(func_name)
Michalis Spyrou20fca522021-06-07 14:23:57 +0100[diff] [blame]71#else /* !defined(ARM_COMPUTE_ENABLE_SVE) */
Georgios Pinitasbdcdc392021-04-22 16:42:03 +0100[diff] [blame]72#define REGISTER_QASYMM8_SIGNED_SVE(func_name) nullptr
Michalis Spyrou20fca522021-06-07 14:23:57 +0100[diff] [blame]73#endif /* defined(ARM_COMPUTE_ENABLE_SVE) */
Georgios Pinitasbdcdc392021-04-22 16:42:03 +0100[diff] [blame]74
Michalis Spyrouc4d45552020-10-19 12:41:30 +0100[diff] [blame]75#else /* defined(ENABLE_QASYMM8_SIGNED_KERNELS) */
76#define REGISTER_QASYMM8_SIGNED_NEON(func_name) nullptr
Michalis Spyrouaa51a5b2020-11-22 00:49:42 +0000[diff] [blame]77#define REGISTER_QASYMM8_SIGNED_SVE(func_name) nullptr
Michalis Spyrouc4d45552020-10-19 12:41:30 +0100[diff] [blame]78#endif /* defined(ENABLE_QASYMM8_SIGNED_KERNELS) */
79
80#if defined(ENABLE_QASYMM8_KERNELS)
81#define REGISTER_QASYMM8_NEON(func_name) &(func_name)
Georgios Pinitasbdcdc392021-04-22 16:42:03 +0100[diff] [blame]82
Michalis Spyrou20fca522021-06-07 14:23:57 +0100[diff] [blame]83#if defined(ARM_COMPUTE_ENABLE_SVE)
Georgios Pinitasbdcdc392021-04-22 16:42:03 +0100[diff] [blame]84#define REGISTER_QASYMM8_SVE(func_name) &(func_name)
Michalis Spyrou20fca522021-06-07 14:23:57 +0100[diff] [blame]85#else /* !defined(ARM_COMPUTE_ENABLE_SVE) */
Georgios Pinitasbdcdc392021-04-22 16:42:03 +0100[diff] [blame]86#define REGISTER_QASYMM8_SVE(func_name) nullptr
Michalis Spyrou20fca522021-06-07 14:23:57 +0100[diff] [blame]87#endif /* defined(ARM_COMPUTE_ENABLE_SVE) */
Georgios Pinitasbdcdc392021-04-22 16:42:03 +0100[diff] [blame]88
Michalis Spyrouc4d45552020-10-19 12:41:30 +0100[diff] [blame]89#else /* defined(ENABLE_QASYMM8_KERNELS) */
90#define REGISTER_QASYMM8_NEON(func_name) nullptr
Michalis Spyrouaa51a5b2020-11-22 00:49:42 +0000[diff] [blame]91#define REGISTER_QASYMM8_SVE(func_name) nullptr
Michalis Spyrouc4d45552020-10-19 12:41:30 +0100[diff] [blame]92#endif /* defined(ENABLE_QASYMM8_KERNELS) */
93
94#if defined(ENABLE_QSYMM16_KERNELS)
Georgios Pinitasbdcdc392021-04-22 16:42:03 +0100[diff] [blame]95
Michalis Spyrouc4d45552020-10-19 12:41:30 +0100[diff] [blame]96#define REGISTER_QSYMM16_NEON(func_name) &(func_name)
Georgios Pinitasbdcdc392021-04-22 16:42:03 +0100[diff] [blame]97
Michalis Spyrou20fca522021-06-07 14:23:57 +0100[diff] [blame]98#if defined(ARM_COMPUTE_ENABLE_SVE)
Georgios Pinitasbdcdc392021-04-22 16:42:03 +0100[diff] [blame]99#define REGISTER_QSYMM16_SVE(func_name) &(func_name)
Michalis Spyrou20fca522021-06-07 14:23:57 +0100[diff] [blame]100#else /* !defined(ARM_COMPUTE_ENABLE_SVE) */
Georgios Pinitasbdcdc392021-04-22 16:42:03 +0100[diff] [blame]101#define REGISTER_QSYMM16_SVE(func_name) nullptr
Michalis Spyrou20fca522021-06-07 14:23:57 +0100[diff] [blame]102#endif /* defined(ARM_COMPUTE_ENABLE_SVE) */
Georgios Pinitasbdcdc392021-04-22 16:42:03 +0100[diff] [blame]103
Michalis Spyrouc4d45552020-10-19 12:41:30 +0100[diff] [blame]104#else /* defined(ENABLE_QSYMM16_KERNELS) */
105#define REGISTER_QSYMM16_NEON(func_name) nullptr
Michalis Spyrouaa51a5b2020-11-22 00:49:42 +0000[diff] [blame]106#define REGISTER_QSYMM16_SVE(func_name) nullptr
Michalis Spyrouc4d45552020-10-19 12:41:30 +0100[diff] [blame]107#endif /* defined(ENABLE_QSYMM16_KERNELS) */
108
Michalis Spyroua3c9a3b2020-12-08 21:02:16 +0000[diff] [blame]109#if defined(ENABLE_INTEGER_KERNELS)
Georgios Pinitasbdcdc392021-04-22 16:42:03 +0100[diff] [blame]110
Michalis Spyrou20fca522021-06-07 14:23:57 +0100[diff] [blame]111#if defined(ARM_COMPUTE_ENABLE_SVE)
Michalis Spyroua3c9a3b2020-12-08 21:02:16 +0000[diff] [blame]112#define REGISTER_INTEGER_SVE(func_name) &(func_name)
Michalis Spyrou20fca522021-06-07 14:23:57 +0100[diff] [blame]113#else /* !defined(ARM_COMPUTE_ENABLE_SVE) */
Georgios Pinitasbdcdc392021-04-22 16:42:03 +0100[diff] [blame]114#define REGISTER_INTEGER_SVE(func_name) nullptr
Michalis Spyrou20fca522021-06-07 14:23:57 +0100[diff] [blame]115#endif /* defined(ARM_COMPUTE_ENABLE_SVE) */
Georgios Pinitasbdcdc392021-04-22 16:42:03 +0100[diff] [blame]116
Michalis Spyrou20fca522021-06-07 14:23:57 +0100[diff] [blame]117#if defined(ARM_COMPUTE_ENABLE_NEON)
Michalis Spyroua3c9a3b2020-12-08 21:02:16 +0000[diff] [blame]118#define REGISTER_INTEGER_NEON(func_name) &(func_name)
Michalis Spyrou20fca522021-06-07 14:23:57 +0100[diff] [blame]119#else /* !defined(ARM_COMPUTE_ENABLE_NEON) */
Georgios Pinitasbdcdc392021-04-22 16:42:03 +0100[diff] [blame]120#define REGISTER_INTEGER_NEON(func_name) nullptr
Michalis Spyrou20fca522021-06-07 14:23:57 +0100[diff] [blame]121#endif /* defined(ARM_COMPUTE_ENABLE_NEON) */
Georgios Pinitasbdcdc392021-04-22 16:42:03 +0100[diff] [blame]122
Michalis Spyroua3c9a3b2020-12-08 21:02:16 +0000[diff] [blame]123#else /* defined(ENABLE_INTEGER_KERNELS) */
124#define REGISTER_INTEGER_NEON(func_name) nullptr
125#define REGISTER_INTEGER_SVE(func_name) nullptr
126#endif /* defined(ENABLE_INTEGER_KERNELS) */
127
Georgios Pinitasff4fca02020-10-02 21:00:00 +0100[diff] [blame]128#endif /* SRC_CORE_COMMON_REGISTRARS_H */