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 struct St {
17   int a, b;
18   St() : a(0), b(0) {}
19   St(const St &st) : a(st.a + st.b), b(0) {}
20   ~St() {}
21 };
22 
23 volatile int g = 1212;
24 volatile int &g1 = g;
25 
26 template <class T>
27 struct S {
28   T f;
29   S(T a) : f(a + g) {}
30   S() : f(g) {}
31   S(const S &s, St t = St()) : f(s.f + t.a) {}
32   operator T() { return T(); }
33   ~S() {}
34 };
35 
36 // CHECK-DAG: [[S_FLOAT_TY:%.+]] = type { float }
37 // CHECK-DAG: [[S_INT_TY:%.+]] = type { i{{[0-9]+}} }
38 
39 template <typename T>
40 T tmain() {
41   S<T> test;
42   T t_var = T();
43   T vec[] = {1, 2};
44   S<T> s_arr[] = {1, 2};
45   S<T> &var = test;
46 #pragma omp target
47 #pragma omp teams distribute private(t_var, vec, s_arr, var)
48   for (int i = 0; i < 2; ++i) {
49     vec[i] = t_var;
50     s_arr[i] = var;
51   }
52   return T();
53 }
54 
55 // CHECK-DAG: [[TEST:@.+]] = global [[S_FLOAT_TY]] zeroinitializer,
56 S<float> test;
57 // CHECK-DAG: [[T_VAR:@.+]] = global i{{[0-9]+}} 333,
58 int t_var = 333;
59 // CHECK-DAG: [[VEC:@.+]] = global [2 x i{{[0-9]+}}] [i{{[0-9]+}} 1, i{{[0-9]+}} 2],
60 int vec[] = {1, 2};
61 // CHECK-DAG: [[S_ARR:@.+]] = global [2 x [[S_FLOAT_TY]]] zeroinitializer,
62 S<float> s_arr[] = {1, 2};
63 // CHECK-DAG: [[VAR:@.+]] = global [[S_FLOAT_TY]] zeroinitializer,
64 S<float> var(3);
65 // CHECK-DAG: [[SIVAR:@.+]] = internal global i{{[0-9]+}} 0,
66 
67 int main() {
68   static int sivar;
69 #ifdef LAMBDA
70   // LAMBDA: [[G:@.+]] = global i{{[0-9]+}} 1212,
71   // LAMBDA-LABEL: @main
72   // LAMBDA: call void [[OUTER_LAMBDA:@.+]](
73   [&]() {
74     // LAMBDA: define{{.*}} internal{{.*}} void [[OUTER_LAMBDA]](
75     // LAMBDA: call i32 @__tgt_target(i32 -1, i8* @{{[^,]+}}, i32 2, i8** %{{[^,]+}}, i8** %{{[^,]+}}, i{{64|32}}* {{.+}}@{{[^,]+}}, i32 0, i32 0), i32* {{.+}}@{{[^,]+}}, i32 0, i32 0))
76     // LAMBDA: call void @[[LOFFL1:.+]](
77     // LAMBDA:  ret
78 #pragma omp target
79 #pragma omp teams distribute private(g, g1, sivar)
80   for (int i = 0; i < 2; ++i) {
81     // LAMBDA: define{{.*}} internal{{.*}} void @[[LOFFL1]](i{{64|32}} {{%.+}}, i{{64|32}} {{%.+}})
82     // LAMBDA: call void {{.+}} @__kmpc_fork_teams({{.+}}, i32 0, {{.+}} @[[LOUTL1:.+]] to {{.+}})
83     // LAMBDA: ret void
84 
85     // LAMBDA: define internal void @[[LOUTL1]]({{.+}})
86     // Skip global, bound tid and loop vars
87     // LAMBDA: {{.+}} = alloca i32*,
88     // LAMBDA: {{.+}} = alloca i32*,
89     // LAMBDA: alloca i32,
90     // LAMBDA: alloca i32,
91     // LAMBDA: alloca i32,
92     // LAMBDA: alloca i32,
93     // LAMBDA: alloca i32,
94     // LAMBDA: [[G_PRIV:%.+]] = alloca i{{[0-9]+}},
95     // LAMBDA: [[G1_PRIV:%.+]] = alloca i{{[0-9]+}}
96     // LAMBDA: [[TMP:%.+]] = alloca i{{[0-9]+}}*,
97     // LAMBDA: [[SIVAR_PRIV:%.+]] = alloca i{{[0-9]+}},
98     // LAMBDA: store{{.+}} [[G1_PRIV]], {{.+}} [[TMP]],
99 
100     g = 1;
101     g1 = 1;
102     sivar = 2;
103     // LAMBDA: call void @__kmpc_for_static_init_4(
104     // LAMBDA-DAG: store{{.+}} 1, {{.+}} [[G_PRIV]],
105     // LAMBDA-DAG: store{{.+}} 2, {{.+}} [[SIVAR_PRIV]],
106     // LAMBDA-DAG: [[G1_REF:%.+]] = load{{.+}}, {{.+}} [[TMP]],
107     // LAMBDA-DAG: store{{.+}} 1, {{.+}} [[G1_REF]],
108     // LAMBDA: call void [[INNER_LAMBDA:@.+]](
109     // LAMBDA: call void @__kmpc_for_static_fini(
110     [&]() {
111       // LAMBDA: define {{.+}} void [[INNER_LAMBDA]](%{{.+}}* [[ARG_PTR:%.+]])
112       // LAMBDA: store %{{.+}}* [[ARG_PTR]], %{{.+}}** [[ARG_PTR_REF:%.+]],
113       g = 2;
114       g1 = 2;
115       sivar = 4;
116       // LAMBDA: [[ARG_PTR:%.+]] = load %{{.+}}*, %{{.+}}** [[ARG_PTR_REF]]
117 
118       // LAMBDA: [[G_PTR_REF:%.+]] = getelementptr inbounds %{{.+}}, %{{.+}}* [[ARG_PTR]], i{{[0-9]+}} 0, i{{[0-9]+}} 0
119       // LAMBDA: [[G_REF:%.+]] = load i{{[0-9]+}}*, i{{[0-9]+}}** [[G_PTR_REF]]
120       // LAMBDA: store i{{[0-9]+}} 2, i{{[0-9]+}}* [[G_REF]]
121       // LAMBDA: [[SIVAR_PTR_REF:%.+]] = getelementptr inbounds %{{.+}}, %{{.+}}* [[ARG_PTR]], i{{[0-9]+}} 0, i{{[0-9]+}} 1
122       // LAMBDA: [[SIVAR_REF:%.+]] = load i{{[0-9]+}}*, i{{[0-9]+}}** [[SIVAR_PTR_REF]]
123       // LAMBDA: store i{{[0-9]+}} 4, i{{[0-9]+}}* [[SIVAR_REF]]
124     }();
125   }
126   }();
127   return 0;
128 #else
129 #pragma omp target
130 #pragma omp teams distribute private(t_var, vec, s_arr, var, sivar)
131   for (int i = 0; i < 2; ++i) {
132     vec[i] = t_var;
133     s_arr[i] = var;
134     sivar += i;
135   }
136   return tmain<int>();
137 #endif
138 }
139 
140 // CHECK: define {{.*}}i{{[0-9]+}} @main()
141 // CHECK: call i32 @__tgt_target(i32 -1, i8* @{{[^,]+}}, i32 5, i8** %{{[^,]+}}, i8** %{{[^,]+}}, i{{64|32}}* {{.+}}@{{[^,]+}}, i32 0, i32 0), i32* {{.+}}@{{[^,]+}}, i32 0, i32 0))
142 // CHECK: call void @[[OFFL1:.+]](i{{64|32}} %{{.+}})
143 // CHECK: {{%.+}} = call{{.*}} i32 @[[TMAIN_INT:.+]]()
144 // CHECK:  ret
145 
146 // CHECK: define{{.*}} void @[[OFFL1]]({{.+}})
147 // CHECK: call void {{.+}} @__kmpc_fork_teams({{.+}}, i32 0, {{.+}} @[[OUTL1:.+]] to {{.+}})
148 // CHECK: ret void
149 
150 // CHECK: define internal void @[[OUTL1]]({{.+}})
151 // Skip global, bound tid and loop vars
152 // CHECK: {{.+}} = alloca i32*,
153 // CHECK: {{.+}} = alloca i32*,
154 // CHECK: {{.+}} = alloca i32,
155 // CHECK: {{.+}} = alloca i32,
156 // CHECK: {{.+}} = alloca i32,
157 // CHECK: {{.+}} = alloca i32,
158 // CHECK: {{.+}} = alloca i32,
159 // CHECK-DAG: [[T_VAR_PRIV:%.+]] = alloca i{{[0-9]+}},
160 // CHECK-DAG: [[VEC_PRIV:%.+]] = alloca [2 x i{{[0-9]+}}],
161 // CHECK-DAG: [[S_ARR_PRIV:%.+]] = alloca [2 x [[S_FLOAT_TY]]],
162 // CHECK-DAG: [[VAR_PRIV:%.+]] = alloca [[S_FLOAT_TY]],
163 // CHECK-DAG: [[SIVAR_PRIV:%.+]] = alloca i{{[0-9]+}},
164 // CHECK: alloca i32,
165 
166 // private(s_arr)
167 // CHECK-DAG: [[S_ARR_PRIV_BGN:%.+]] = getelementptr{{.*}} [2 x [[S_FLOAT_TY]]], [2 x [[S_FLOAT_TY]]]* [[S_ARR_PRIV]],
168 // CHECK-DAG: [[S_ARR_PTR_ALLOC:%.+]] = phi{{.+}} [ [[S_ARR_PRIV_BGN]], {{.+}} ], [ [[S_ARR_NEXT:%.+]], {{.+}} ]
169 // CHECK-DAG: call void @{{.+}}({{.+}} [[S_ARR_PTR_ALLOC]])
170 // CHECK-DAG: [[S_ARR_NEXT]] = getelementptr {{.+}} [[S_ARR_PTR_ALLOC]],
171 
172 // private(var)
173 // CHECK-DAG: call void @{{.+}}({{.+}} [[VAR_PRIV]])
174 
175 // CHECK: call void @__kmpc_for_static_init_4(
176 // CHECK-DAG: {{.+}} = {{.+}} [[T_VAR_PRIV]]
177 // CHECK-DAG: {{.+}} = {{.+}} [[VEC_PRIV]]
178 // CHECK-DAG: {{.+}} = {{.+}} [[S_ARR_PRIV]]
179 // CHECK-DAG: {{.+}} = {{.+}} [[VAR_PRIV]]
180 // CHECK-DAG: {{.+}} = {{.+}} [[SIVAR_PRIV]]
181 // CHECK: call void @__kmpc_for_static_fini(
182 // CHECK: ret void
183 
184 
185 // CHECK: define{{.*}} i{{[0-9]+}} @[[TMAIN_INT]]()
186 // CHECK: call i32 @__tgt_target(i32 -1, i8* @{{[^,]+}}, i32 0,
187 // CHECK: call void @[[TOFFL1:.+]]()
188 // CHECK:  ret
189 
190 // CHECK: define {{.*}}void @[[TOFFL1]]()
191 // CHECK: call void {{.+}} @__kmpc_fork_teams({{.+}}, i32 0, {{.+}} @[[TOUTL1:.+]] to {{.+}})
192 // CHECK: ret void
193 
194 // CHECK: define internal void @[[TOUTL1]]({{.+}})
195 // Skip global, bound tid and loop vars
196 // CHECK: {{.+}} = alloca i32*,
197 // CHECK: {{.+}} = alloca i32*,
198 // CHECK: alloca i{{[0-9]+}},
199 // CHECK: alloca i{{[0-9]+}},
200 // CHECK: alloca i{{[0-9]+}},
201 // CHECK: alloca i{{[0-9]+}},
202 // CHECK: alloca i{{[0-9]+}},
203 // CHECK: [[T_VAR_PRIV:%.+]] = alloca i{{[0-9]+}},
204 // CHECK: [[VEC_PRIV:%.+]] = alloca [2 x i{{[0-9]+}}],
205 // CHECK: [[S_ARR_PRIV:%.+]] = alloca [2 x [[S_INT_TY]]],
206 // CHECK: [[VAR_PRIV:%.+]] = alloca [[S_INT_TY]],
207 // CHECK: [[TMP:%.+]] = alloca [[S_INT_TY]]*,
208 // CHECK: alloca i32,
209 
210 // private(s_arr)
211 // CHECK-DAG: [[S_ARR_PRIV_BGN:%.+]] = getelementptr{{.*}} [2 x [[S_INT_TY]]], [2 x [[S_INT_TY]]]* [[S_ARR_PRIV]],
212 // CHECK-DAG: [[S_ARR_PTR_ALLOC:%.+]] = phi{{.+}} [ [[S_ARR_PRIV_BGN]], {{.+}} ], [ [[S_ARR_NEXT:%.+]], {{.+}} ]
213 // CHECK-DAG: call void @{{.+}}({{.+}} [[S_ARR_PTR_ALLOC]])
214 // CHECK-DAG: [[S_ARR_NEXT]] = getelementptr {{.+}} [[S_ARR_PTR_ALLOC]],
215 
216 // CHECK-DAG: [[S_ARR_PRIV_BGN:%.+]] = getelementptr{{.*}} [2 x [[S_INT_TY]]], [2 x [[S_INT_TY]]]* [[S_ARR_PRIV]],
217 // CHECK-DAG: [[S_ARR_PTR_ALLOC:%.+]] = phi{{.+}} [ [[S_ARR_PRIV_BGN]], {{.+}} ], [ [[S_ARR_NEXT:%.+]], {{.+}} ]
218 // CHECK-DAG: call void @{{.+}}({{.+}} [[S_ARR_PTR_ALLOC]])
219 // CHECK-DAG: [[S_ARR_NEXT]] = getelementptr {{.+}} [[S_ARR_PTR_ALLOC]],
220 
221 // private(var)
222 // CHECK-DAG: call void @{{.+}}({{.+}} [[VAR_PRIV]])
223 // CHECK-DAG: store{{.+}} [[VAR_PRIV]], {{.+}} [[TMP]]
224 
225 // CHECK: call void @__kmpc_for_static_init_4(
226 // CHECK-DAG: {{.+}} = {{.+}} [[T_VAR_PRIV]]
227 // CHECK-DAG: {{.+}} = {{.+}} [[VEC_PRIV]]
228 // CHECK-DAG: {{.+}} = {{.+}} [[S_ARR_PRIV]]
229 // CHECK-DAG: {{.+}} = {{.+}} [[TMP]]
230 // CHECK: call void @__kmpc_for_static_fini(
231 // CHECK: ret void
232 
233 #endif
234