1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
|
/*===-- flang/runtime/complex-reduction.c ---------------------------*- C -*-===
*
* Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
* See https://llvm.org/LICENSE.txt for license information.
* SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
*
* ===-----------------------------------------------------------------------===
*/
#include "complex-reduction.h"
#include "flang/Common/long-double.h"
struct CppComplexFloat {
float r, i;
};
struct CppComplexDouble {
double r, i;
};
struct CppComplexLongDouble {
long double r, i;
};
/* Not all environments define CMPLXF, CMPLX, CMPLXL. */
#ifndef CMPLXF
#if __clang_major__ >= 12
#define CMPLXF __builtin_complex
#else
static float_Complex_t CMPLXF(float r, float i) {
union {
struct CppComplexFloat x;
float_Complex_t result;
} u;
u.x.r = r;
u.x.i = i;
return u.result;
}
#endif
#endif
#ifndef CMPLX
#if __clang_major__ >= 12
#define CMPLX __builtin_complex
#else
static double_Complex_t CMPLX(double r, double i) {
union {
struct CppComplexDouble x;
double_Complex_t result;
} u;
u.x.r = r;
u.x.i = i;
return u.result;
}
#endif
#endif
#ifndef CMPLXL
#if __clang_major__ >= 12
#define CMPLXL __builtin_complex
#else
static long_double_Complex_t CMPLXL(long double r, long double i) {
union {
struct CppComplexLongDouble x;
long_double_Complex_t result;
} u;
u.x.r = r;
u.x.i = i;
return u.result;
}
#endif
#endif
/* RTNAME(SumComplex4) calls RTNAME(CppSumComplex4) with the same arguments
* and converts the members of its C++ complex result to C _Complex.
*/
#define CPP_NAME(name) Cpp##name
#define ADAPT_REDUCTION(name, cComplex, cpptype, cmplxMacro, ARGS, ARG_NAMES) \
struct cpptype RTNAME(CPP_NAME(name))(struct cpptype *, ARGS); \
cComplex RTNAME(name)(ARGS) { \
struct cpptype result; \
RTNAME(CPP_NAME(name))(&result, ARG_NAMES); \
return cmplxMacro(result.r, result.i); \
}
/* TODO: COMPLEX(2 & 3) */
/* SUM() */
ADAPT_REDUCTION(SumComplex4, float_Complex_t, CppComplexFloat, CMPLXF,
REDUCTION_ARGS, REDUCTION_ARG_NAMES)
ADAPT_REDUCTION(SumComplex8, double_Complex_t, CppComplexDouble, CMPLX,
REDUCTION_ARGS, REDUCTION_ARG_NAMES)
#if LONG_DOUBLE == 80
ADAPT_REDUCTION(SumComplex10, long_double_Complex_t, CppComplexLongDouble,
CMPLXL, REDUCTION_ARGS, REDUCTION_ARG_NAMES)
#elif LONG_DOUBLE == 128
ADAPT_REDUCTION(SumComplex16, long_double_Complex_t, CppComplexLongDouble,
CMPLXL, REDUCTION_ARGS, REDUCTION_ARG_NAMES)
#endif
/* PRODUCT() */
ADAPT_REDUCTION(ProductComplex4, float_Complex_t, CppComplexFloat, CMPLXF,
REDUCTION_ARGS, REDUCTION_ARG_NAMES)
ADAPT_REDUCTION(ProductComplex8, double_Complex_t, CppComplexDouble, CMPLX,
REDUCTION_ARGS, REDUCTION_ARG_NAMES)
#if LONG_DOUBLE == 80
ADAPT_REDUCTION(ProductComplex10, long_double_Complex_t, CppComplexLongDouble,
CMPLXL, REDUCTION_ARGS, REDUCTION_ARG_NAMES)
#elif LONG_DOUBLE == 128
ADAPT_REDUCTION(ProductComplex16, long_double_Complex_t, CppComplexLongDouble,
CMPLXL, REDUCTION_ARGS, REDUCTION_ARG_NAMES)
#endif
/* DOT_PRODUCT() */
ADAPT_REDUCTION(DotProductComplex4, float_Complex_t, CppComplexFloat, CMPLXF,
DOT_PRODUCT_ARGS, DOT_PRODUCT_ARG_NAMES)
ADAPT_REDUCTION(DotProductComplex8, double_Complex_t, CppComplexDouble, CMPLX,
DOT_PRODUCT_ARGS, DOT_PRODUCT_ARG_NAMES)
#if LONG_DOUBLE == 80
ADAPT_REDUCTION(DotProductComplex10, long_double_Complex_t,
CppComplexLongDouble, CMPLXL, DOT_PRODUCT_ARGS, DOT_PRODUCT_ARG_NAMES)
#elif LONG_DOUBLE == 128
ADAPT_REDUCTION(DotProductComplex16, long_double_Complex_t,
CppComplexLongDouble, CMPLXL, DOT_PRODUCT_ARGS, DOT_PRODUCT_ARG_NAMES)
#endif
|
