FLANG
complex-reduction.h
1/*===-- flang/runtime/complex-reduction.h ---------------------------*- C -*-===
2 *
3 * Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
4 * See https://llvm.org/LICENSE.txt for license information.
5 * SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
6 *
7 * ===-----------------------------------------------------------------------===
8 */
9
10/* Wraps the C++-coded complex-valued SUM and PRODUCT reductions with
11 * C-coded wrapper functions returning _Complex values, to avoid problems
12 * with C++ build compilers that don't support C's _Complex.
13 */
14
15#ifndef FORTRAN_RUNTIME_COMPLEX_REDUCTION_H_
16#define FORTRAN_RUNTIME_COMPLEX_REDUCTION_H_
17
18#include "flang/Common/float128.h"
19#include "flang/Runtime/entry-names.h"
20#include <complex.h>
21
22struct CppDescriptor; /* dummy type name for Fortran::runtime::Descriptor */
23
24#if defined(_MSC_VER) && !(defined(__clang_major__) && __clang_major__ >= 12)
25typedef _Fcomplex float_Complex_t;
26typedef _Dcomplex double_Complex_t;
27typedef _Lcomplex long_double_Complex_t;
28#else
29typedef float _Complex float_Complex_t;
30typedef double _Complex double_Complex_t;
31typedef long double _Complex long_double_Complex_t;
32#endif
33
34#define REDUCTION_ARGS \
35 const struct CppDescriptor *x, const char *source, int line, int dim /*=0*/, \
36 const struct CppDescriptor *mask /*=NULL*/
37#define REDUCTION_ARG_NAMES x, source, line, dim, mask
38
39float_Complex_t RTNAME(SumComplex2)(REDUCTION_ARGS);
40float_Complex_t RTNAME(SumComplex3)(REDUCTION_ARGS);
41float_Complex_t RTNAME(SumComplex4)(REDUCTION_ARGS);
42double_Complex_t RTNAME(SumComplex8)(REDUCTION_ARGS);
43long_double_Complex_t RTNAME(SumComplex10)(REDUCTION_ARGS);
44#if HAS_LDBL128 || HAS_FLOAT128
45CFloat128ComplexType RTNAME(SumComplex16)(REDUCTION_ARGS);
46#endif
47
48float_Complex_t RTNAME(ProductComplex2)(REDUCTION_ARGS);
49float_Complex_t RTNAME(ProductComplex3)(REDUCTION_ARGS);
50float_Complex_t RTNAME(ProductComplex4)(REDUCTION_ARGS);
51double_Complex_t RTNAME(ProductComplex8)(REDUCTION_ARGS);
52long_double_Complex_t RTNAME(ProductComplex10)(REDUCTION_ARGS);
53#if HAS_LDBL128 || HAS_FLOAT128
54CFloat128ComplexType RTNAME(ProductComplex16)(REDUCTION_ARGS);
55#endif
56
57#define DOT_PRODUCT_ARGS \
58 const struct CppDescriptor *x, const struct CppDescriptor *y, \
59 const char *source, int line, int dim /*=0*/, \
60 const struct CppDescriptor *mask /*=NULL*/
61#define DOT_PRODUCT_ARG_NAMES x, y, source, line, dim, mask
62
63float_Complex_t RTNAME(DotProductComplex2)(DOT_PRODUCT_ARGS);
64float_Complex_t RTNAME(DotProductComplex3)(DOT_PRODUCT_ARGS);
65float_Complex_t RTNAME(DotProductComplex4)(DOT_PRODUCT_ARGS);
66double_Complex_t RTNAME(DotProductComplex8)(DOT_PRODUCT_ARGS);
67long_double_Complex_t RTNAME(DotProductComplex10)(DOT_PRODUCT_ARGS);
68#if HAS_LDBL128 || HAS_FLOAT128
69CFloat128ComplexType RTNAME(DotProductComplex16)(DOT_PRODUCT_ARGS);
70#endif
71
72#define REDUCE_ARGS(T, OP) \
73 OP operation, const struct CppDescriptor *x, const struct CppDescriptor *y, \
74 const char *source, int line, int dim /*=0*/, \
75 const struct CppDescriptor *mask /*=NULL*/, const T *identity /*=NULL*/, \
76 _Bool ordered /*=true*/
77#define REDUCE_ARG_NAMES \
78 operation, x, y, source, line, dim, mask, identity, ordered
79
80typedef float_Complex_t (*float_Complex_t_ref_op)(
81 const float_Complex_t *, const float_Complex_t *);
82typedef float_Complex_t (*float_Complex_t_value_op)(
83 float_Complex_t, float_Complex_t);
84typedef double_Complex_t (*double_Complex_t_ref_op)(
85 const double_Complex_t *, const double_Complex_t *);
86typedef double_Complex_t (*double_Complex_t_value_op)(
87 double_Complex_t, double_Complex_t);
88typedef long_double_Complex_t (*long_double_Complex_t_ref_op)(
89 const long_double_Complex_t *, const long_double_Complex_t *);
90typedef long_double_Complex_t (*long_double_Complex_t_value_op)(
91 long_double_Complex_t, long_double_Complex_t);
92
93float_Complex_t RTNAME(ReduceComplex2Ref)(
94 REDUCE_ARGS(float_Complex_t, float_Complex_t_ref_op));
95float_Complex_t RTNAME(ReduceComplex2Value)(
96 REDUCE_ARGS(float_Complex_t, float_Complex_t_value_op));
97float_Complex_t RTNAME(ReduceComplex3Ref)(
98 REDUCE_ARGS(float_Complex_t, float_Complex_t_ref_op));
99float_Complex_t RTNAME(ReduceComplex3Value)(
100 REDUCE_ARGS(float_Complex_t, float_Complex_t_value_op));
101float_Complex_t RTNAME(ReduceComplex4Ref)(
102 REDUCE_ARGS(float_Complex_t, float_Complex_t_ref_op));
103float_Complex_t RTNAME(ReduceComplex4Value)(
104 REDUCE_ARGS(float_Complex_t, float_Complex_t_value_op));
105double_Complex_t RTNAME(ReduceComplex8Ref)(
106 REDUCE_ARGS(double_Complex_t, double_Complex_t_ref_op));
107double_Complex_t RTNAME(ReduceComplex8Value)(
108 REDUCE_ARGS(double_Complex_t, double_Complex_t_value_op));
109long_double_Complex_t RTNAME(ReduceComplex10Ref)(
110 REDUCE_ARGS(long_double_Complex_t, long_double_Complex_t_ref_op));
111long_double_Complex_t RTNAME(ReduceComplex10Value)(
112 REDUCE_ARGS(long_double_Complex_t, long_double_Complex_t_value_op));
113#if HAS_LDBL128 || HAS_FLOAT128
114typedef CFloat128ComplexType (*CFloat128ComplexType_ref_op)(
115 const CFloat128ComplexType *, const CFloat128ComplexType *);
116typedef CFloat128ComplexType (*CFloat128ComplexType_value_op)(
117 CFloat128ComplexType, CFloat128ComplexType);
118CFloat128ComplexType RTNAME(ReduceComplex16Ref)(
119 REDUCE_ARGS(CFloat128ComplexType, CFloat128ComplexType_ref_op));
120CFloat128ComplexType RTNAME(ReduceComplex16Value)(
121 REDUCE_ARGS(CFloat128ComplexType, CFloat128ComplexType_value_op));
122#endif
123
124#define REDUCE_DIM_ARGS(T, OP) \
125 struct CppDescriptor *result, OP operation, const struct CppDescriptor *x, \
126 const struct CppDescriptor *y, const char *source, int line, int dim, \
127 const struct CppDescriptor *mask /*=NULL*/, const T *identity /*=NULL*/, \
128 _Bool ordered /*=true*/
129#define REDUCE_DIM_ARG_NAMES \
130 result, operation, x, y, source, line, dim, mask, identity, ordered
131
132void RTNAME(ReduceComplex2DimRef)(
133 REDUCE_DIM_ARGS(float_Complex_t, float_Complex_t_ref_op));
134void RTNAME(ReduceComplex2DimValue)(
135 REDUCE_DIM_ARGS(float_Complex_t, float_Complex_t_value_op));
136void RTNAME(ReduceComplex3DimRef)(
137 REDUCE_DIM_ARGS(float_Complex_t, float_Complex_t_ref_op));
138void RTNAME(ReduceComplex3DimValue)(
139 REDUCE_DIM_ARGS(float_Complex_t, float_Complex_t_value_op));
140void RTNAME(ReduceComplex4DimRef)(
141 REDUCE_DIM_ARGS(float_Complex_t, float_Complex_t_ref_op));
142void RTNAME(ReduceComplex4DimValue)(
143 REDUCE_DIM_ARGS(float_Complex_t, float_Complex_t_value_op));
144void RTNAME(ReduceComplex8DimRef)(
145 REDUCE_DIM_ARGS(double_Complex_t, double_Complex_t_ref_op));
146void RTNAME(ReduceComplex8DimValue)(
147 REDUCE_DIM_ARGS(double_Complex_t, double_Complex_t_value_op));
148void RTNAME(ReduceComplex10DimRef)(
149 REDUCE_DIM_ARGS(long_double_Complex_t, long_double_Complex_t_ref_op));
150void RTNAME(ReduceComplex10DimValue)(
151 REDUCE_DIM_ARGS(long_double_Complex_t, long_double_Complex_t_value_op));
152#if HAS_LDBL128 || HAS_FLOAT128
153void RTNAME(ReduceComplex16DimRef)(
154 REDUCE_DIM_ARGS(CFloat128ComplexType, CFloat128ComplexType_ref_op));
155void RTNAME(ReduceComplex16DimValue)(
156 REDUCE_DIM_ARGS(CFloat128ComplexType, CFloat128ComplexType_value_op));
157#endif
158
159#endif // FORTRAN_RUNTIME_COMPLEX_REDUCTION_H_
Generated on Fri Apr 25 2025 19:44:09 for FLANG by doxygen 1.9.6