• Home
  • Line#
  • Scopes#
  • Navigate#
  • Raw
  • Download
1 // RUN: %clang_cc1 -verify -fopenmp -ast-print %s -Wno-openmp-mapping | 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 -fsyntax-only -verify %s -ast-print -Wno-openmp-mapping | FileCheck %s
4 
5 // RUN: %clang_cc1 -verify -fopenmp-simd -ast-print %s -Wno-openmp-mapping | 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 -fsyntax-only -verify %s -ast-print -Wno-openmp-mapping | FileCheck %s
8 // expected-no-diagnostics
9 
10 #ifndef HEADER
11 #define HEADER
12 
13 typedef void **omp_allocator_handle_t;
14 extern const omp_allocator_handle_t omp_null_allocator;
15 extern const omp_allocator_handle_t omp_default_mem_alloc;
16 extern const omp_allocator_handle_t omp_large_cap_mem_alloc;
17 extern const omp_allocator_handle_t omp_const_mem_alloc;
18 extern const omp_allocator_handle_t omp_high_bw_mem_alloc;
19 extern const omp_allocator_handle_t omp_low_lat_mem_alloc;
20 extern const omp_allocator_handle_t omp_cgroup_mem_alloc;
21 extern const omp_allocator_handle_t omp_pteam_mem_alloc;
22 extern const omp_allocator_handle_t omp_thread_mem_alloc;
23 
foo()24 void foo() {}
25 
26 struct S {
SS27   S(): a(0) {}
SS28   S(int v) : a(v) {}
29   int a;
30   typedef int type;
31 };
32 
33 template <typename T>
34 class S7 : public T {
35 protected:
36   T a;
S7()37   S7() : a(0) {}
38 
39 public:
S7(typename T::type v)40   S7(typename T::type v) : a(v) {
41 #pragma omp target
42 #pragma omp teams distribute private(a) private(this->a) private(T::a)
43     for (int k = 0; k < a.a; ++k)
44       ++this->a.a;
45   }
operator =(S7 & s)46   S7 &operator=(S7 &s) {
47 #pragma omp target
48 #pragma omp teams distribute private(a) private(this->a)
49     for (int k = 0; k < s.a.a; ++k)
50       ++s.a.a;
51     return *this;
52   }
53 
foo()54   void foo() {
55     int b, argv, d, c, e, f;
56 #pragma omp target
57 #pragma omp teams distribute default(none), private(b) firstprivate(argv) shared(d) reduction(+:c) reduction(max:e) num_teams(f) thread_limit(d)
58     for (int k = 0; k < a.a; ++k)
59       ++a.a;
60   }
61 };
62 // CHECK: #pragma omp target
63 // CHECK-NEXT: #pragma omp teams distribute private(this->a) private(this->a) private(T::a)
64 // CHECK: #pragma omp target
65 // CHECK-NEXT: #pragma omp teams distribute private(this->a) private(this->a)
66 // CHECK: #pragma omp target
67 // CHECK-NEXT: #pragma omp teams distribute default(none) private(b) firstprivate(argv) shared(d) reduction(+: c) reduction(max: e) num_teams(f) thread_limit(d)
68 // CHECK: #pragma omp target
69 // CHECK-NEXT: #pragma omp teams distribute private(this->a) private(this->a) private(this->S::a)
70 
71 class S8 : public S7<S> {
S8()72   S8() {}
73 
74 public:
S8(int v)75   S8(int v) : S7<S>(v){
76 #pragma omp target
77 #pragma omp teams distribute private(a) private(this->a) private(S7<S>::a)
78     for (int k = 0; k < a.a; ++k)
79       ++this->a.a;
80   }
operator =(S8 & s)81   S8 &operator=(S8 &s) {
82 #pragma omp target
83 #pragma omp teams distribute private(a) private(this->a)
84     for (int k = 0; k < s.a.a; ++k)
85       ++s.a.a;
86     return *this;
87   }
88 
bar()89   void bar() {
90     int b, argv, d, c, e, f;
91 #pragma omp target
92 #pragma omp teams distribute allocate(omp_thread_mem_alloc:argv) default(none), private(b) firstprivate(argv) shared(d) reduction(+:c) reduction(max:e) num_teams(f) thread_limit(d) allocate(omp_default_mem_alloc:c) shared(omp_default_mem_alloc, omp_thread_mem_alloc)
93     for (int k = 0; k < a.a; ++k)
94       ++a.a;
95   }
96 };
97 // CHECK: #pragma omp target
98 // CHECK-NEXT: #pragma omp teams distribute private(this->a) private(this->a) private(this->S7<S>::a)
99 // CHECK: #pragma omp target
100 // CHECK-NEXT: #pragma omp teams distribute private(this->a) private(this->a)
101 // CHECK: #pragma omp target
102 // CHECK-NEXT: #pragma omp teams distribute allocate(omp_thread_mem_alloc: argv) default(none) private(b) firstprivate(argv) shared(d) reduction(+: c) reduction(max: e) num_teams(f) thread_limit(d) allocate(omp_default_mem_alloc: c) shared(omp_default_mem_alloc,omp_thread_mem_alloc)
103 
104 template <class T, int N>
tmain(T argc)105 T tmain(T argc) {
106   T b = argc, c, d, e, f, g;
107   static T a;
108 // CHECK: static T a;
109 #pragma omp target
110 #pragma omp teams distribute
111   for (int i=0; i < 2; ++i)
112     a = 2;
113 // CHECK: #pragma omp target
114 // CHECK-NEXT: #pragma omp teams distribute{{$}}
115 // CHECK-NEXT: for (int i = 0; i < 2; ++i)
116 // CHECK-NEXT: a = 2;
117 #pragma omp target
118 #pragma omp teams distribute private(argc, b), firstprivate(c, d), collapse(2)
119   for (int i = 0; i < 10; ++i)
120     for (int j = 0; j < 10; ++j)
121       foo();
122 // CHECK: #pragma omp target
123 // CHECK-NEXT: #pragma omp teams distribute private(argc,b) firstprivate(c,d) collapse(2)
124 // CHECK-NEXT: for (int i = 0; i < 10; ++i)
125 // CHECK-NEXT: for (int j = 0; j < 10; ++j)
126 // CHECK-NEXT: foo();
127   for (int i = 0; i < 10; ++i)
128     foo();
129 // CHECK: for (int i = 0; i < 10; ++i)
130 // CHECK-NEXT: foo();
131 #pragma omp target
132 #pragma omp teams distribute
133   for (int i = 0; i < 10; ++i)
134     foo();
135 // CHECK: #pragma omp target
136 // CHECK-NEXT: #pragma omp teams distribute
137 // CHECK-NEXT: for (int i = 0; i < 10; ++i)
138 // CHECK-NEXT: foo();
139 #pragma omp target
140 #pragma omp teams distribute default(none), private(b) firstprivate(argc) shared(d) reduction(+:c) reduction(max:e) num_teams(f) thread_limit(d)
141     for (int k = 0; k < 10; ++k)
142       e += d + argc;
143 // CHECK: #pragma omp target
144 // CHECK-NEXT: #pragma omp teams distribute default(none) private(b) firstprivate(argc) shared(d) reduction(+: c) reduction(max: e) num_teams(f) thread_limit(d)
145 // CHECK-NEXT: for (int k = 0; k < 10; ++k)
146 // CHECK-NEXT: e += d + argc;
147   return T();
148 }
149 
main(int argc,char ** argv)150 int main (int argc, char **argv) {
151   int b = argc, c, d, e, f, g;
152   static int a;
153 // CHECK: static int a;
154 #pragma omp target
155 #pragma omp teams distribute
156   for (int i=0; i < 2; ++i)
157     a = 2;
158 // CHECK: #pragma omp target
159 // CHECK-NEXT: #pragma omp teams distribute
160 // CHECK-NEXT: for (int i = 0; i < 2; ++i)
161 // CHECK-NEXT: a = 2;
162 #pragma omp target
163 #pragma omp teams distribute private(argc,b),firstprivate(argv, c), collapse(2)
164   for (int i = 0; i < 10; ++i)
165     for (int j = 0; j < 10; ++j)
166       foo();
167 // CHECK: #pragma omp target
168 // CHECK-NEXT: #pragma omp teams distribute private(argc,b) firstprivate(argv,c) collapse(2)
169 // CHECK-NEXT: for (int i = 0; i < 10; ++i)
170 // CHECK-NEXT: for (int j = 0; j < 10; ++j)
171 // CHECK-NEXT: foo();
172   for (int i = 0; i < 10; ++i)
173     foo();
174 // CHECK: for (int i = 0; i < 10; ++i)
175 // CHECK-NEXT: foo();
176 #pragma omp target
177 #pragma omp teams distribute
178   for (int i = 0; i < 10; ++i)foo();
179 // CHECK: #pragma omp target
180 // CHECK-NEXT: #pragma omp teams distribute
181 // CHECK-NEXT: for (int i = 0; i < 10; ++i)
182 // CHECK-NEXT: foo();
183 #pragma omp target
184 #pragma omp teams distribute 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
188 // CHECK-NEXT: #pragma omp teams distribute default(none) private(b) firstprivate(argc) shared(d) reduction(+: c) reduction(max: e) num_teams(f) thread_limit(d)
189 // CHECK-NEXT: for (int k = 0; k < 10; ++k)
190 // CHECK-NEXT: e += d + argc;
191   return (0);
192 }
193 
194 #endif
195