xref: /aosp_15_r20/external/arm-optimized-routines/pl/math/pl_sig.h (revision 412f47f9e737e10ed5cc46ec6a8d7fa2264f8a14)
1 /*
2  * PL macros for emitting various ulp/bench entries based on function signature
3  *
4  * Copyright (c) 2022-2024, Arm Limited.
5  * SPDX-License-Identifier: MIT OR Apache-2.0 WITH LLVM-exception.
6  */
7 
8 #define PL_DECL_SF1(fun) float fun##f (float);
9 #define PL_DECL_SF2(fun) float fun##f (float, float);
10 #define PL_DECL_SD1(fun) double fun (double);
11 #define PL_DECL_SD2(fun) double fun (double, double);
12 
13 #if WANT_VMATH
14 # define V_NAME_F1(fun) _ZGVnN4v_##fun##f
15 # define V_NAME_D1(fun) _ZGVnN2v_##fun
16 # define V_NAME_F2(fun) _ZGVnN4vv_##fun##f
17 # define V_NAME_D2(fun) _ZGVnN2vv_##fun
18 # define PL_DECL_VF1(fun)                                                    \
19     VPCS_ATTR float32x4_t V_NAME_F1 (fun##f) (float32x4_t);
20 # define PL_DECL_VF2(fun)                                                    \
21     VPCS_ATTR float32x4_t V_NAME_F2 (fun##f) (float32x4_t, float32x4_t);
22 # define PL_DECL_VD1(fun) VPCS_ATTR float64x2_t V_NAME_D1 (fun) (float64x2_t);
23 # define PL_DECL_VD2(fun)                                                    \
24     VPCS_ATTR float64x2_t V_NAME_D2 (fun) (float64x2_t, float64x2_t);
25 #endif
26 
27 #if WANT_SVE_MATH
28 # define SV_NAME_F1(fun) _ZGVsMxv_##fun##f
29 # define SV_NAME_D1(fun) _ZGVsMxv_##fun
30 # define SV_NAME_F2(fun) _ZGVsMxvv_##fun##f
31 # define SV_NAME_D2(fun) _ZGVsMxvv_##fun
32 # define PL_DECL_SVF1(fun)                                                   \
33     svfloat32_t SV_NAME_F1 (fun) (svfloat32_t, svbool_t);
34 # define PL_DECL_SVF2(fun)                                                   \
35     svfloat32_t SV_NAME_F2 (fun) (svfloat32_t, svfloat32_t, svbool_t);
36 # define PL_DECL_SVD1(fun)                                                   \
37     svfloat64_t SV_NAME_D1 (fun) (svfloat64_t, svbool_t);
38 # define PL_DECL_SVD2(fun)                                                   \
39     svfloat64_t SV_NAME_D2 (fun) (svfloat64_t, svfloat64_t, svbool_t);
40 #endif
41 
42 /* For building the routines, emit function prototype from PL_SIG. This
43    ensures that the correct signature has been chosen (wrong one will be a
44    compile error). PL_SIG is defined differently by various components of the
45    build system to emit entries in the wrappers and entries for mathbench and
46    ulp.  */
47 #ifndef _PL_SIG
48 # if defined(EMIT_ULP_FUNCS)
49 #  define _PL_SIG(v, t, a, f, ...) PL_SIG _Z##v##t##a (f)
50 # elif defined(EMIT_ULP_WRAPPERS)
51 #  define _PL_SIG(v, t, a, f, ...) PL_SIG Z##v##N##t##a##_WRAP (f)
52 # elif defined(EMIT_MATHBENCH_FUNCS)
53 #  define _PL_SIG(v, t, a, f, ...) PL_SIG _Z##v##t##a (f, ##__VA_ARGS__)
54 # else
55 #  define _PL_SIG(v, t, a, f, ...) PL_DECL_##v##t##a (f)
56 # endif
57 #endif
58 
59 #define PL_SIG(...) _PL_SIG (__VA_ARGS__)
60