1 // 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
2 // 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
3 // RUN: %clang_cc1 -DLAMBDA -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 LAMBDA --check-prefix LAMBDA-64
4 // RUN: %clang_cc1 -DLAMBDA -verify -fopenmp -x c++ -std=c++11 -triple i386-unknown-unknown -fopenmp-targets=i386-pc-linux-gnu -emit-llvm %s -o - | FileCheck %s --check-prefix LAMBDA --check-prefix LAMBDA-32
5 // RUN: %clang_cc1 -DLAMBDA -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 -DLAMBDA -fopenmp -x c++ -std=c++11 -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 LAMBDA --check-prefix LAMBDA-32
7 
8 // RUN: %clang_cc1  -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 CHECK --check-prefix CHECK-64
9 // RUN: %clang_cc1  -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  -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
11 // RUN: %clang_cc1  -verify -fopenmp -x c++ -std=c++11 -triple i386-unknown-unknown -fopenmp-targets=i386-pc-linux-gnu -emit-llvm %s -o - | FileCheck %s --check-prefix CHECK --check-prefix CHECK-32
12 // RUN: %clang_cc1  -fopenmp -x c++ -std=c++11 -triple i386-unknown-unknown -fopenmp-targets=i386-pc-linux-gnu -emit-pch -o %t %s
13 // RUN: %clang_cc1  -fopenmp -x c++ -std=c++11 -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
14 // expected-no-diagnostics
15 #ifndef HEADER
16 #define HEADER
17 
18 template <class T>
19 struct S {
20   T f;
21   S(T a) : f(a) {}
22   S() : f() {}
23   operator T() { return T(); }
24   ~S() {}
25 };
26 
27 // CHECK: [[S_FLOAT_TY:%.+]] = type { float }
28 // CHECK: [[S_INT_TY:%.+]] = type { i{{[0-9]+}} }
29 template <typename T>
30 T tmain() {
31   S<T> test;
32   T t_var = T();
33   T vec[] = {1, 2};
34   S<T> s_arr[] = {1, 2};
35   S<T> &var = test;
36   #pragma omp target
37   #pragma omp teams
38 #pragma omp distribute private(t_var, vec, s_arr, s_arr, var, var)
39   for (int i = 0; i < 2; ++i) {
40     vec[i] = t_var;
41     s_arr[i] = var;
42   }
43   return T();
44 }
45 
46 int main() {
47   static int svar;
48   volatile double g;
49   volatile double &g1 = g;
50 
51   #ifdef LAMBDA
52   // LAMBDA-LABEL: @main
53   // LAMBDA: call{{.*}} void [[OUTER_LAMBDA:@.+]](
54   [&]() {
55     static float sfvar;
56     // LAMBDA: define{{.*}} internal{{.*}} void [[OUTER_LAMBDA]](
57     // LAMBDA: call i{{[0-9]+}} @__tgt_target_teams(
58     // LAMBDA: call void [[OFFLOADING_FUN:@.+]](
59 
60     // LAMBDA: define{{.+}} void [[OFFLOADING_FUN]]()
61     // LAMBDA: call {{.*}}void {{.+}} @__kmpc_fork_teams({{.+}}, i32 0, {{.+}}* [[OMP_OUTLINED:@.+]] to {{.+}})
62     #pragma omp target
63     #pragma omp teams
64 #pragma omp distribute private(g, g1, svar, sfvar)
65     for (int i = 0; i < 2; ++i) {
66       // LAMBDA: define{{.*}} internal{{.*}} void [[OMP_OUTLINED]](i32* noalias %{{.+}}, i32* noalias %{{.+}})
67       // LAMBDA: [[G_PRIVATE_ADDR:%.+]] = alloca double,
68       // LAMBDA: [[G1_PRIVATE_ADDR:%.+]] = alloca double,
69       // LAMBDA: [[TMP_PRIVATE_ADDR:%.+]] = alloca double*,
70       // LAMBDA: [[SVAR_PRIVATE_ADDR:%.+]] = alloca i{{[0-9]+}},
71       // LAMBDA: [[SFVAR_PRIVATE_ADDR:%.+]] = alloca float,
72       // LAMBDA: store double* [[G1_PRIVATE_ADDR]], double** [[TMP_PRIVATE_ADDR]],
73       g = 1;
74       g1 = 1;
75       svar = 3;
76       sfvar = 4.0;
77       // LAMBDA: call {{.*}}void @__kmpc_for_static_init_4(
78       // LAMBDA: store double 1.0{{.+}}, double* [[G_PRIVATE_ADDR]],
79       // LAMBDA: store i{{[0-9]+}} 3, i{{[0-9]+}}* [[SVAR_PRIVATE_ADDR]],
80       // LAMBDA: store float 4.0{{.+}}, float* [[SFVAR_PRIVATE_ADDR]],
81       // LAMBDA: [[G_PRIVATE_ADDR_REF:%.+]] = getelementptr inbounds %{{.+}}, %{{.+}}* [[ARG:%.+]], i{{[0-9]+}} 0, i{{[0-9]+}} 0
82       // LAMBDA: store double* [[G_PRIVATE_ADDR]], double** [[G_PRIVATE_ADDR_REF]],
83       // LAMBDA: [[TMP_PRIVATE_ADDR_REF:%.+]] = getelementptr inbounds %{{.+}}, %{{.+}}* [[ARG:%.+]], i{{[0-9]+}} 0, i{{[0-9]+}} 1
84       // LAMBDA: [[G1_PRIVATE_ADDR_FROM_TMP:%.+]] = load double*, double** [[TMP_PRIVATE_ADDR]],
85       // LAMBDA: store double* [[G1_PRIVATE_ADDR_FROM_TMP]], double** [[TMP_PRIVATE_ADDR_REF]],
86       // LAMBDA: [[SVAR_PRIVATE_ADDR_REF:%.+]] = getelementptr inbounds %{{.+}}, %{{.+}}* [[ARG:%.+]], i{{[0-9]+}} 0, i{{[0-9]+}} 2
87       // LAMBDA: store i{{[0-9]+}}* [[SVAR_PRIVATE_ADDR]], i{{[0-9]+}}** [[SVAR_PRIVATE_ADDR_REF]]
88       // LAMBDA: [[SFVAR_PRIVATE_ADDR_REF:%.+]] = getelementptr inbounds %{{.+}}, %{{.+}}* [[ARG:%.+]], i{{[0-9]+}} 0, i{{[0-9]+}} 3
89       // LAMBDA: store float* [[SFVAR_PRIVATE_ADDR]], float** [[SFVAR_PRIVATE_ADDR_REF]]
90       // LAMBDA: call{{.*}} void [[INNER_LAMBDA:@.+]](%{{.+}}* [[ARG]])
91       // LAMBDA: call {{.*}}void @__kmpc_for_static_fini(
92       [&]() {
93 	// LAMBDA: define {{.+}} void [[INNER_LAMBDA]](%{{.+}}* [[ARG_PTR:%.+]])
94 	// LAMBDA: store %{{.+}}* [[ARG_PTR]], %{{.+}}** [[ARG_PTR_REF:%.+]],
95 	g = 2;
96 	g1 = 2;
97 	svar = 4;
98 	sfvar = 8.0;
99 	// LAMBDA: [[ARG_PTR:%.+]] = load %{{.+}}*, %{{.+}}** [[ARG_PTR_REF]]
100 	// LAMBDA: [[G_PTR_REF:%.+]] = getelementptr inbounds %{{.+}}, %{{.+}}* [[ARG_PTR]], i{{[0-9]+}} 0, i{{[0-9]+}} 0
101 	// LAMBDA: [[G_REF:%.+]] = load double*, double** [[G_PTR_REF]]
102 	// LAMBDA: store double 2.0{{.+}}, double* [[G_REF]]
103 
104 	// LAMBDA: [[TMP_PTR_REF:%.+]] = getelementptr inbounds %{{.+}}, %{{.+}}* [[ARG_PTR]], i{{[0-9]+}} 0, i{{[0-9]+}} 1
105 	// LAMBDA: [[G1_REF:%.+]] = load double*, double** [[TMP_PTR_REF]]
106 	// LAMBDA: store double 2.0{{.+}}, double* [[G1_REF]],
107 	// LAMBDA: [[SVAR_PTR_REF:%.+]] = getelementptr inbounds %{{.+}}, %{{.+}}* [[ARG_PTR]], i{{[0-9]+}} 0, i{{[0-9]+}} 2
108 	// LAMBDA: [[SVAR_REF:%.+]] = load i{{[0-9]+}}*, i{{[0-9]+}}** [[SVAR_PTR_REF]]
109 	// LAMBDA: store i{{[0-9]+}} 4, i{{[0-9]+}}* [[SVAR_REF]]
110 	// LAMBDA: [[SFVAR_PTR_REF:%.+]] = getelementptr inbounds %{{.+}}, %{{.+}}* [[ARG_PTR]], i{{[0-9]+}} 0, i{{[0-9]+}} 3
111 	// LAMBDA: [[SFVAR_REF:%.+]] = load float*, float** [[SFVAR_PTR_REF]]
112 	// LAMBDA: store float 8.0{{.+}}, float* [[SFVAR_REF]]
113       }();
114     }
115   }();
116   return 0;
117   #else
118   S<float> test;
119   int t_var = 0;
120   int vec[] = {1, 2};
121   S<float> s_arr[] = {1, 2};
122   S<float> &var = test;
123 
124   #pragma omp target
125   #pragma omp teams
126 #pragma omp distribute private(t_var, vec, s_arr, s_arr, var, var, svar)
127   for (int i = 0; i < 2; ++i) {
128     vec[i] = t_var;
129     s_arr[i] = var;
130   }
131   int i;
132 
133   #pragma omp target
134   #pragma omp teams
135 #pragma omp distribute private(i)
136   for (i = 0; i < 2; ++i) {
137     ;
138   }
139   return tmain<int>();
140   #endif
141 }
142 
143 // CHECK: define{{.*}} i{{[0-9]+}} @main()
144 // CHECK: [[TEST:%.+]] = alloca [[S_FLOAT_TY]],
145 // CHECK: call {{.*}} [[S_FLOAT_TY_DEF_CONSTR:@.+]]([[S_FLOAT_TY]]* [[TEST]])
146 // CHECK: call i{{[0-9]+}} @__tgt_target_teams(
147 // CHECK: call void [[OFFLOAD_FUN:@.+]](
148 // CHECK: ret
149 
150 // CHECK: define{{.+}} [[OFFLOAD_FUN]]()
151 // CHECK: call void (%{{.+}}*, i{{[0-9]+}}, void (i{{[0-9]+}}*, i{{[0-9]+}}*, ...)*, ...) @__kmpc_fork_teams(%{{.+}}* @{{.+}}, i{{[0-9]+}} 0, void (i{{[0-9]+}}*, i{{[0-9]+}}*, ...)* bitcast (void (i{{[0-9]+}}*, i{{[0-9]+}}*)* [[OMP_OUTLINED:@.+]] to void
152 // CHECK: ret
153 //
154 // CHECK: define internal void [[OMP_OUTLINED]](i{{[0-9]+}}* noalias [[GTID_ADDR:%.+]], i{{[0-9]+}}* noalias %{{.+}})
155 // CHECK: [[T_VAR_PRIV:%.+]] = alloca i{{[0-9]+}},
156 // CHECK: [[VEC_PRIV:%.+]] = alloca [2 x i{{[0-9]+}}],
157 // CHECK: [[S_ARR_PRIV:%.+]] = alloca [2 x [[S_FLOAT_TY]]],
158 // CHECK-NOT: alloca [2 x [[S_FLOAT_TY]]],
159 // CHECK: [[VAR_PRIV:%.+]] = alloca [[S_FLOAT_TY]],
160 // CHECK-NOT: alloca [[S_FLOAT_TY]],
161 // CHECK: [[S_VAR_PRIV:%.+]] = alloca i{{[0-9]+}},
162 // CHECK: store i{{[0-9]+}}* [[GTID_ADDR]], i{{[0-9]+}}** [[GTID_ADDR_REF:%.+]]
163 // CHECK-NOT: [[T_VAR_PRIV]]
164 // CHECK-NOT: [[VEC_PRIV]]
165 // CHECK: {{.+}}:
166 // CHECK: [[S_ARR_PRIV_ITEM:%.+]] = phi [[S_FLOAT_TY]]*
167 // CHECK: call {{.*}} [[S_FLOAT_TY_DEF_CONSTR]]([[S_FLOAT_TY]]* [[S_ARR_PRIV_ITEM]])
168 // CHECK-NOT: [[T_VAR_PRIV]]
169 // CHECK-NOT: [[VEC_PRIV]]
170 // CHECK: call {{.*}} [[S_FLOAT_TY_DEF_CONSTR]]([[S_FLOAT_TY]]* [[VAR_PRIV]])
171 // CHECK: call void @__kmpc_for_static_init_4(
172 // CHECK: call void @__kmpc_for_static_fini(
173 // CHECK: ret void
174 
175 // CHECK: define{{.*}} i{{[0-9]+}} [[TMAIN_INT:@.+]]()
176 // CHECK: [[TEST:%.+]] = alloca [[S_INT_TY]],
177 // CHECK: call {{.*}} [[S_INT_TY_DEF_CONSTR:@.+]]([[S_INT_TY]]* [[TEST]])
178 // CHECK: call i{{[0-9]+}} @__tgt_target_teams(
179 // CHECK: call void [[OFFLOAD_FUN_1:@.+]](
180 // CHECK: ret
181 
182 
183 // CHECK: define internal void [[OFFLOAD_FUN_1]]()
184 // CHECK: call void (%{{.+}}*, i{{[0-9]+}}, void (i{{[0-9]+}}*, i{{[0-9]+}}*, ...)*, ...) @__kmpc_fork_teams(%{{.+}}* @{{.+}}, i{{[0-9]+}} 0, void (i{{[0-9]+}}*, i{{[0-9]+}}*, ...)* bitcast (void (i{{[0-9]+}}*, i{{[0-9]+}}*)* [[OMP_OUTLINED_1:@.+]] to void
185 // CHECK: ret
186 //
187 // CHECK: define internal void [[OMP_OUTLINED_1]](i{{[0-9]+}}* noalias [[GTID_ADDR:%.+]], i{{[0-9]+}}* noalias %{{.+}})
188 // CHECK: [[T_VAR_PRIV:%.+]] = alloca i{{[0-9]+}},
189 // CHECK: [[VEC_PRIV:%.+]] = alloca [2 x i{{[0-9]+}}],
190 // CHECK: [[S_ARR_PRIV:%.+]] = alloca [2 x [[S_INT_TY]]],
191 // CHECK-NOT: alloca [2 x [[S_INT_TY]]],
192 // CHECK: [[VAR_PRIV:%.+]] = alloca [[S_INT_TY]],
193 // CHECK-NOT: alloca [[S_INT_TY]],
194 // CHECK: store i{{[0-9]+}}* [[GTID_ADDR]], i{{[0-9]+}}** [[GTID_ADDR_REF:%.+]]
195 // CHECK-NOT: [[T_VAR_PRIV]]
196 // CHECK-NOT: [[VEC_PRIV]]
197 // CHECK: {{.+}}:
198 // CHECK: [[S_ARR_PRIV_ITEM:%.+]] = phi [[S_INT_TY]]*
199 // CHECK: call {{.*}} [[S_INT_TY_DEF_CONSTR]]([[S_INT_TY]]* [[S_ARR_PRIV_ITEM]])
200 // CHECK-NOT: [[T_VAR_PRIV]]
201 // CHECK-NOT: [[VEC_PRIV]]
202 // CHECK: call {{.*}} [[S_INT_TY_DEF_CONSTR]]([[S_INT_TY]]* [[VAR_PRIV]])
203 // CHECK: call void @__kmpc_for_static_init_4(
204 // CHECK: call void @__kmpc_for_static_fini(
205 // CHECK: ret void
206 #endif
207