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