1 // expected-no-diagnostics
2 #ifndef HEADER
3 #define HEADER
4 
5 // Test host codegen.
6 // 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
7 // 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
8 // 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
9 // 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
10 // 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
11 // 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
12 #ifdef CK1
13 
14 template <typename T, int X, long long Y>
15 struct SS{
16   T a[X];
17   float b;
18   // CK1: define {{.*}}i32 @{{.+}}foo{{.+}}(
19   int foo(void) {
20 
21   // CK1: call i32 @__tgt_target_teams(
22   // CK1: call void @[[OFFL1:.+]](
23     #pragma omp target
24     #pragma omp teams distribute simd
25     for(int i = 0; i < X; i++) {
26       a[i] = (T)0;
27     }
28   // CK1: call i32 @__tgt_target_teams(
29   // CK1: call void @[[OFFL2:.+]](
30     #pragma omp target
31     #pragma omp teams distribute simd dist_schedule(static)
32     for(int i = 0; i < X; i++) {
33       a[i] = (T)0;
34     }
35   // CK1: call i32 @__tgt_target_teams(
36   // CK1: call void @[[OFFL3:.+]](
37     #pragma omp target
38     #pragma omp teams distribute simd dist_schedule(static, X/2)
39     for(int i = 0; i < X; i++) {
40       a[i] = (T)0;
41     }
42   // CK1: define internal void @[[OFFL1]](
43   // CK1: call void {{.+}} @__kmpc_fork_teams({{.+}}, i32 1, {{.+}} @[[OUTL1:.+]] to {{.+}},
44   // CK1: ret void
45 
46   // CK1: define internal void @[[OUTL1]]({{.+}})
47   // CK1: call void @__kmpc_for_static_init_4({{.+}}, {{.+}}, i32 92
48   // CK1: call void @__kmpc_for_static_fini(
49   // CK1: ret void
50 
51   // CK1: define internal void @[[OFFL2]](
52   // CK1: call void {{.+}} @__kmpc_fork_teams({{.+}}, i32 1, {{.+}} @[[OUTL2:.+]] to {{.+}},
53   // CK1: ret void
54 
55   // CK1: define internal void @[[OUTL2]]({{.+}})
56   // CK1: call void @__kmpc_for_static_init_4({{.+}}, {{.+}}, i32 92
57   // CK1: call void @__kmpc_for_static_fini(
58   // CK1: ret void
59 
60   // CK1: define internal void @[[OFFL3]](
61   // CK1: call void {{.+}} @__kmpc_fork_teams({{.+}}, i32 1, {{.+}} @[[OUTL3:.+]] to {{.+}},
62   // CK1: ret void
63 
64   // CK1: define internal void @[[OUTL3]]({{.+}})
65   // CK1: call void @__kmpc_for_static_init_4({{.+}}, {{.+}}, i32 91
66   // CK1: call void @__kmpc_for_static_fini(
67   // CK1: ret void
68 
69     return a[0];
70   }
71 };
72 
73 int teams_template_struct(void) {
74   SS<int, 123, 456> V;
75   return V.foo();
76 
77 }
78 // CK1: !{!"llvm.loop.vectorize.enable", i1 true}
79 #endif // CK1
80 
81 // Test host codegen.
82 // 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
83 // 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
84 // 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
85 // 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
86 // 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
87 // 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
88 #ifdef CK2
89 
90 template <typename T, int n>
91 int tmain(T argc) {
92   T a[n];
93 #pragma omp target
94 #pragma omp teams distribute simd
95   for(int i = 0; i < n; i++) {
96     a[i] = (T)0;
97   }
98 #pragma omp target
99 #pragma omp teams distribute simd dist_schedule(static)
100   for(int i = 0; i < n; i++) {
101     a[i] = (T)0;
102   }
103 #pragma omp target
104 #pragma omp teams distribute simd dist_schedule(static, n)
105   for(int i = 0; i < n; i++) {
106     a[i] = (T)0;
107   }
108   return 0;
109 }
110 
111 int main (int argc, char **argv) {
112   int n = 100;
113   int a[n];
114 #pragma omp target
115 #pragma omp teams distribute simd
116   for(int i = 0; i < n; i++) {
117     a[i] = 0;
118   }
119 #pragma omp target
120 #pragma omp teams distribute simd dist_schedule(static)
121   for(int i = 0; i < n; i++) {
122     a[i] = 0;
123   }
124 #pragma omp target
125 #pragma omp teams distribute simd dist_schedule(static, n)
126   for(int i = 0; i < n; i++) {
127     a[i] = 0;
128   }
129   return tmain<int, 10>(argc);
130 }
131 
132 // CK2: define {{.*}}i32 @{{[^,]+}}(i{{.+}}{{.+}} %[[ARGC:.+]], {{.+}})
133 // CK2: call i32 @__tgt_target_teams(
134 // CK2: call void @[[OFFL1:.+]]({{.+}})
135 // CK2: call i32 @__tgt_target_teams(
136 // CK2: call void @[[OFFL2:.+]]({{.+}})
137 // CK2: call i32 @__tgt_target_teams(
138 // CK2: call void @[[OFFL3:.+]]({{.+}})
139 // CK2: {{%.+}} = call{{.*}} i32 @[[TMAIN:.+]]({{.+}})
140 // CK2: ret
141 
142 // CK2:  define {{.*}}void @[[OFFL1]]({{.+}})
143 // CK2: call void {{.+}} @__kmpc_fork_teams({{.+}}, i32 3, {{.+}} @[[OUTL1:.+]] to {{.+}},
144 // CK2: ret void
145 
146 // CK2: define internal void @[[OUTL1]]({{.+}})
147 // CK2: call void @__kmpc_for_static_init_4({{.+}}, {{.+}}, i32 92
148 // CK2: call void @__kmpc_for_static_fini(
149 // CK2: ret void
150 
151 // CK2: define {{.*}}void @[[OFFL2]]({{.+}})
152 // CK2: call void {{.+}} @__kmpc_fork_teams({{.+}}, i32 3, {{.+}} @[[OUTL2:.+]] to {{.+}},
153 // CK2: ret void
154 
155 // CK2: define internal void @[[OUTL2]]({{.+}})
156 // CK2: call void @__kmpc_for_static_init_4({{.+}}, {{.+}}, i32 92
157 // CK2: call void @__kmpc_for_static_fini(
158 // CK2: ret void
159 
160 // CK2:  define {{.*}}void @[[OFFL3]]({{.+}})
161 // CK2: call void {{.+}} @__kmpc_fork_teams({{.+}}, i32 4, {{.+}} @[[OUTL3:.+]] to {{.+}},
162 // CK2: ret void
163 
164 // CK2: define internal void @[[OUTL3]]({{.+}})
165 // CK2: call void @__kmpc_for_static_init_4({{.+}}, {{.+}}, i32 91
166 // CK2: call void @__kmpc_for_static_fini(
167 // CK2: ret void
168 
169 // CK2: define {{.*}}i32 @[[TMAIN]]({{.+}})
170 // CK2: call i32 @__tgt_target_teams(
171 // CK2: call void @[[OFFLT1:.+]]({{.+}})
172 // CK2: call i32 @__tgt_target_teams(
173 // CK2: call void @[[OFFLT2:.+]]({{.+}})
174 // CK2: call i32 @__tgt_target_teams(
175 // CK2: call void @[[OFFLT3:.+]]({{.+}})
176 // CK2:  ret
177 // CK2-NEXT: }
178 
179 // CK2:  define {{.*}}void @[[OFFLT1]]({{.+}})
180 // CK2: call void {{.+}} @__kmpc_fork_teams({{.+}}, i32 1, {{.+}} @[[OUTLT1:.+]] to {{.+}},
181 // CK2: ret void
182 
183 // CK2: define internal void @[[OUTLT1]]({{.+}})
184 // CK2: call void @__kmpc_for_static_init_4({{.+}}, {{.+}}, i32 92
185 // CK2: call void @__kmpc_for_static_fini(
186 // CK2: ret void
187 
188 // CK2:  define {{.*}}void @[[OFFLT2]]({{.+}})
189 // CK2: call void {{.+}} @__kmpc_fork_teams({{.+}}, i32 1, {{.+}} @[[OUTLT2:.+]] to {{.+}},
190 // CK2: ret void
191 
192 // CK2: define internal void @[[OUTLT2]]({{.+}})
193 // CK2: call void @__kmpc_for_static_init_4({{.+}}, {{.+}}, i32 92
194 // CK2: call void @__kmpc_for_static_fini(
195 // CK2: ret void
196 
197 // CK2:  define {{.*}}void @[[OFFLT3]]({{.+}})
198 // CK2: call void {{.+}} @__kmpc_fork_teams({{.+}}, i32 1, {{.+}} @[[OUTLT3:.+]] to {{.+}},
199 // CK2: ret void
200 
201 // CK2: define internal void @[[OUTLT3]]({{.+}})
202 // CK2: call void @__kmpc_for_static_init_4({{.+}}, {{.+}}, i32 91
203 // CK2: call void @__kmpc_for_static_fini(
204 // CK2: ret void
205 
206 // CK2: !{!"llvm.loop.vectorize.enable", i1 true}
207 #endif // CK2
208 #endif // #ifndef HEADER
209