xref: /llvm-project/clang/test/OpenMP/target_teams_distribute_parallel_for_simd_ast_print.cpp (revision 7c1d9b15eee3a34678addab2bab66f3020ac0753)
1 // RUN: %clang_cc1 -verify -fopenmp -fopenmp-version=50 -ast-print %s -Wno-openmp-mapping | FileCheck %s --check-prefix=CHECK --check-prefix=OMP50
2 // RUN: %clang_cc1 -fopenmp -fopenmp-version=50 -x c++ -std=c++11 -emit-pch -o %t %s -Wno-openmp-mapping
3 // RUN: %clang_cc1 -fopenmp -fopenmp-version=50 -std=c++11 -include-pch %t -verify %s -ast-print -Wno-openmp-mapping | FileCheck %s --check-prefix=CHECK --check-prefix=OMP50
4 // RUN: %clang_cc1 -verify -fopenmp -ast-print %s -Wno-openmp-mapping -DOMP51 | FileCheck %s --check-prefix=CHECK --check-prefix=OMP51
5 // RUN: %clang_cc1 -fopenmp -x c++ -std=c++11 -emit-pch -o %t %s -Wno-openmp-mapping -DOMP51
6 // RUN: %clang_cc1 -fopenmp -std=c++11 -include-pch %t -verify %s -ast-print -Wno-openmp-mapping -DOMP51 | FileCheck %s --check-prefix=CHECK --check-prefix=OMP51
7 
8 // RUN: %clang_cc1 -verify -fopenmp-simd -fopenmp-version=50 -ast-print %s -Wno-openmp-mapping | FileCheck %s --check-prefix=CHECK --check-prefix=OMP50
9 // RUN: %clang_cc1 -fopenmp-simd -fopenmp-version=50 -x c++ -std=c++11 -emit-pch -o %t %s -Wno-openmp-mapping
10 // RUN: %clang_cc1 -fopenmp-simd -fopenmp-version=50 -std=c++11 -include-pch %t -verify %s -ast-print -Wno-openmp-mapping | FileCheck %s --check-prefix=CHECK --check-prefix=OMP50
11 // RUN: %clang_cc1 -verify -fopenmp-simd -ast-print %s -Wno-openmp-mapping -DOMP51 | FileCheck %s --check-prefix=CHECK --check-prefix=OMP51
12 // RUN: %clang_cc1 -fopenmp-simd -x c++ -std=c++11 -emit-pch -o %t %s -Wno-openmp-mapping -DOMP51
13 // RUN: %clang_cc1 -fopenmp-simd -std=c++11 -include-pch %t -verify %s -ast-print -Wno-openmp-mapping -DOMP51 | FileCheck %s --check-prefix=CHECK --check-prefix=OMP51
14 // expected-no-diagnostics
15 
16 #ifndef HEADER
17 #define HEADER
18 
19 typedef void **omp_allocator_handle_t;
20 extern const omp_allocator_handle_t omp_null_allocator;
21 extern const omp_allocator_handle_t omp_default_mem_alloc;
22 extern const omp_allocator_handle_t omp_large_cap_mem_alloc;
23 extern const omp_allocator_handle_t omp_const_mem_alloc;
24 extern const omp_allocator_handle_t omp_high_bw_mem_alloc;
25 extern const omp_allocator_handle_t omp_low_lat_mem_alloc;
26 extern const omp_allocator_handle_t omp_cgroup_mem_alloc;
27 extern const omp_allocator_handle_t omp_pteam_mem_alloc;
28 extern const omp_allocator_handle_t omp_thread_mem_alloc;
29 
foo()30 void foo() {}
31 
32 struct S {
SS33   S(): a(0) {}
SS34   S(int v) : a(v) {}
35   int a;
36   typedef int type;
37 };
38 
39 template <typename T>
40 class S7 : public T {
41 protected:
42   T a;
S7()43   S7() : a(0) {}
44 
45 public:
S7(typename T::type v)46   S7(typename T::type v) : a(v) {
47 #pragma omp target teams distribute parallel for simd private(a) private(this->a) private(T::a)
48     for (int k = 0; k < a.a; ++k)
49       ++this->a.a;
50   }
operator =(S7 & s)51   S7 &operator=(S7 &s) {
52     int k;
53 #pragma omp target teams distribute parallel for simd allocate(omp_pteam_mem_alloc: a) private(a) private(this->a) linear(k) allocate(k) uses_allocators(omp_pteam_mem_alloc)
54     for (k = 0; k < s.a.a; ++k)
55       ++s.a.a;
56 
57     foo();
58     bar();
59     return *this;
60   }
foo()61   void foo() {
62     int b, argv, d, c, e, f;
63 #pragma omp target teams distribute parallel for simd default(none), private(b) firstprivate(argv) shared(d) reduction(+:c) reduction(max:e) num_teams(f) thread_limit(d)
64     for (int k = 0; k < a.a; ++k)
65       ++a.a;
66   }
bar()67   void bar() {
68     int arr[10];
69     const int alen = 16;
70     const int slen1 = 8;
71     const int slen2 = 8;
72 #pragma omp target teams distribute parallel for simd simdlen(slen1) safelen(slen2) aligned(arr:alen)
73     for (int k = 0; k < a.a; ++k)
74       ++a.a;
75   }
76 };
77 // CHECK: #pragma omp target teams distribute parallel for simd private(this->a) private(this->a) private(T::a)
78 // CHECK: #pragma omp target teams distribute parallel for simd allocate(omp_pteam_mem_alloc: this->a) private(this->a) private(this->a) linear(k) allocate(k) uses_allocators(omp_pteam_mem_alloc)
79 // CHECK: #pragma omp target teams distribute parallel for simd default(none) private(b) firstprivate(argv) shared(d) reduction(+: c) reduction(max: e) num_teams(f) thread_limit(d)
80 // CHECK: #pragma omp target teams distribute parallel for simd simdlen(slen1) safelen(slen2) aligned(arr: alen)
81 
82 class S8 : public S7<S> {
S8()83   S8() {}
84 
85 public:
S8(int v)86   S8(int v) : S7<S>(v){
87 #pragma omp target teams distribute parallel for simd private(a) private(this->a) private(S7<S>::a)
88     for (int k = 0; k < a.a; ++k)
89       ++this->a.a;
90   }
operator =(S8 & s)91   S8 &operator=(S8 &s) {
92 #pragma omp target teams distribute parallel for simd private(a) private(this->a)
93     for (int k = 0; k < s.a.a; ++k)
94       ++s.a.a;
95 
96     foo();
97     bar();
98     return *this;
99   }
bar()100   void bar() {
101     int b, argv, d, c, e, f8;
102 #pragma omp target teams distribute parallel for simd default(none), private(b) firstprivate(argv) shared(d) reduction(+:c) reduction(max:e) num_teams(f8) thread_limit(d)
103     for (int k = 0; k < a.a; ++k)
104       ++a.a;
105   }
foo()106   void foo() {
107     const int alen = 16;
108     const int slen1 = 8;
109     const int slen2 = 8;
110     int arr[10];
111 #pragma omp target teams distribute parallel for simd simdlen(slen1) safelen(slen2) aligned(arr:alen)
112     for (int k = 0; k < a.a; ++k)
113       ++a.a;
114   }
115 };
116 // CHECK: #pragma omp target teams distribute parallel for simd private(this->a) private(this->a) private(this->S::a)
117 // CHECK: #pragma omp target teams distribute parallel for simd private(this->a) private(this->a) private(this->S7<S>::a)
118 // CHECK: #pragma omp target teams distribute parallel for simd private(this->a) private(this->a)
119 // CHECK: #pragma omp target teams distribute parallel for simd default(none) private(b) firstprivate(argv) shared(d) reduction(+: c) reduction(max: e) num_teams(f8) thread_limit(d)
120 // CHECK: #pragma omp target teams distribute parallel for simd simdlen(slen1) safelen(slen2) aligned(arr: alen)
121 
122 template <class T, int N>
tmain(T argc)123 T tmain(T argc) {
124   T b = argc, c, d, e, f, g;
125   static T a;
126 // CHECK: static T a;
127   const T clen = 5;
128   const T alen = 16;
129   int arr[10];
130 #pragma omp target teams distribute parallel for simd
131   for (int i=0; i < 2; ++i)
132     a = 2;
133 // CHECK: #pragma omp target teams distribute parallel for simd{{$}}
134 // CHECK-NEXT: for (int i = 0; i < 2; ++i)
135 // CHECK-NEXT: a = 2;
136 #pragma omp target teams distribute parallel for simd private(argc, b), firstprivate(c, d), collapse(2)
137   for (int i = 0; i < 10; ++i)
138     for (int j = 0; j < 10; ++j)
139       foo();
140 // CHECK: #pragma omp target teams distribute parallel for simd private(argc,b) firstprivate(c,d) collapse(2)
141 // CHECK-NEXT: for (int i = 0; i < 10; ++i)
142 // CHECK-NEXT: for (int j = 0; j < 10; ++j)
143 // CHECK-NEXT: foo();
144   for (int i = 0; i < 10; ++i)
145     foo();
146 // CHECK: for (int i = 0; i < 10; ++i)
147 // CHECK-NEXT: foo();
148 #pragma omp target teams distribute parallel for simd
149   for (int i = 0; i < 10; ++i)
150     foo();
151 // CHECK: #pragma omp target teams distribute parallel for simd
152 // CHECK-NEXT: for (int i = 0; i < 10; ++i)
153 // CHECK-NEXT: foo();
154 #pragma omp target teams distribute parallel for simd default(none), private(b) firstprivate(argc) shared(d) reduction(+:c) reduction(max:e) num_teams(f) thread_limit(d)
155     for (int k = 0; k < 10; ++k)
156       e += d + argc;
157 // CHECK: #pragma omp target teams distribute parallel for simd default(none) private(b) firstprivate(argc) shared(d) reduction(+: c) reduction(max: e) num_teams(f) thread_limit(d)
158 // CHECK-NEXT: for (int k = 0; k < 10; ++k)
159 // CHECK-NEXT: e += d + argc;
160 #pragma omp target teams distribute parallel for simd simdlen(clen-1)
161   for (int k = 0; k < 10; ++k)
162     e += d + argc;
163 // CHECK: #pragma omp target teams distribute parallel for simd simdlen(clen - 1)
164 // CHECK-NEXT: for (int k = 0; k < 10; ++k)
165 // CHECK-NEXT: e += d + argc;
166 #pragma omp target teams distribute parallel for simd safelen(clen-1) aligned(arr:alen)
167   for (int k = 0; k < 10; ++k)
168     e += d + argc + arr[k];
169 // CHECK: #pragma omp target teams distribute parallel for simd safelen(clen - 1) aligned(arr: alen)
170 // CHECK-NEXT: for (int k = 0; k < 10; ++k)
171 // CHECK-NEXT: e += d + argc + arr[k];
172   return T();
173 }
174 
main(int argc,char ** argv)175 int main (int argc, char **argv) {
176   int b = argc, c, d, e, f, g;
177   static int a;
178 // CHECK: static int a;
179   const int clen = 5;
180   const int N = 10;
181   int arr[10];
182 #pragma omp target teams distribute parallel for simd
183   for (int i=0; i < 2; ++i)
184     a = 2;
185 // CHECK: #pragma omp target teams distribute parallel for simd
186 // CHECK-NEXT: for (int i = 0; i < 2; ++i)
187 // CHECK-NEXT: a = 2;
188 #ifdef OMP51
189 #pragma omp target teams distribute parallel for simd private(argc,b),firstprivate(argv, c), collapse(2) order(reproducible:concurrent)
190 #else
191 #pragma omp target teams distribute parallel for simd private(argc,b),firstprivate(argv, c), collapse(2) order(concurrent)
192 #endif // OMP51
193   for (int i = 0; i < 10; ++i)
194     for (int j = 0; j < 10; ++j)
195       foo();
196 // OMP50: #pragma omp target teams distribute parallel for simd private(argc,b) firstprivate(argv,c) collapse(2) order(concurrent)
197 // OMP51: #pragma omp target teams distribute parallel for simd private(argc,b) firstprivate(argv,c) collapse(2) order(reproducible: concurrent)
198 // CHECK-NEXT: for (int i = 0; i < 10; ++i)
199 // CHECK-NEXT: for (int j = 0; j < 10; ++j)
200 // CHECK-NEXT: foo();
201   for (int i = 0; i < 10; ++i)
202     foo();
203 // CHECK: for (int i = 0; i < 10; ++i)
204 // CHECK-NEXT: foo();
205 #pragma omp target teams distribute parallel for simd
206   for (int i = 0; i < 10; ++i)foo();
207 // CHECK: #pragma omp target teams distribute parallel for simd
208 // CHECK-NEXT: for (int i = 0; i < 10; ++i)
209 // CHECK-NEXT: foo();
210 #pragma omp target teams distribute parallel for simd default(none), private(b) firstprivate(argc) shared(d) reduction(+:c) reduction(max:e) num_teams(f) thread_limit(d)
211   for (int k = 0; k < 10; ++k)
212     e += d + argc;
213 // CHECK: #pragma omp target teams distribute parallel for simd default(none) private(b) firstprivate(argc) shared(d) reduction(+: c) reduction(max: e) num_teams(f) thread_limit(d)
214 // CHECK-NEXT: for (int k = 0; k < 10; ++k)
215 // CHECK-NEXT: e += d + argc;
216 #pragma omp target teams distribute parallel for simd simdlen(clen-1)
217   for (int k = 0; k < 10; ++k)
218     e += d + argc;
219 // CHECK: #pragma omp target teams distribute parallel for simd simdlen(clen - 1)
220 // CHECK-NEXT: for (int k = 0; k < 10; ++k)
221 // CHECK-NEXT: e += d + argc;
222 #pragma omp target teams distribute parallel for simd safelen(clen-1) aligned(arr:N+6)
223   for (int k = 0; k < 10; ++k)
224     e += d + argc + arr[k];
225 // CHECK: #pragma omp target teams distribute parallel for simd safelen(clen - 1) aligned(arr: N + 6)
226 // CHECK-NEXT: for (int k = 0; k < 10; ++k)
227 // CHECK-NEXT: e += d + argc + arr[k];
228   return (0);
229 }
230 
231 #endif
232