xref: /llvm-project/clang/test/OpenMP/target_teams_distribute_parallel_for_ast_print.cpp (revision 7c1d9b15eee3a34678addab2bab66f3020ac0753)
1 // RUN: %clang_cc1 -verify -fopenmp -fopenmp-version=50 -ast-print %s -Wno-openmp-mapping -Wsign-conversion | 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 -Wsign-conversion -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 -Wsign-conversion | 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 -Wsign-conversion -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 private(a) private(this->a) private(T::a) allocate(omp_cgroup_mem_alloc:a) uses_allocators(omp_cgroup_mem_alloc)
48     for (int k = 0; k < a.a; ++k) {
49       ++this->a.a;
50 #pragma omp cancel for
51     }
52   }
operator =(S7 & s)53   S7 &operator=(S7 &s) {
54 #pragma omp target teams distribute parallel for private(a) private(this->a)
55     for (int k = 0; k < s.a.a; ++k)
56       ++s.a.a;
57 
58     foo();
59     return *this;
60   }
foo()61   void foo() {
62     int b, argv, d, c, e, f;
63 #pragma omp target teams distribute parallel for 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   }
67 };
68 // CHECK: #pragma omp target teams distribute parallel for private(this->a) private(this->a) private(T::a) allocate(omp_cgroup_mem_alloc: this->a) uses_allocators(omp_cgroup_mem_alloc)
69 // CHECK: #pragma omp cancel for
70 // CHECK: #pragma omp target teams distribute parallel for private(this->a) private(this->a)
71 // CHECK: #pragma omp target teams distribute parallel for default(none) private(b) firstprivate(argv) shared(d) reduction(+: c) reduction(max: e) num_teams(f) thread_limit(d)
72 
73 class S8 : public S7<S> {
S8()74   S8() {}
75 
76 public:
S8(int v)77   S8(int v) : S7<S>(v){
78 #pragma omp target teams distribute parallel for private(a) private(this->a) private(S7<S>::a)
79     for (int k = 0; k < a.a; ++k)
80       ++this->a.a;
81   }
operator =(S8 & s)82   S8 &operator=(S8 &s) {
83 #pragma omp target teams distribute parallel for private(a) private(this->a)
84     for (int k = 0; k < s.a.a; ++k)
85       ++s.a.a;
86 
87     bar();
88     return *this;
89   }
bar()90   void bar() {
91     int b, argv, d, c, e, f8;
92 #pragma omp target teams distribute parallel for allocate(b) default(none), private(b) firstprivate(argv) shared(d) reduction(+:c) reduction(max:e) num_teams(f8) thread_limit(d) allocate(e)
93     for (int k = 0; k < a.a; ++k)
94       ++a.a;
95   }
96 };
97 // CHECK: #pragma omp target teams distribute parallel for private(this->a) private(this->a) private(this->S::a)
98 // CHECK: #pragma omp target teams distribute parallel for private(this->a) private(this->a) private(this->S7<S>::a)
99 // CHECK: #pragma omp target teams distribute parallel for private(this->a) private(this->a)
100 // CHECK: #pragma omp target teams distribute parallel for allocate(b) default(none) private(b) firstprivate(argv) shared(d) reduction(+: c) reduction(max: e) num_teams(f8) thread_limit(d) allocate(e)
101 
102 template <class T, int N>
tmain(T argc)103 T tmain(T argc) {
104   T b = argc, c, d, e, f, g;
105   static T a;
106 // CHECK: static T a;
107   const T clen = 5;
108   const T alen = 16;
109   int arr[10];
110 #pragma omp target teams distribute parallel for
111   for (int i=0; i < 2; ++i)
112     a = 2;
113 // CHECK: #pragma omp target teams distribute parallel for{{$}}
114 // CHECK-NEXT: for (int i = 0; i < 2; ++i)
115 // CHECK-NEXT: a = 2;
116 #ifdef OMP51
117 #pragma omp target teams distribute parallel for private(argc, b), firstprivate(c, d), collapse(2) order(unconstrained:concurrent)
118 #else
119 #pragma omp target teams distribute parallel for private(argc, b), firstprivate(c, d), collapse(2) order(concurrent)
120 #endif // OMP51
121   for (int i = 0; i < 10; ++i)
122     for (int j = 0; j < 10; ++j)
123       foo();
124 // OMP51: #pragma omp target teams distribute parallel for private(argc,b) firstprivate(c,d) collapse(2) order(unconstrained: concurrent)
125 // OMP50: #pragma omp target teams distribute parallel for private(argc,b) firstprivate(c,d) collapse(2) order(concurrent)
126 // CHECK-NEXT: for (int i = 0; i < 10; ++i)
127 // CHECK-NEXT: for (int j = 0; j < 10; ++j)
128 // CHECK-NEXT: foo();
129   for (int i = 0; i < 10; ++i)
130     foo();
131 // CHECK: for (int i = 0; i < 10; ++i)
132 // CHECK-NEXT: foo();
133 #pragma omp target teams distribute parallel for
134   for (int i = 0; i < 10; ++i)
135     foo();
136 // CHECK: #pragma omp target teams distribute parallel for
137 // CHECK-NEXT: for (int i = 0; i < 10; ++i)
138 // CHECK-NEXT: foo();
139 #pragma omp target teams distribute parallel for default(none), private(b) firstprivate(argc) shared(d) reduction(+:c) reduction(max:e) num_teams(f) thread_limit(d)
140     for (int k = 0; k < 10; ++k)
141       e += d + argc;
142 // CHECK: #pragma omp target teams distribute parallel for default(none) private(b) firstprivate(argc) shared(d) reduction(+: c) reduction(max: e) num_teams(f) thread_limit(d)
143 // CHECK-NEXT: for (int k = 0; k < 10; ++k)
144 // CHECK-NEXT: e += d + argc;
145 #pragma omp target teams distribute parallel for
146   for (int k = 0; k < 10; ++k)
147     e += d + argc;
148 // CHECK: #pragma omp target teams distribute parallel for
149 // CHECK-NEXT: for (int k = 0; k < 10; ++k)
150 // CHECK-NEXT: e += d + argc;
151   return T();
152 }
153 
main(int argc,char ** argv)154 int main (int argc, char **argv) {
155   int b = argc, c, d, e, f, g;
156   static int a;
157 // CHECK: static int a;
158   const int clen = 5;
159   const int N = 10;
160   int arr[10];
161 #pragma omp target teams distribute parallel for
162   for (int i=0; i < 2; ++i)
163     a = 2;
164 // CHECK: #pragma omp target teams distribute parallel for
165 // CHECK-NEXT: for (int i = 0; i < 2; ++i)
166 // CHECK-NEXT: a = 2;
167 #pragma omp target teams distribute parallel for private(argc,b),firstprivate(argv, c), collapse(2)
168   for (int i = 0; i < 10; ++i)
169     for (int j = 0; j < 10; ++j)
170       foo();
171 // CHECK: #pragma omp target teams distribute parallel for private(argc,b) firstprivate(argv,c) collapse(2)
172 // CHECK-NEXT: for (int i = 0; i < 10; ++i)
173 // CHECK-NEXT: for (int j = 0; j < 10; ++j)
174 // CHECK-NEXT: foo();
175   for (int i = 0; i < 10; ++i)
176     foo();
177 // CHECK: for (int i = 0; i < 10; ++i)
178 // CHECK-NEXT: foo();
179 #pragma omp target teams distribute parallel for
180   for (int i = 0; i < 10; ++i)foo();
181 // CHECK: #pragma omp target teams distribute parallel for
182 // CHECK-NEXT: for (int i = 0; i < 10; ++i)
183 // CHECK-NEXT: foo();
184 #pragma omp target teams distribute parallel for default(none), private(b) firstprivate(argc) shared(d) reduction(+:c) reduction(max:e) num_teams(f) thread_limit(d)
185   for (int k = 0; k < 10; ++k)
186     e += d + argc;
187 // CHECK: #pragma omp target teams distribute parallel for default(none) private(b) firstprivate(argc) shared(d) reduction(+: c) reduction(max: e) num_teams(f) thread_limit(d)
188 // CHECK-NEXT: for (int k = 0; k < 10; ++k)
189 // CHECK-NEXT: e += d + argc;
190 #pragma omp target teams distribute parallel for reduction(task,+:argc)
191   for (int k = 0; k < 10; ++k)
192     e += d + argc;
193 // CHECK: #pragma omp target teams distribute parallel for reduction(task, +: argc)
194 // CHECK-NEXT: for (int k = 0; k < 10; ++k)
195 // CHECK-NEXT: e += d + argc;
196   return (0);
197 }
198 
199 #endif
200