1 // RUN: %clang_cc1 -verify -fopenmp -ast-print %s | FileCheck %s --check-prefix=CHECK --check-prefix=OMP45
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 | FileCheck %s --check-prefix=CHECK --check-prefix=OMP45
4 // RUN: %clang_cc1 -verify -fopenmp -fopenmp-version=50 -ast-print %s -DOMP5 | FileCheck %s --check-prefix=CHECK --check-prefix=OMP50
5 // RUN: %clang_cc1 -fopenmp -fopenmp-version=50 -x c++ -std=c++11 -emit-pch -o %t %s -DOMP5
6 // RUN: %clang_cc1 -fopenmp -fopenmp-version=50 -std=c++11 -include-pch %t -fsyntax-only -verify %s -ast-print -DOMP5 | FileCheck %s --check-prefix=CHECK --check-prefix=OMP50
7 
8 // RUN: %clang_cc1 -verify -fopenmp-simd -ast-print %s | FileCheck %s --check-prefix=CHECK --check-prefix=OMP45
9 // RUN: %clang_cc1 -fopenmp-simd -x c++ -std=c++11 -emit-pch -o %t %s
10 // RUN: %clang_cc1 -fopenmp-simd -std=c++11 -include-pch %t -fsyntax-only -verify %s -ast-print | FileCheck %s --check-prefix=CHECK --check-prefix=OMP45
11 // RUN: %clang_cc1 -verify -fopenmp-simd -fopenmp-version=50 -ast-print %s -DOMP5 | FileCheck %s --check-prefix=CHECK --check-prefix=OMP50
12 // RUN: %clang_cc1 -fopenmp-simd -fopenmp-version=50 -x c++ -std=c++11 -emit-pch -o %t %s -DOMP5
13 // RUN: %clang_cc1 -fopenmp-simd -fopenmp-version=50 -std=c++11 -include-pch %t -fsyntax-only -verify %s -ast-print -DOMP5 | FileCheck %s --check-prefix=CHECK --check-prefix=OMP50
14 // expected-no-diagnostics
15 
16 #ifndef HEADER
17 #define HEADER
18 
19 struct SS {
20   SS(): a(0) {}
21   SS(int v) : a(v) {}
22   int a;
23   typedef int type;
24 };
25 
26 template <typename T>
27 class S7 : public T {
28 protected:
29   T *a;
30   T b[2];
31   S7() : a(0) {}
32 
33 public:
34   S7(typename T::type &v) : a((T*)&v) {
35 #pragma omp simd aligned(a)
36     for (int k = 0; k < a->a; ++k)
37       ++this->a->a;
38   }
39   S7 &operator=(S7 &s) {
40 #pragma omp simd aligned(this->b : 8)
41     for (int k = 0; k < s.a->a; ++k)
42       ++s.a->a;
43     return *this;
44   }
45 };
46 
47 // CHECK: #pragma omp simd aligned(this->a){{$}}
48 // CHECK: #pragma omp simd aligned(this->b: 8)
49 // CHECK: #pragma omp simd aligned(this->a)
50 
51 class S8 : public S7<SS> {
52   S8() {}
53 
54 public:
55   S8(int v) : S7<SS>(v){
56 #pragma omp simd aligned(S7<SS>::a)
57     for (int k = 0; k < a->a; ++k)
58       ++this->a->a;
59   }
60   S8 &operator=(S8 &s) {
61 #pragma omp simd aligned(this->b: 4)
62     for (int k = 0; k < s.a->a; ++k)
63       ++s.a->a;
64     return *this;
65   }
66 };
67 
68 // CHECK: #pragma omp simd aligned(this->S7<SS>::a)
69 // CHECK: #pragma omp simd aligned(this->b: 4)
70 
71 void foo() {}
72 int g_ind = 1;
73 template<class T, class N> T reduct(T* arr, N num) {
74   N i;
75   N ind;
76   N myind;
77   N &ref = i;
78   T sum = (T)0;
79 // CHECK: T sum = (T)0;
80 #pragma omp simd private(myind, g_ind), linear(ind), aligned(arr), linear(uval(ref))
81 // CHECK-NEXT: #pragma omp simd private(myind,g_ind) linear(ind) aligned(arr) linear(uval(ref))
82   for (i = 0; i < num; ++i) {
83     myind = ind;
84     T cur = arr[myind];
85     ind += g_ind;
86     sum += cur;
87   }
88 }
89 
90 template<class T> struct S {
91   S(const T &a)
92     :m_a(a)
93   {}
94   T result(T *v) const {
95     T res;
96     T val;
97     T lin = 0;
98     T &ref = res;
99 // CHECK: T res;
100 // CHECK: T val;
101 // CHECK: T lin = 0;
102 // CHECK: T &ref = res;
103     #pragma omp simd allocate(res) private(val)  safelen(7) linear(lin : -5) lastprivate(res) simdlen(5) linear(ref(ref))
104 // CHECK-NEXT: #pragma omp simd allocate(res) private(val) safelen(7) linear(lin: -5) lastprivate(res) simdlen(5) linear(ref(ref))
105     for (T i = 7; i < m_a; ++i) {
106       val = v[i-7] + m_a;
107       res = val;
108       lin -= 5;
109     }
110     const T clen = 3;
111 // CHECK: T clen = 3;
112     #pragma omp simd safelen(clen-1) simdlen(clen-1)
113 // CHECK-NEXT: #pragma omp simd safelen(clen - 1) simdlen(clen - 1)
114     for(T i = clen+2; i < 20; ++i) {
115 // CHECK-NEXT: for (T i = clen + 2; i < 20; ++i) {
116       v[i] = v[v-clen] + 1;
117 // CHECK-NEXT: v[i] = v[v - clen] + 1;
118     }
119 // CHECK-NEXT: }
120     return res;
121   }
122   ~S()
123   {}
124   T m_a;
125 };
126 
127 template<int LEN> struct S2 {
128   static void func(int n, float *a, float *b, float *c) {
129     int k1 = 0, k2 = 0;
130 #pragma omp simd safelen(LEN) linear(k1,k2:LEN) aligned(a:LEN) simdlen(LEN) allocate(k1)
131     for(int i = 0; i < n; i++) {
132       c[i] = a[i] + b[i];
133       c[k1] = a[k1] + b[k1];
134       c[k2] = a[k2] + b[k2];
135       k1 = k1 + LEN;
136       k2 = k2 + LEN;
137     }
138   }
139 };
140 
141 // S2<4>::func is called below in main.
142 // CHECK: template<> struct S2<4> {
143 // CHECK-NEXT: static void func(int n, float *a, float *b, float *c)     {
144 // CHECK-NEXT:   int k1 = 0, k2 = 0;
145 // CHECK-NEXT: #pragma omp simd safelen(4) linear(k1,k2: 4) aligned(a: 4) simdlen(4) allocate(k1)
146 // CHECK-NEXT:   for (int i = 0; i < n; i++) {
147 // CHECK-NEXT:     c[i] = a[i] + b[i];
148 // CHECK-NEXT:     c[k1] = a[k1] + b[k1];
149 // CHECK-NEXT:     c[k2] = a[k2] + b[k2];
150 // CHECK-NEXT:     k1 = k1 + 4;
151 // CHECK-NEXT:     k2 = k2 + 4;
152 // CHECK-NEXT:   }
153 // CHECK-NEXT: }
154 
155 int main (int argc, char **argv) {
156   int b = argc, c, d, e, f, g;
157   int k1=0,k2=0;
158   int &ref = b;
159   static int *a;
160 // CHECK: static int *a;
161 #pragma omp simd
162 // CHECK-NEXT: #pragma omp simd{{$}}
163   for (int i=0; i < 2; ++i)*a=2;
164 // CHECK-NEXT: for (int i = 0; i < 2; ++i)
165 // CHECK-NEXT: *a = 2;
166 #ifdef OMP5
167 #pragma omp simd private(argc, b),lastprivate(d,f) collapse(2) aligned(a : 4) if(simd:a) nontemporal(argc, c, d) lastprivate(conditional: e, f) order(concurrent)
168 // OMP50-NEXT: #pragma omp simd private(argc,b) lastprivate(d,f) collapse(2) aligned(a: 4) if(simd: a) nontemporal(argc,c,d) lastprivate(conditional: e,f) order(concurrent)
169 #else
170 #pragma omp simd private(argc, b),lastprivate(d,f) collapse(2) aligned(a : 4)
171 // OMP45-NEXT: #pragma omp simd private(argc,b) lastprivate(d,f) collapse(2) aligned(a: 4)
172 #endif // OMP5
173   for (int i = 0; i < 10; ++i)
174   for (int j = 0; j < 10; ++j) {foo(); k1 += 8; k2 += 8;}
175 // CHECK-NEXT: for (int i = 0; i < 10; ++i)
176 // CHECK-NEXT: for (int j = 0; j < 10; ++j) {
177 // CHECK-NEXT: foo();
178 // CHECK-NEXT: k1 += 8;
179 // CHECK-NEXT: k2 += 8;
180 // CHECK-NEXT: }
181   for (int i = 0; i < 10; ++i)foo();
182 // CHECK-NEXT: for (int i = 0; i < 10; ++i)
183 // CHECK-NEXT: foo();
184   const int CLEN = 4;
185 // CHECK-NEXT: const int CLEN = 4;
186 #ifdef OMP5
187   #pragma omp simd aligned(a:CLEN) linear(a:CLEN) safelen(CLEN) collapse( 1 ) simdlen(CLEN) linear(val(ref): CLEN) if(a)
188 // OMP50-NEXT: #pragma omp simd aligned(a: CLEN) linear(a: CLEN) safelen(CLEN) collapse(1) simdlen(CLEN) linear(val(ref): CLEN) if(a)
189 #else
190   #pragma omp simd aligned(a:CLEN) linear(a:CLEN) safelen(CLEN) collapse( 1 ) simdlen(CLEN) linear(val(ref): CLEN)
191 // OMP45-NEXT: #pragma omp simd aligned(a: CLEN) linear(a: CLEN) safelen(CLEN) collapse(1) simdlen(CLEN) linear(val(ref): CLEN)
192 #endif // OMP5
193   for (int i = 0; i < 10; ++i)foo();
194 // CHECK-NEXT: for (int i = 0; i < 10; ++i)
195 // CHECK-NEXT: foo();
196 
197   float arr[16];
198   S2<4>::func(0,arr,arr,arr);
199   return (0);
200 }
201 
202 #endif
203