1 // RUN: %clang_cc1 -DCHECK -verify -fopenmp -x c++ -triple powerpc64le-unknown-unknown -fopenmp-targets=powerpc64le-ibm-linux-gnu -emit-llvm %s -o - | FileCheck %s --check-prefix CHECK --check-prefix CHECK-64
2 // RUN: %clang_cc1 -DCHECK -fopenmp -x c++ -std=c++11 -triple powerpc64le-unknown-unknown -fopenmp-targets=powerpc64le-ibm-linux-gnu -emit-pch -o %t %s
3 // RUN: %clang_cc1 -DCHECK -fopenmp -x c++ -triple powerpc64le-unknown-unknown -fopenmp-targets=powerpc64le-ibm-linux-gnu -std=c++11 -include-pch %t -verify %s -emit-llvm -o - | FileCheck %s --check-prefix CHECK --check-prefix CHECK-64
4 // RUN: %clang_cc1 -DCHECK -verify -fopenmp -x c++ -triple i386-unknown-unknown -fopenmp-targets=i386-pc-linux-gnu -emit-llvm %s -o - | FileCheck %s --check-prefix CHECK --check-prefix CHECK-32
5 // RUN: %clang_cc1 -DCHECK -fopenmp -x c++ -std=c++11 -triple i386-unknown-unknown -fopenmp-targets=i386-pc-linux-gnu -emit-pch -o %t %s
6 // RUN: %clang_cc1 -DCHECK -fopenmp -x c++ -triple i386-unknown-unknown -fopenmp-targets=i386-pc-linux-gnu -std=c++11 -include-pch %t -verify %s -emit-llvm -o - | FileCheck %s --check-prefix CHECK --check-prefix CHECK-32
7 
8 // RUN: %clang_cc1 -DLAMBDA -verify -fopenmp -x c++ -std=c++11 -triple powerpc64le-unknown-unknown -fopenmp-targets=powerpc64le-ibm-linux-gnu -emit-llvm %s -o - | FileCheck %s --check-prefix LAMBDA --check-prefix LAMBDA-64
9 // RUN: %clang_cc1 -DLAMBDA -fopenmp -x c++ -std=c++11 -triple powerpc64le-unknown-unknown -fopenmp-targets=powerpc64le-ibm-linux-gnu -emit-pch -o %t %s
10 // RUN: %clang_cc1 -DLAMBDA -fopenmp -x c++  -std=c++11 -triple powerpc64le-unknown-unknown -fopenmp-targets=powerpc64le-ibm-linux-gnu -std=c++11 -include-pch %t -verify %s -emit-llvm -o - | FileCheck %s --check-prefix LAMBDA --check-prefix LAMBDA-64
11 
12 // expected-no-diagnostics
13 #ifndef HEADER
14 #define HEADER
15 
16 template <typename T>
17 T tmain() {
18   T t_var = T();
19   T vec[] = {1, 2};
20 #pragma omp target teams distribute reduction(+: t_var)
21   for (int i = 0; i < 2; ++i) {
22     t_var += (T) i;
23   }
24   return T();
25 }
26 
27 int main() {
28   static int sivar;
29 #ifdef LAMBDA
30   // LAMBDA: [[RED_VAR:@.+]] = common global [8 x {{.+}}] zeroinitializer
31 
32   // LAMBDA-LABEL: @main
33   // LAMBDA: call void [[OUTER_LAMBDA:@.+]](
34   [&]() {
35     // LAMBDA: define{{.*}} internal{{.*}} void [[OUTER_LAMBDA]](
36     // LAMBDA: call i32 @__tgt_target_teams(i64 -1, i8* @{{[^,]+}}, i32 1, i8** %{{[^,]+}}, i8** %{{[^,]+}}, i{{64|32}}* {{.+}}@{{[^,]+}}, i32 0, i32 0), i64* {{.+}}@{{[^,]+}}, i32 0, i32 0), i32 0, i32 0)
37     // LAMBDA: call void @[[LOFFL1:.+]](
38     // LAMBDA:  ret
39 #pragma omp target teams distribute reduction(+: sivar)
40   for (int i = 0; i < 2; ++i) {
41     // LAMBDA: define{{.*}} internal{{.*}} void @[[LOFFL1]](i{{64|32}} [[SIVAR_ARG:%.+]])
42     // LAMBDA: [[SIVAR_ADDR:%.+]] = alloca i{{.+}},
43     // LAMBDA: [[SIVAR_CASTED:%.+]] = alloca i{{.+}},
44     // LAMBDA: store{{.+}} [[SIVAR_ARG]], {{.+}} [[SIVAR_ADDR]],
45     // LAMBDA: [[SIVAR_CONV:%.+]] = bitcast{{.+}} [[SIVAR_ADDR]] to
46     // LAMBDA: [[SIVAR:%.+]] = load i64, i64* [[SIVAR_CASTED]],
47     // LAMBDA: call void {{.+}} @__kmpc_fork_teams({{.+}}, i32 1, {{.+}} @[[LOUTL1:.+]] to {{.+}}, {{.+}} [[SIVAR]])
48     // LAMBDA: ret void
49 
50     // LAMBDA: define internal void @[[LOUTL1]]({{.+}}, {{.+}}, {{.+}} [[SIVAR_ARG:%.+]])
51     // Skip global and bound tid vars
52     // LAMBDA: {{.+}} = alloca i32*,
53     // LAMBDA: {{.+}} = alloca i32*,
54     // LAMBDA: [[SIVAR_ADDR:%.+]] = alloca i{{.+}},
55     // LAMBDA: [[SIVAR_PRIV:%.+]] = alloca i{{.+}},
56     // LAMBDA: [[RED_LIST:%.+]] = alloca [1 x {{.+}}],
57     // LAMBDA: store{{.+}} [[SIVAR_ARG]], {{.+}} [[SIVAR_ADDR]],
58     // LAMBDA: [[SIVAR_REF:%.+]] = bitcast i64* [[SIVAR_ADDR]] to i32*
59     // LAMBDA: store{{.+}} 0, {{.+}} [[SIVAR_PRIV]],
60 
61     // LAMBDA: call void @__kmpc_for_static_init_4(
62     // LAMBDA: store{{.+}}, {{.+}} [[SIVAR_PRIV]],
63     // LAMBDA: call void [[INNER_LAMBDA:@.+]](
64     // LAMBDA: call void @__kmpc_for_static_fini(
65     // LAMBDA: [[RED_LIST_GEP:%.+]] = getelementptr{{.+}} [[RED_LIST]],
66     // LAMBDA: [[SIVAR_PRIV_CAST:%.+]] = bitcast{{.+}} [[SIVAR_PRIV]] to
67     // LAMBDA: store{{.+}} [[SIVAR_PRIV_CAST]], {{.+}} [[RED_LIST_GEP]],
68     // LAMBDA: [[RED_LIST_BCAST:%.+]] = bitcast{{.+}} [[RED_LIST]] to
69     // LAMBDA: [[K_RED_RET:%.+]] = call{{.+}} @__kmpc_reduce({{.+}}, {{.+}}, {{.+}}, {{.+}}, {{.+}} [[RED_LIST_BCAST]], {{.+}} [[RED_FUN:@.+]], {{.+}} [[RED_VAR]])
70     // LAMBDA: switch{{.+}} [[K_RED_RET]], label{{.+}} [
71     // LAMBDA: {{.+}}, label %[[CASE1:.+]]
72     // LAMBDA: {{.+}}, label %[[CASE2:.+]]
73     // LAMBDA: ]
74     // LAMBDA: [[CASE1]]:
75     // LAMBDA-DAG: [[SIVAR_VAL:%.+]] = load{{.+}}, {{.+}} [[SIVAR_REF]],
76     // LAMBDA-DAG: [[SIVAR_PRIV_VAL:%.+]] = load{{.+}}, {{.+}} [[SIVAR_PRIV]],
77     // LAMBDA-DAG: [[SIVAR_INC:%.+]] = add{{.+}} [[SIVAR_VAL]], [[SIVAR_PRIV_VAL]]
78     // LAMBDA: store{{.+}} [[SIVAR_INC]], {{.+}} [[SIVAR_REF]],
79     // LAMBDA: call void @__kmpc_end_reduce({{.+}}, {{.+}}, {{.+}} [[RED_VAR]])
80     // LAMBDA: br
81     // LAMBDA: [[CASE2]]:
82     // LAMBDA-DAG: [[SIVAR_PRIV_VAL:%.+]] = load{{.+}}, {{.+}} [[SIVAR_PRIV]],
83     // LAMBDA-DAG: [[ATOMIC_RES:%.+]] = atomicrmw add{{.+}} [[SIVAR_REF]], {{.+}} [[SIVAR_PRIV_VAL]]
84     // LAMBDA: call void @__kmpc_end_reduce({{.+}}, {{.+}}, {{.+}} [[RED_VAR]])
85     // LAMBDA: br
86     sivar += i;
87 
88     [&]() {
89       // LAMBDA: define {{.+}} void [[INNER_LAMBDA]](%{{.+}}* [[ARG_PTR:%.+]])
90       // LAMBDA: store %{{.+}}* [[ARG_PTR]], %{{.+}}** [[ARG_PTR_REF:%.+]],
91 
92       sivar += 4;
93       // LAMBDA: [[ARG_PTR:%.+]] = load %{{.+}}*, %{{.+}}** [[ARG_PTR_REF]]
94 
95       // LAMBDA: [[SIVAR_PTR_REF:%.+]] = getelementptr inbounds %{{.+}}, %{{.+}}* [[ARG_PTR]], i{{[0-9]+}} 0, i{{[0-9]+}} 0
96       // LAMBDA: [[SIVAR_REF:%.+]] = load i{{[0-9]+}}*, i{{[0-9]+}}** [[SIVAR_PTR_REF]]
97       // LAMBDA: [[SIVAR_VAL:%.+]] = load i{{[0-9]+}}, i{{[0-9]+}}* [[SIVAR_REF]]
98       // LAMBDA: [[SIVAR_INC:%.+]] = add{{.+}} [[SIVAR_VAL]], 4
99       // LAMBDA: store i{{[0-9]+}} [[SIVAR_INC]], i{{[0-9]+}}* [[SIVAR_REF]]
100     }();
101   }
102   }();
103   return 0;
104 #else
105 #pragma omp target teams distribute reduction(+: sivar)
106   for (int i = 0; i < 2; ++i) {
107     sivar += i;
108   }
109   return tmain<int>();
110 #endif
111 }
112 
113 // CHECK: [[RED_VAR:@.+]] = common global [8 x {{.+}}] zeroinitializer
114 
115 // CHECK: define {{.*}}i{{[0-9]+}} @main()
116 // CHECK: call i32 @__tgt_target_teams(i64 -1, i8* @{{[^,]+}}, i32 1, i8** %{{[^,]+}}, i8** %{{[^,]+}}, i{{64|32}}* {{.+}}@{{[^,]+}}, i32 0, i32 0), i64* {{.+}}@{{[^,]+}}, i32 0, i32 0), i32 0, i32 0)
117 // CHECK: call void @[[OFFL1:.+]](i{{64|32}} %{{.+}})
118 // CHECK: {{%.+}} = call{{.*}} i32 @[[TMAIN_INT:.+]]()
119 // CHECK:  ret
120 
121 // CHECK: define{{.*}} void @[[OFFL1]](i{{64|32}} [[SIVAR_ARG:%.+]])
122 // CHECK: [[SIVAR_ADDR:%.+]] = alloca i{{.+}},
123 // CHECK: [[SIVAR_CASTED:%.+]] = alloca i{{.+}},
124 // CHECK: store{{.+}} [[SIVAR_ARG]], {{.+}} [[SIVAR_ADDR]],
125 // CHECK-64: [[SIVAR_LOAD:%.+]] = load i64, i64* [[SIVAR_CASTED]],
126 // CHECK-32: [[SIVAR_LOAD:%.+]] = load i32, i32* [[SIVAR_CASTED]],
127 // CHECK: call void {{.+}} @__kmpc_fork_teams({{.+}}, i32 1, {{.+}} @[[OUTL1:.+]] to {{.+}}, {{.+}} [[SIVAR_LOAD]])
128 // CHECK: ret void
129 
130 // CHECK: define internal void @[[OUTL1]]({{.+}}, {{.+}}, {{.+}} [[SIVAR_ARG:%.+]])
131 // Skip global and bound tid vars
132 // CHECK: {{.+}} = alloca i32*,
133 // CHECK: {{.+}} = alloca i32*,
134 // CHECK: [[SIVAR_ADDR:%.+]] = alloca i{{.+}},
135 // CHECK: [[SIVAR_PRIV:%.+]] = alloca i{{.+}},
136 // CHECK: [[RED_LIST:%.+]] = alloca [1 x {{.+}}],
137 // CHECK: store{{.+}} [[SIVAR_ARG]], {{.+}} [[SIVAR_ADDR]],
138 // CHECK-64: [[SIVAR_REF:%.+]] = bitcast i64* [[SIVAR_ADDR]] to i32*
139 // CHECK: store{{.+}} 0, {{.+}} [[SIVAR_PRIV]],
140 
141 // CHECK: call void @__kmpc_for_static_init_4(
142 // CHECK: store{{.+}}, {{.+}} [[SIVAR_PRIV]],
143 // CHECK: call void @__kmpc_for_static_fini(
144 // CHECK: [[RED_LIST_GEP:%.+]] = getelementptr{{.+}} [[RED_LIST]],
145 // CHECK: [[SIVAR_PRIV_CAST:%.+]] = bitcast{{.+}} [[SIVAR_PRIV]] to
146 // CHECK: store{{.+}} [[SIVAR_PRIV_CAST]], {{.+}} [[RED_LIST_GEP]],
147 // CHECK: [[RED_LIST_BCAST:%.+]] = bitcast{{.+}} [[RED_LIST]] to
148 // CHECK: [[K_RED_RET:%.+]] = call{{.+}} @__kmpc_reduce({{.+}}, {{.+}}, {{.+}}, {{.+}}, {{.+}} [[RED_LIST_BCAST]], {{.+}} [[RED_FUN:@.+]], {{.+}} [[RED_VAR]])
149 // CHECK: switch{{.+}} [[K_RED_RET]], label{{.+}} [
150 // CHECK: {{.+}}, label %[[CASE1:.+]]
151 // CHECK: {{.+}}, label %[[CASE2:.+]]
152 // CHECK: ]
153 // CHECK: [[CASE1]]:
154 // CHECK-64-DAG: [[SIVAR_VAL:%.+]] = load{{.+}}, {{.+}} [[SIVAR_REF]],
155 // CHECK-32-DAG: [[SIVAR_VAL:%.+]] = load{{.+}}, {{.+}} [[SIVAR_ADDR]],
156 // CHECK-DAG: [[SIVAR_PRIV_VAL:%.+]] = load{{.+}}, {{.+}} [[SIVAR_PRIV]],
157 // CHECK-DAG: [[SIVAR_INC:%.+]] = add{{.+}} [[SIVAR_VAL]], [[SIVAR_PRIV_VAL]]
158 // CHECK-64: store{{.+}} [[SIVAR_INC]], {{.+}} [[SIVAR_REF]],
159 // CHECK-32: store{{.+}} [[SIVAR_INC]], {{.+}} [[SIVAR_ADDR]],
160 // CHECK: call void @__kmpc_end_reduce({{.+}}, {{.+}}, {{.+}} [[RED_VAR]])
161 // CHECK: br
162 // CHECK: [[CASE2]]:
163 // CHECK-DAG: [[SIVAR_PRIV_VAL:%.+]] = load{{.+}}, {{.+}} [[SIVAR_PRIV]],
164 // CHECK-64-DAG: [[ATOMIC_RES:%.+]] = atomicrmw add{{.+}} [[SIVAR_REF]], {{.+}} [[SIVAR_PRIV_VAL]]
165 // CHECK-32-DAG: [[ATOMIC_RES:%.+]] = atomicrmw add{{.+}} [[SIVAR_ADDR]], {{.+}} [[SIVAR_PRIV_VAL]]
166 // CHECK: call void @__kmpc_end_reduce({{.+}}, {{.+}}, {{.+}} [[RED_VAR]])
167 // CHECK: br
168 
169 
170 // CHECK: define{{.*}} i{{[0-9]+}} @[[TMAIN_INT]]()
171 // CHECK: call i32 @__tgt_target_teams(i64 -1, i8* @{{[^,]+}}, i32 1,
172 // CHECK: call void @[[TOFFL1:.+]]({{.+}})
173 // CHECK:  ret
174 
175 // CHECK: define{{.*}} void @[[TOFFL1]](i{{64|32}} [[TVAR_ARG:%.+]])
176 // CHECK: [[TVAR_ADDR:%.+]] = alloca i{{.+}},
177 // CHECK: [[TVAR_CASTED_ADDR:%.+]] = alloca i{{.+}},
178 // CHECK: store{{.+}} [[TVAR_ARG]], {{.+}} [[TVAR_ADDR]],
179 // CHECK-64: [[TVAR_CONV:%.+]] = bitcast{{.+}} [[TVAR_ADDR]] to
180 // CHECK-64: [[TVAR:%.+]] = load i64, i64* [[TVAR_CASTED_ADDR]],
181 // CHECK-32: [[TVAR:%.+]] = load i32, i32* [[TVAR_CASTED_ADDR]],
182 // CHECK: call void {{.+}} @__kmpc_fork_teams({{.+}}, i32 1, {{.+}} @[[TOUTL1:.+]] to {{.+}}, {{.+}} [[TVAR]])
183 // CHECK: ret void
184 
185 // CHECK: define internal void @[[TOUTL1]]({{.+}}, {{.+}}, {{.+}} [[TVAR_ARG:%.+]])
186 // Skip global and bound tid vars
187 // CHECK: {{.+}} = alloca i32*,
188 // CHECK: {{.+}} = alloca i32*,
189 // CHECK: [[TVAR_ADDR:%.+]] = alloca i{{.+}},
190 // CHECK: [[TVAR_PRIV:%.+]] = alloca i{{.+}},
191 // CHECK: [[RED_LIST:%.+]] = alloca [1 x {{.+}}],
192 // CHECK: store{{.+}} [[TVAR_ARG]], {{.+}} [[TVAR_ADDR]],
193 // CHECK-64: [[TVAR_REF:%.+]] = bitcast i64* [[TVAR_ADDR]] to i32*
194 // CHECK: store{{.+}} 0, {{.+}} [[TVAR_PRIV]],
195 
196 // CHECK: call void @__kmpc_for_static_init_4(
197 // CHECK: store{{.+}}, {{.+}} [[TVAR_PRIV]],
198 // CHECK: call void @__kmpc_for_static_fini(
199 // CHECK: [[RED_LIST_GEP:%.+]] = getelementptr{{.+}} [[RED_LIST]],
200 // CHECK: [[TVAR_PRIV_CAST:%.+]] = bitcast{{.+}} [[TVAR_PRIV]] to
201 // CHECK: store{{.+}} [[TVAR_PRIV_CAST]], {{.+}} [[RED_LIST_GEP]],
202 // CHECK: [[RED_LIST_BCAST:%.+]] = bitcast{{.+}} [[RED_LIST]] to
203 // CHECK: [[K_RED_RET:%.+]] = call{{.+}} @__kmpc_reduce({{.+}}, {{.+}}, {{.+}}, {{.+}}, {{.+}} [[RED_LIST_BCAST]], {{.+}} [[RED_FUN:@.+]], {{.+}} [[RED_VAR]])
204 // CHECK: switch{{.+}} [[K_RED_RET]], label{{.+}} [
205 // CHECK: {{.+}}, label %[[CASE1:.+]]
206 // CHECK: {{.+}}, label %[[CASE2:.+]]
207 // CHECK: ]
208 // CHECK: [[CASE1]]:
209 // CHECK-64-DAG: [[TVAR_VAL:%.+]] = load{{.+}}, {{.+}} [[TVAR_REF]],
210 // CHECK-32-DAG: [[TVAR_VAL:%.+]] = load{{.+}}, {{.+}} [[TVAR_ADDR]],
211 // CHECK-DAG: [[TVAR_PRIV_VAL:%.+]] = load{{.+}}, {{.+}} [[TVAR_PRIV]],
212 // CHECK-DAG: [[TVAR_INC:%.+]] = add{{.+}} [[TVAR_VAL]], [[TVAR_PRIV_VAL]]
213 // CHECK-64: store{{.+}} [[TVAR_INC]], {{.+}} [[TVAR_REF]],
214 // CHECK-32: store{{.+}} [[TVAR_INC]], {{.+}} [[TVAR_ADDR]],
215 // CHECK: call void @__kmpc_end_reduce({{.+}}, {{.+}}, {{.+}} [[RED_VAR]])
216 // CHECK: br
217 // CHECK: [[CASE2]]:
218 // CHECK-DAG: [[TVAR_PRIV_VAL:%.+]] = load{{.+}}, {{.+}} [[TVAR_PRIV]],
219 // CHECK-64-DAG: [[ATOMIC_RES:%.+]] = atomicrmw add{{.+}} [[TVAR_REF]], {{.+}} [[TVAR_PRIV_VAL]]
220 // CHECK-32-DAG: [[ATOMIC_RES:%.+]] = atomicrmw add{{.+}} [[TVAR_ADDR]], {{.+}} [[TVAR_PRIV_VAL]]
221 // CHECK: call void @__kmpc_end_reduce({{.+}}, {{.+}}, {{.+}} [[RED_VAR]])
222 // CHECK: br
223 
224 #endif
225