xref: /llvm-project/clang/test/OpenMP/task_ast_print.cpp (revision 20e904950967c125abc1e91f57e5a373987ff016)
1 // RUN: %clang_cc1 -verify -Wno-vla -fopenmp -ast-print %s | FileCheck %s
2 // RUN: %clang_cc1 -fopenmp -x c++ -std=c++11 -emit-pch -o %t %s
3 // RUN: %clang_cc1 -fopenmp -std=c++11 -include-pch %t -verify -Wno-vla %s -ast-print | FileCheck %s
4 
5 // RUN: %clang_cc1 -verify -Wno-vla -fopenmp-simd -ast-print %s | FileCheck %s
6 // RUN: %clang_cc1 -fopenmp-simd -x c++ -std=c++11 -emit-pch -o %t %s
7 // RUN: %clang_cc1 -fopenmp-simd -std=c++11 -include-pch %t -verify -Wno-vla %s -ast-print | FileCheck %s
8 // RUN: %clang_cc1 -triple x86_64-pc-linux-gnu -fopenmp -ast-dump  %s | FileCheck %s --check-prefix=DUMP
9 // expected-no-diagnostics
10 
11 #ifndef HEADER
12 #define HEADER
13 
14 typedef void *omp_depend_t;
15 typedef unsigned long omp_event_handle_t;
16 
17 void foo() {}
18 
19 struct S1 {
20   S1(): a(0) {}
21   S1(int v) : a(v) {}
22   int a;
23   typedef int type;
24   S1 operator +(const S1&);
25 };
26 
27 template <typename T>
28 class S7 : public T {
29 protected:
30   T a, b, c[10], d[10];
31   S7() : a(0) {}
32 
33 public:
34   S7(typename T::type v) : a(v) {
35     omp_depend_t x;
36     omp_event_handle_t evt;
37 #pragma omp taskgroup allocate(b) task_reduction(+:b)
38 #pragma omp task private(a) private(this->a) private(T::a) in_reduction(+:this->b) allocate(b) depend(depobj:x) detach(evt) depend(iterator(i=0:10:1, T *k = &a:&b), in: c[i], d[(int)(k-&a)]) affinity(iterator(i=0:10:1, T *k = &a:&b): c[i], d[(int)(k-&a)])
39     for (int k = 0; k < a.a; ++k)
40       ++this->a.a;
41   }
42   S7 &operator=(S7 &s) {
43 #pragma omp task private(a) private(this->a)
44     for (int k = 0; k < s.a.a; ++k)
45       ++s.a.a;
46     return *this;
47   }
48 };
49 
50 // CHECK: #pragma omp taskgroup allocate(this->b) task_reduction(+: this->b)
51 // CHECK: #pragma omp task private(this->a) private(this->a) private(T::a) in_reduction(+: this->b) allocate(this->b) depend(depobj : x) detach(evt) depend(iterator(int i = 0:10:1, T * k = &this->a:&this->b), in : this->c[i],this->d[(int)(k - &this->a)]) affinity(iterator(int i = 0:10:1, T * k = &this->a:&this->b) : this->c[i],this->d[(int)(k - &this->a)]){{$}}
52 // CHECK: #pragma omp task private(this->a) private(this->a)
53 // CHECK: #pragma omp task private(this->a) private(this->a) private(this->S1::a) in_reduction(+: this->b) allocate(this->b) depend(depobj : x) detach(evt) depend(iterator(int i = 0:10:1, S1 * k = &this->a:&this->b), in : this->c[i],this->d[(int)(k - &this->a)]) affinity(iterator(int i = 0:10:1, S1 * k = &this->a:&this->b) : this->c[i],this->d[(int)(k - &this->a)])
54 
55 class S8 : public S7<S1> {
56   S8() {}
57 
58 public:
59   S8(int v) : S7<S1>(v){
60 #pragma omp task private(a) private(this->a) private(S7<S1>::a)
61     for (int k = 0; k < a.a; ++k)
62       ++this->a.a;
63   }
64   S8 &operator=(S8 &s) {
65 #pragma omp task private(a) private(this->a)
66     for (int k = 0; k < s.a.a; ++k)
67       ++s.a.a;
68     return *this;
69   }
70 };
71 
72 // CHECK: #pragma omp task private(this->a) private(this->a) private(this->S7<S1>::a)
73 // CHECK: #pragma omp task private(this->a) private(this->a)
74 
75 template <class T>
76 struct S {
77   operator T() { return T(); }
78   static T TS;
79 #pragma omp threadprivate(TS)
80 };
81 
82 // CHECK:      template <class T> struct S {
83 // CHECK:        static T TS;
84 // CHECK-NEXT:   #pragma omp threadprivate(S::TS)
85 // CHECK:      };
86 // CHECK:      template<> struct S<int> {
87 // CHECK:        static int TS;
88 // CHECK-NEXT:   #pragma omp threadprivate(S<int>::TS)
89 // CHECK-NEXT: }
90 
91 template <typename T, int C>
92 T tmain(T argc, T *argv) {
93   T b = argc, c, d, e, f, g;
94   static T a;
95   S<T> s;
96   T arr[argc];
97   omp_depend_t x;
98   omp_event_handle_t evt;
99   double *arr_double;
100 #pragma omp task untied depend(in : argc, argv[b:argc], arr[:], ([argc][sizeof(T)])argv, arr_double[argc]) if (task : argc > 0) depend(depobj: x) detach(evt)
101   a = 2;
102 #pragma omp task default(none), private(argc, b) firstprivate(argv) shared(d) if (argc > 0) final(S<T>::TS > 0) priority(argc) affinity(argc, argv[b:argc], arr[:], ([argc][sizeof(T)])argv)
103   foo();
104 #pragma omp taskgroup task_reduction(-: argc)
105 #pragma omp task if (C) mergeable priority(C) in_reduction(-: argc)
106   foo();
107   return 0;
108 }
109 
110 // CHECK: template <typename T, int C> T tmain(T argc, T *argv) {
111 // CHECK-NEXT: T b = argc, c, d, e, f, g;
112 // CHECK-NEXT: static T a;
113 // CHECK-NEXT: S<T> s;
114 // CHECK-NEXT: T arr[argc];
115 // CHECK-NEXT: omp_depend_t x;
116 // CHECK-NEXT: omp_event_handle_t evt;
117 // CHECK-NEXT: double *arr_double;
118 // CHECK-NEXT: #pragma omp task untied depend(in : argc,argv[b:argc],arr[:],([argc][sizeof(T)])argv,arr_double[argc]) if(task: argc > 0) depend(depobj : x) detach(evt)
119 // CHECK-NEXT: a = 2;
120 // CHECK-NEXT: #pragma omp task default(none) private(argc,b) firstprivate(argv) shared(d) if(argc > 0) final(S<T>::TS > 0) priority(argc) affinity(argc,argv[b:argc],arr[:],([argc][sizeof(T)])argv)
121 // CHECK-NEXT: foo()
122 // CHECK-NEXT: #pragma omp taskgroup task_reduction(-: argc)
123 // CHECK-NEXT: #pragma omp task if(C) mergeable priority(C) in_reduction(-: argc)
124 // CHECK-NEXT: foo()
125 // CHECK: template<> int tmain<int, 5>(int argc, int *argv) {
126 // CHECK-NEXT: int b = argc, c, d, e, f, g;
127 // CHECK-NEXT: static int a;
128 // CHECK-NEXT: S<int> s;
129 // CHECK-NEXT: int arr[argc];
130 // CHECK-NEXT: omp_depend_t x;
131 // CHECK-NEXT: omp_event_handle_t evt;
132 // CHECK-NEXT: double *arr_double;
133 // CHECK-NEXT: #pragma omp task untied depend(in : argc,argv[b:argc],arr[:],([argc][sizeof(int)])argv,arr_double[argc]) if(task: argc > 0) depend(depobj : x) detach(evt)
134 // CHECK-NEXT: a = 2;
135 // CHECK-NEXT: #pragma omp task default(none) private(argc,b) firstprivate(argv) shared(d) if(argc > 0) final(S<int>::TS > 0) priority(argc) affinity(argc,argv[b:argc],arr[:],([argc][sizeof(int)])argv)
136 // CHECK-NEXT: foo()
137 // CHECK-NEXT: #pragma omp taskgroup task_reduction(-: argc)
138 // CHECK-NEXT: #pragma omp task if(5) mergeable priority(5) in_reduction(-: argc)
139 // CHECK-NEXT: foo()
140 // CHECK: template<> long tmain<long, 1>(long argc, long *argv) {
141 // CHECK-NEXT: long b = argc, c, d, e, f, g;
142 // CHECK-NEXT: static long a;
143 // CHECK-NEXT: S<long> s;
144 // CHECK-NEXT: long arr[argc];
145 // CHECK-NEXT: omp_depend_t x;
146 // CHECK-NEXT: omp_event_handle_t evt;
147 // CHECK-NEXT: double *arr_double;
148 // CHECK-NEXT: #pragma omp task untied depend(in : argc,argv[b:argc],arr[:],([argc][sizeof(long)])argv,arr_double[argc]) if(task: argc > 0) depend(depobj : x) detach(evt)
149 // CHECK-NEXT: a = 2;
150 // CHECK-NEXT: #pragma omp task default(none) private(argc,b) firstprivate(argv) shared(d) if(argc > 0) final(S<long>::TS > 0) priority(argc) affinity(argc,argv[b:argc],arr[:],([argc][sizeof(long)])argv)
151 // CHECK-NEXT: foo()
152 // CHECK-NEXT: #pragma omp taskgroup task_reduction(-: argc)
153 // CHECK-NEXT: #pragma omp task if(1) mergeable priority(1) in_reduction(-: argc)
154 // CHECK-NEXT: foo()
155 
156 enum Enum {};
157 
158 int main(int argc, char **argv) {
159   long x;
160   int b = argc, c, d, e, f, g;
161   static int a;
162   int arr[10], arr1[argc];
163   omp_depend_t y;
164   omp_event_handle_t evt;
165 #pragma omp threadprivate(a)
166   Enum ee;
167 // CHECK: Enum ee;
168 #pragma omp task untied mergeable depend(out:argv[:a][1], (arr)[0:],([argc][10])argv,b) if(task: argc > 0) priority(f) depend(depobj:y)
169   // CHECK-NEXT: #pragma omp task untied mergeable depend(out : argv[:a][1],(arr)[0:],([argc][10])argv,b) if(task: argc > 0) priority(f) depend(depobj : y)
170   a = 2;
171 // CHECK-NEXT: a = 2;
172 #pragma omp taskgroup task_reduction(min: arr1)
173 #pragma omp task default(none), private(argc, b) firstprivate(argv, evt) if (argc > 0) final(a > 0) depend(inout : a, argv[:argc],arr[:a], ([10][argc])argv) priority(23) in_reduction(min: arr1), detach(evt)
174   // CHECK-NEXT: #pragma omp taskgroup task_reduction(min: arr1)
175   // CHECK-NEXT: #pragma omp task default(none) private(argc,b) firstprivate(argv,evt) if(argc > 0) final(a > 0) depend(inout : a,argv[:argc],arr[:a],([10][argc])argv) priority(23) in_reduction(min: arr1) detach(evt)
176   foo();
177   // CHECK-NEXT: foo();
178 #pragma omp taskgroup task_reduction(min: arr1)
179 #pragma omp parallel reduction(+:arr1)
180 #pragma omp task in_reduction(min: arr1) depend(iterator(i=0:argc, unsigned j=argc:0:a), out: argv[i][j])
181   // CHECK-NEXT: #pragma omp taskgroup task_reduction(min: arr1)
182   // CHECK-NEXT: #pragma omp parallel reduction(+: arr1)
183   // CHECK-NEXT: #pragma omp task in_reduction(min: arr1) depend(iterator(int i = 0:argc, unsigned int j = argc:0:a), out : argv[i][j])
184   foo();
185   // CHECK-NEXT: foo();
186   // CHECK-NEXT: #pragma omp task in_reduction(+: arr1)
187 #pragma omp task in_reduction(+: arr1)
188   foo();
189   // CHECK-NEXT: foo();
190   // CHECK-NEXT: #pragma omp task depend(out : arr,omp_all_memory)
191 #pragma omp task depend(out: omp_all_memory, arr)
192   foo();
193   // CHECK-NEXT: foo();
194   // CHECK-NEXT: #pragma omp task depend(inout : b,arr,a,x,omp_all_memory)
195 #pragma omp task depend(inout: b, arr, omp_all_memory, a, x)
196   foo();
197   // CHECK-NEXT: foo();
198   // CHECK-NEXT: #pragma omp task depend(inout : omp_all_memory)
199 #pragma omp task depend(inout: omp_all_memory)
200   foo();
201   // CHECK-NEXT: foo();
202   return tmain<int, 5>(b, &b) + tmain<long, 1>(x, &x);
203 }
204 
205 extern template int S<int>::TS;
206 extern template long S<long>::TS;
207 
208 // DUMP-LABEL:  FunctionDecl {{.*}} implicit_firstprivate
209 void
210 implicit_firstprivate() {
211 
212 #pragma omp parallel num_threads(1)
213   {
214     int i = 0;
215     // DUMP: OMPTaskDirective
216     // DUMP-NEXT: OMPFirstprivateClause
217     // DUMP-NOT: DeclRefExpr {{.+}} 'i' {{.+}} non_odr_use_unevaluated
218     // DUMP: DeclRefExpr {{.+}} 'i' 'int' refers_to_enclosing_variable_or_capture
219     // DUMP: CapturedStmt
220     // DUMP: BinaryOperator {{.+}} 'int' lvalue '='
221     // DUMP-NEXT: DeclRefExpr {{.+}} 'j' 'int'
222     // DUMP: DeclRefExpr {{.+}} 'i' {{.+}} non_odr_use_unevaluated
223     #pragma omp task
224     {
225 	int j = sizeof(i);
226 	j = i;
227     }
228   }
229 }
230 
231 // DUMP-LABEL:  FunctionDecl {{.*}} no_implicit_firstprivate
232 void
233 no_implicit_firstprivate() {
234 
235 #pragma omp parallel num_threads(1)
236   {
237     int i = 0;
238     // DUMP: OMPTaskDirective
239     // DUMP-NEXT: CapturedStmt
240     // DUMP: DeclRefExpr {{.+}} 'i' {{.+}} non_odr_use_unevaluated refers_to_enclosing_variable_or_capture
241     #pragma omp task
242     {
243 	int j = sizeof(i);
244     }
245   }
246 }
247 
248 #endif
249