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 
foo()13 void foo() {}
14 
15 struct S {
SS16   S(): a(0) {}
SS17   S(int v) : a(v) {}
18   int a;
19   typedef int type;
20 };
21 
22 template <typename T>
23 class S7 : public T {
24 protected:
25   T a;
S7()26   S7() : a(0) {}
27 
28 public:
S7(typename T::type v)29   S7(typename T::type v) : a(v) {
30 #pragma omp target
31 #pragma omp teams
32 #pragma omp distribute private(a) private(this->a) private(T::a) allocate(a)
33     for (int k = 0; k < a.a; ++k)
34       ++this->a.a;
35   }
operator =(S7 & s)36   S7 &operator=(S7 &s) {
37 #pragma omp target
38 #pragma omp teams
39 #pragma omp distribute allocate(a) private(a) private(this->a)
40     for (int k = 0; k < s.a.a; ++k)
41       ++s.a.a;
42     return *this;
43   }
44 };
45 
46 // CHECK: #pragma omp target
47 // CHECK-NEXT: #pragma omp teams
48 // CHECK-NEXT: #pragma omp distribute private(this->a) private(this->a) private(T::a) allocate(this->a)
49 // CHECK: #pragma omp target
50 // CHECK-NEXT: #pragma omp teams
51 // CHECK-NEXT: #pragma omp distribute allocate(this->a) private(this->a) private(this->a)
52 // CHECK: #pragma omp target
53 // CHECK-NEXT: #pragma omp teams
54 // CHECK-NEXT: #pragma omp distribute private(this->a) private(this->a) private(this->S::a) allocate(this->a)
55 
56 class S8 : public S7<S> {
S8()57   S8() {}
58 
59 public:
S8(int v)60   S8(int v) : S7<S>(v){
61 #pragma omp target
62 #pragma omp teams
63 #pragma omp distribute private(a) private(this->a) private(S7<S>::a)
64     for (int k = 0; k < a.a; ++k)
65       ++this->a.a;
66   }
operator =(S8 & s)67   S8 &operator=(S8 &s) {
68 #pragma omp target
69 #pragma omp teams
70 #pragma omp distribute private(a) private(this->a)
71     for (int k = 0; k < s.a.a; ++k)
72       ++s.a.a;
73     return *this;
74   }
75 };
76 
77 // CHECK: #pragma omp target
78 // CHECK-NEXT: #pragma omp teams
79 // CHECK-NEXT: #pragma omp distribute private(this->a) private(this->a) private(this->S7<S>::a)
80 // CHECK: #pragma omp target
81 // CHECK-NEXT: #pragma omp teams
82 // CHECK-NEXT: #pragma omp distribute private(this->a) private(this->a)
83 
84 template <class T, int N>
85 T tmain(T argc) {
86   T b = argc, c, d, e, f, g;
87   static T a;
88 // CHECK: static T a;
89 #pragma omp distribute
90 // CHECK-NEXT: #pragma omp distribute{{$}}
91   for (int i=0; i < 2; ++i)a=2;
92 // CHECK-NEXT: for (int i = 0; i < 2; ++i)
93 // CHECK-NEXT: a = 2;
94 #pragma omp target
95 #pragma omp teams
96 #pragma omp distribute private(argc, b), firstprivate(c, d), collapse(2)
97   for (int i = 0; i < 10; ++i)
98   for (int j = 0; j < 10; ++j)foo();
99 // CHECK-NEXT: #pragma omp target
100 // CHECK-NEXT: #pragma omp teams
101 // CHECK-NEXT: #pragma omp distribute private(argc,b) firstprivate(c,d) collapse(2)
102 // CHECK-NEXT: for (int i = 0; i < 10; ++i)
103 // CHECK-NEXT: for (int j = 0; j < 10; ++j)
104 // CHECK-NEXT: foo();
105   for (int i = 0; i < 10; ++i)foo();
106 // CHECK-NEXT: for (int i = 0; i < 10; ++i)
107 // CHECK-NEXT: foo();
108 #pragma omp distribute
109 // CHECK: #pragma omp distribute
110   for (int i = 0; i < 10; ++i)foo();
111 // CHECK-NEXT: for (int i = 0; i < 10; ++i)
112 // CHECK-NEXT: foo();
113   return T();
114 }
115 
main(int argc,char ** argv)116 int main (int argc, char **argv) {
117   int b = argc, c, d, e, f, g;
118   static int a;
119 // CHECK: static int a;
120 #pragma omp distribute
121 // CHECK-NEXT: #pragma omp distribute
122   for (int i=0; i < 2; ++i)a=2;
123 // CHECK-NEXT: for (int i = 0; i < 2; ++i)
124 // CHECK-NEXT: a = 2;
125 #pragma omp target
126 #pragma omp teams
127 #pragma omp distribute private(argc,b),firstprivate(argv, c), collapse(2)
128   for (int i = 0; i < 10; ++i)
129   for (int j = 0; j < 10; ++j)foo();
130 // CHECK-NEXT: #pragma omp target
131 // CHECK-NEXT: #pragma omp teams
132 // CHECK-NEXT: #pragma omp distribute private(argc,b) firstprivate(argv,c) collapse(2)
133 // CHECK-NEXT: for (int i = 0; i < 10; ++i)
134 // CHECK-NEXT: for (int j = 0; j < 10; ++j)
135 // CHECK-NEXT: foo();
136   for (int i = 0; i < 10; ++i)foo();
137 // CHECK-NEXT: for (int i = 0; i < 10; ++i)
138 // CHECK-NEXT: foo();
139 #pragma omp distribute
140 // CHECK: #pragma omp distribute
141   for (int i = 0; i < 10; ++i)foo();
142 // CHECK-NEXT: for (int i = 0; i < 10; ++i)
143 // CHECK-NEXT: foo();
144   return (0);
145 }
146 
147 #endif
148