1 // expected-no-diagnostics
2 #ifndef HEADER
3 #define HEADER
4 // Test host codegen.
5 // RUN: %clang_cc1 -DCK1 -verify -fopenmp -x c++ -triple powerpc64le-unknown-unknown -fopenmp-targets=powerpc64le-ibm-linux-gnu -emit-llvm %s -o - | FileCheck %s --check-prefix CK1 --check-prefix CK1-64
6 // RUN: %clang_cc1 -DCK1 -fopenmp -x c++ -std=c++11 -triple powerpc64le-unknown-unknown -fopenmp-targets=powerpc64le-ibm-linux-gnu -emit-pch -o %t %s
7 // RUN: %clang_cc1 -DCK1 -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 CK1 --check-prefix CK1-64
8 // RUN: %clang_cc1 -DCK1 -verify -fopenmp -x c++ -triple i386-unknown-unknown -fopenmp-targets=i386-pc-linux-gnu -emit-llvm %s -o - | FileCheck %s --check-prefix CK1 --check-prefix CK1-32
9 // RUN: %clang_cc1 -DCK1 -fopenmp -x c++ -std=c++11 -triple i386-unknown-unknown -fopenmp-targets=i386-pc-linux-gnu -emit-pch -o %t %s
10 // RUN: %clang_cc1 -DCK1 -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 CK1 --check-prefix CK1-32
11 #ifdef CK1
12 
13 int a[100];
14 
15 // CK1: define {{.*}}i32 @{{.+}}teams_argument_globali(
16 int teams_argument_global(int n){
17   int te = n / 128;
18   int th = 128;
19   // discard n_addr
20   // CK1: alloca i32,
21   // CK1: [[TE:%.+]] = alloca i32,
22   // CK1: [[TH:%.+]] = alloca i32,
23   // CK1: [[TE_CAST:%.+]] = alloca i{{32|64}},
24   // CK1: [[TH_CAST:%.+]] = alloca i{{32|64}},
25   // CK1: [[TE_PAR:%.+]] = load{{.+}}, {{.+}} [[TE_CAST]],
26   // CK1: [[TH_PAR:%.+]] = load{{.+}}, {{.+}} [[TH_CAST]],
27   // CK1: call i32 @__tgt_target_teams(i64 -1, i8* @{{[^,]+}}, i32 4, i8** %{{[^,]+}}, i8** %{{[^,]+}}, i{{64|32}}* {{.+}}@{{[^,]+}}, i32 0, i32 0), i64* {{.+}}@{{[^,]+}}, i32 0, i32 0), i32 {{.+}}, i32 {{.+}})
28 
29   // CK1: call void @[[OFFL1:.+]](i{{32|64}} [[TE_PAR]], i{{32|64}} [[TH_PAR]],
30   #pragma omp target
31   #pragma omp teams distribute parallel for num_teams(te), thread_limit(th)
32   for(int i = 0; i < n; i++) {
33     a[i] = 0;
34     #pragma omp cancel for
35   }
36 
37   // CK1: call i32 @__tgt_target_teams(i64 -1, i8* @{{[^,]+}}, i32 2, i8** %{{[^,]+}}, i8** %{{[^,]+}}, i{{64|32}}* {{.+}}@{{[^,]+}}, i32 0, i32 0), i64* {{.+}}@{{[^,]+}}, i32 0, i32 0), i32 0, i32 0)
38   // CK1: call void @[[OFFL2:.+]](i{{64|32}} %{{.+}})
39   #pragma omp target
40   {{{
41   #pragma omp teams distribute parallel for
42   for(int i = 0; i < n; i++) {
43     a[i] = 0;
44   }
45   }}}
46 
47   // outlined target regions
48   // CK1: define internal void @[[OFFL1]](i{{32|64}} [[TE_ARG:%.+]], i{{32|64}} [[TH_ARG:%.+]], i{{32|64}} {{.+}}, {{.+}})
49   // CK1: [[TE_ADDR:%.+]] = alloca i{{32|64}},
50   // CK1: [[TH_ADDR:%.+]] = alloca i{{32|64}},
51   // CK1: store{{.+}} [[TE_ARG]], {{.+}} [[TE_ADDR]],
52   // CK1: store{{.+}} [[TH_ARG]], {{.+}} [[TH_ADDR]],
53   // CK1-64: [[TE_CONV:%.+]] = bitcast{{.+}} [[TE_ADDR]] to
54   // CK1-64: [[TH_CONV:%.+]] = bitcast{{.+}} [[TH_ADDR]] to
55   // CK1-64: [[TE_VAL:%.+]] = load i32, i32* [[TE_CONV]],
56   // CK1-64: [[TH_VAL:%.+]] = load i32, i32* [[TH_CONV]],
57   // CK1-32: [[TE_VAL:%.+]] = load i32, i32* [[TE_ADDR]],
58   // CK1-32: [[TH_VAL:%.+]] = load i32, i32* [[TH_ADDR]],
59   // CK1: {{%.+}} = call i32 @__kmpc_push_num_teams({{.+}}, {{.+}}, i32 [[TE_VAL]], i32 [[TH_VAL]])
60   // CK1: call void {{.+}} @__kmpc_fork_teams({{.+}}, i32 2, {{.+}} @[[OUTL1:.+]] to {{.+}}, {{.+}}, {{.+}})
61   // CK1: ret void
62 
63   // CK1: define internal void @[[OUTL1]]({{.+}})
64   // CK1: call void @__kmpc_for_static_init_4(
65   // CK1: call void {{.+}} @__kmpc_fork_call(
66   // CK1: call void @__kmpc_for_static_fini(
67   // CK1: ret void
68 
69   // CK1: define internal void @[[OFFL2]]({{.+}}, {{.+}})
70   // CK1: call void {{.+}} @__kmpc_fork_teams({{.+}}, i32 2, {{.+}} @[[OUTL2:.+]] to {{.+}}, {{.+}}, {{.+}})
71   // CK1: ret void
72 
73   // CK1: define internal void @[[OUTL2]]({{.+}})
74   // CK1: call void @__kmpc_for_static_init_4(
75   // CK1: call void {{.+}} @__kmpc_fork_call(
76   // CK1: call void @__kmpc_for_static_fini(
77   // CK1: ret void
78 
79   return a[0];
80 }
81 
82 #endif // CK1
83 
84 // Test host codegen.
85 // RUN: %clang_cc1 -DCK2 -verify -fopenmp -x c++ -triple powerpc64le-unknown-unknown -fopenmp-targets=powerpc64le-ibm-linux-gnu -emit-llvm %s -o - | FileCheck %s --check-prefix CK2 --check-prefix CK2-64
86 // RUN: %clang_cc1 -DCK2 -fopenmp -x c++ -std=c++11 -triple powerpc64le-unknown-unknown -fopenmp-targets=powerpc64le-ibm-linux-gnu -emit-pch -o %t %s
87 // RUN: %clang_cc1 -DCK2 -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 CK2 --check-prefix CK2-64
88 // RUN: %clang_cc1 -DCK2 -verify -fopenmp -x c++ -triple i386-unknown-unknown -fopenmp-targets=i386-pc-linux-gnu -emit-llvm %s -o - | FileCheck %s --check-prefix CK2 --check-prefix CK2-32
89 // RUN: %clang_cc1 -DCK2 -fopenmp -x c++ -std=c++11 -triple i386-unknown-unknown -fopenmp-targets=i386-pc-linux-gnu -emit-pch -o %t %s
90 // RUN: %clang_cc1 -DCK2 -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 CK2 --check-prefix CK2-32
91 #ifdef CK2
92 
93 // CK2: define {{.*}}i32 @{{.+}}teams_local_argv(
94 int teams_local_arg(void) {
95   int n = 100;
96   int a[n];
97 
98   // CK2: call i32 @__tgt_target_teams(i64 -1, i8* @{{[^,]+}}, i32 3, i8** %{{[^,]+}}, i8** %{{[^,]+}}, i{{64|32}}* {{.+}}, i64* {{.+}}@{{[^,]+}}, i32 0, i32 0), i32 0, i32 0)
99   // CK2: call void @[[OFFL1:.+]](i{{64|32}} %{{.+}})
100   #pragma omp target
101   #pragma omp teams distribute parallel for
102   for(int i = 0; i < n; i++) {
103     a[i] = 0;
104   }
105 
106   // outlined target region
107   // CK2: define internal void @[[OFFL1]]({{.+}}, {{.+}})
108   // CK2: call void {{.+}} @__kmpc_fork_teams({{.+}}, i32 3, {{.+}} @[[OUTL1:.+]] to {{.+}}, {{.+}}, {{.+}})
109   // CK2: ret void
110 
111   // CK2: define internal void @[[OUTL1]]({{.+}})
112   // CK2: call void @__kmpc_for_static_init_4(
113   // CK2: call void {{.+}} @__kmpc_fork_call(
114   // CK2: call void @__kmpc_for_static_fini(
115   // CK2: ret void
116 
117   return a[0];
118 }
119 #endif // CK2
120 
121 // Test host codegen.
122 // RUN: %clang_cc1 -DCK3 -verify -fopenmp -x c++ -triple powerpc64le-unknown-unknown -fopenmp-targets=powerpc64le-ibm-linux-gnu -emit-llvm %s -o - | FileCheck %s --check-prefix CK3 --check-prefix CK3-64
123 // RUN: %clang_cc1 -DCK3 -fopenmp -x c++ -std=c++11 -triple powerpc64le-unknown-unknown -fopenmp-targets=powerpc64le-ibm-linux-gnu -emit-pch -o %t %s
124 // RUN: %clang_cc1 -DCK3 -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 CK3 --check-prefix CK3-64
125 // RUN: %clang_cc1 -DCK3 -verify -fopenmp -x c++ -triple i386-unknown-unknown -fopenmp-targets=i386-pc-linux-gnu -emit-llvm %s -o - | FileCheck %s --check-prefix CK3 --check-prefix CK3-32
126 // RUN: %clang_cc1 -DCK3 -fopenmp -x c++ -std=c++11 -triple i386-unknown-unknown -fopenmp-targets=i386-pc-linux-gnu -emit-pch -o %t %s
127 // RUN: %clang_cc1 -DCK3 -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 CK3 --check-prefix CK3-32
128 #ifdef CK3
129 
130 // CK3: [[SSI:%.+]] = type { [{{.+}} x i32], float }
131 
132 template <typename T, int X, long long Y>
133 struct SS{
134   T a[X];
135   float b;
136   // CK3: define {{.*}}i32 @{{.+}}foo{{.+}}(
137   int foo(void) {
138 
139   // CK3: 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)
140   // CK3: call void @[[OFFL1:.+]]([[SSI]]* %{{.+}})
141     #pragma omp target
142     #pragma omp teams distribute parallel for
143     for(int i = 0; i < X; i++) {
144       a[i] = (T)0;
145     }
146 
147       // outlined target region
148   // CK3: define internal void @[[OFFL1]]([[SSI]]* {{.+}})
149   // CK3: call void {{.+}} @__kmpc_fork_teams({{.+}}, i32 1, {{.+}} @[[OUTL1:.+]] to {{.+}}, {{.+}}, {{.+}})
150   // CK3: ret void
151 
152   // CK3: define internal void @[[OUTL1]]({{.+}})
153   // CK3: call void @__kmpc_for_static_init_4(
154   // CK3: call void {{.+}} @__kmpc_fork_call(
155   // CK3: call void @__kmpc_for_static_fini(
156   // CK3: ret void
157 
158     return a[0];
159   }
160 };
161 
162 int teams_template_struct(void) {
163   SS<int, 123, 456> V;
164   return V.foo();
165 
166 }
167 #endif // CK3
168 
169 // Test host codegen.
170 // RUN: %clang_cc1 -DCK4 -verify -fopenmp -x c++ -triple powerpc64le-unknown-unknown -fopenmp-targets=powerpc64le-ibm-linux-gnu -emit-llvm %s -o - | FileCheck %s --check-prefix CK4 --check-prefix CK4-64
171 // RUN: %clang_cc1 -DCK4 -fopenmp -x c++ -std=c++11 -triple powerpc64le-unknown-unknown -fopenmp-targets=powerpc64le-ibm-linux-gnu -emit-pch -o %t %s
172 // RUN: %clang_cc1 -DCK4 -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 CK4 --check-prefix CK4-64
173 // RUN: %clang_cc1 -DCK4 -verify -fopenmp -x c++ -triple i386-unknown-unknown -fopenmp-targets=i386-pc-linux-gnu -emit-llvm %s -o - | FileCheck %s --check-prefix CK4 --check-prefix CK4-32
174 // RUN: %clang_cc1 -DCK4 -fopenmp -x c++ -std=c++11 -triple i386-unknown-unknown -fopenmp-targets=i386-pc-linux-gnu -emit-pch -o %t %s
175 // RUN: %clang_cc1 -DCK4 -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 CK4 --check-prefix CK4-32
176 
177 #ifdef CK4
178 
179 template <typename T, int n>
180 int tmain(T argc) {
181   T a[n];
182   int te = n/128;
183   int th = 128;
184 #pragma omp target
185 #pragma omp teams distribute parallel for num_teams(te) thread_limit(th)
186   for(int i = 0; i < n; i++) {
187     a[i] = (T)0;
188   }
189   return 0;
190 }
191 
192 int main (int argc, char **argv) {
193   int n = 100;
194   int a[n];
195 #pragma omp target
196 #pragma omp teams distribute parallel for
197   for(int i = 0; i < n; i++) {
198     a[i] = 0;
199   }
200   return tmain<int, 10>(argc);
201 }
202 
203 // CK4:  define {{.*}}i32 @{{[^,]+}}(i{{.+}}{{.+}} %[[ARGC:.+]], {{.+}})
204 // CK4:   call i32 @__tgt_target_teams(i64 -1, i8* @{{[^,]+}}, i32 3, i8** %{{[^,]+}}, i8** %{{[^,]+}}, i{{64|32}}* {{.+}}, i64* {{.+}}@{{[^,]+}}, i32 0, i32 0), i32 0, i32 0)
205 // CK4: call void @[[OFFL1:.+]]({{.+}})
206 // CK4: {{%.+}} = call{{.*}} i32 @[[TMAIN:.+]]({{.+}})
207 // CK4:  ret
208 
209 // CK4:  define {{.*}}void @[[OFFL1]]({{.+}})
210 // CK4: call void {{.+}} @__kmpc_fork_teams({{.+}}, i32 3, {{.+}} @[[OUTL1:.+]] to {{.+}}, {{.+}}, {{.+}})
211 // CK4: ret void
212 
213 // CK4: define internal void @[[OUTL1]]({{.+}})
214 // CK4: call void @__kmpc_for_static_init_4(
215 // CK4: call void {{.+}} @__kmpc_fork_call(
216 // CK4: call void @__kmpc_for_static_fini(
217 // CK4: ret void
218 
219 // CK4:  define {{.*}}i32 @[[TMAIN]]({{.+}})
220 // CK4:   call i32 @__tgt_target_teams(i64 -1, i8* @{{[^,]+}}, i32 3, i8** %{{[^,]+}}, i8** %{{[^,]+}}, i{{64|32}}* {{.+}}@{{[^,]+}}, i32 0, i32 0), i64* {{.+}}@{{[^,]+}}, i32 0, i32 0), i32 {{.+}}, i32 {{.+}})
221 // CK4: call void @[[OFFLT:.+]]({{.+}})
222 // CK4:  ret
223 // CK4-NEXT: }
224 
225 // CK4: define {{.*}}void @[[OFFLT]](i{{32|64}} [[TE_ARG:%.+]], i{{32|64}} [[TH_ARG:%.+]], {{.+}})
226 // CK4: [[TE_ADDR:%.+]] = alloca i{{32|64}},
227 // CK4: [[TH_ADDR:%.+]] = alloca i{{32|64}},
228 // CK4: store{{.+}} [[TE_ARG]], {{.+}} [[TE_ADDR]],
229 // CK4: store{{.+}} [[TH_ARG]], {{.+}} [[TH_ADDR]],
230 // CK4-64: [[TE_CONV:%.+]] = bitcast{{.+}} [[TE_ADDR]] to
231 // CK4-64: [[TH_CONV:%.+]] = bitcast{{.+}} [[TH_ADDR]] to
232 // CK4-64: [[TE_VAL:%.+]] = load i32, i32* [[TE_CONV]],
233 // CK4-64: [[TH_VAL:%.+]] = load i32, i32* [[TH_CONV]],
234 // CK4-32: [[TE_VAL:%.+]] = load i32, i32* [[TE_ADDR]],
235 // CK4-32: [[TH_VAL:%.+]] = load i32, i32* [[TH_ADDR]],
236 // CK4: {{%.+}} = call i32 @__kmpc_push_num_teams({{.+}}, {{.+}}, i32 [[TE_VAL]], i32 [[TH_VAL]])
237 // CK4: call void {{.+}} @__kmpc_fork_teams({{.+}}, i32 1, {{.+}} @[[OUTLT:.+]] to {{.+}}, {{.+}}, {{.+}})
238 // CK4: ret void
239 
240 // CK4: define internal void @[[OUTLT]]({{.+}})
241 // CK4: call void @__kmpc_for_static_init_4(
242 // CK4: call void {{.+}} @__kmpc_fork_call(
243 // CK4: call void @__kmpc_for_static_fini(
244 // CK4: ret void
245 
246 #endif // CK4
247 #endif
248