1 // RUN: %clang_cc1 -verify -fopenmp -x c++ -std=c++11 -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 -x c++ -std=c++11 -include-pch %t -fsyntax-only -verify %s -ast-print | FileCheck %s
4 // expected-no-diagnostics
5 
6 #ifndef HEADER
7 #define HEADER
8 
9 #pragma omp declare simd linear(d: 8)
10 #pragma omp declare simd inbranch simdlen(32)
11 #pragma omp declare simd notinbranch
12 void add_1(float *d) __attribute__((cold));
13 
14 // CHECK: #pragma omp declare simd notinbranch
15 // CHECK-NEXT: #pragma omp declare simd inbranch simdlen(32)
16 // CHECK-NEXT: #pragma omp declare simd linear(val(d): 8)
17 // CHECK-NEXT: void add_1(float *d) __attribute__((cold));
18 //
19 
20 #pragma omp declare simd aligned(hp, hp2)
21 template <class C> void h(C *hp, C *hp2, C *hq, C *lin) {
22 }
23 
24 // CHECK: #pragma omp declare simd aligned(hp) aligned(hp2)
25 // CHECK-NEXT: template <class C> void h(C *hp, C *hp2, C *hq, C *lin) {
26 // CHECK-NEXT: }
27 
28 // CHECK: #pragma omp declare simd aligned(hp) aligned(hp2)
29 // CHECK-NEXT: template<> void h<float>(float *hp, float *hp2, float *hq, float *lin) {
30 // CHECK-NEXT: }
31 
32 // CHECK-NEXT: template<> void h<int>(int *hp, int *hp2, int *hq, int *lin) {
33 // CHECK-NEXT: h((float *)hp, (float *)hp2, (float *)hq, (float *)lin);
34 // CHECK-NEXT: }
35 
36 // Explicit specialization with <C=int>.
37 // Pragmas need to be same, otherwise standard says that's undefined behavior.
38 #pragma omp declare simd aligned(hp, hp2)
39 template <>
40 void h(int *hp, int *hp2, int *hq, int *lin)
41 {
42   // Implicit specialization with <C=float>.
43   // This is special case where the directive is stored by Sema and is
44   // generated together with the (pending) function instatiation.
45   h((float*) hp, (float*) hp2, (float*) hq, (float*) lin);
46 }
47 
48 class VV {
49   // CHECK: #pragma omp declare simd uniform(this, a) linear(val(b): a)
50   // CHECK-NEXT: int add(int a, int b) __attribute__((cold))    {
51   // CHECK-NEXT: return a + b;
52   // CHECK-NEXT: }
53   #pragma omp declare simd uniform(this, a) linear(val(b): a)
54   int add(int a, int b) __attribute__((cold)) { return a + b; }
55 
56   // CHECK: #pragma omp declare simd aligned(b: 4) aligned(a) linear(ref(b): 4) linear(val(this)) linear(val(a))
57   // CHECK-NEXT: float taddpf(float *a, float *&b)     {
58   // CHECK-NEXT: return *a + *b;
59   // CHECK-NEXT: }
60   #pragma omp declare simd aligned (b: 4) aligned(a) linear(ref(b): 4) linear(this, a)
61   float taddpf(float *a, float *&b) { return *a + *b; }
62 
63 // CHECK: #pragma omp declare simd aligned(b: 8)
64 // CHECK-NEXT: #pragma omp declare simd linear(uval(c): 8)
65 // CHECK-NEXT: int tadd(int (&b)[], int &c) {
66 // CHECK-NEXT: return this->x[b[0]] + b[0];
67 // CHECK-NEXT: }
68   #pragma omp declare simd linear(uval(c): 8)
69   #pragma omp declare simd aligned(b : 8)
70   int tadd(int (&b)[], int &c) { return x[b[0]] + b[0]; }
71 
72 private:
73   int x[10];
74 };
75 
76 // CHECK: template <int X, typename T> class TVV {
77 // CHECK: #pragma omp declare simd simdlen(X)
78 // CHECK-NEXT: int tadd(int a, int b) {
79 // CHECK: #pragma omp declare simd aligned(a: X * 2) aligned(b) linear(ref(b): X)
80 // CHECK-NEXT: float taddpf(float *a, T *&b) {
81 // CHECK-NEXT: return *a + *b;
82 // CHECK-NEXT: }
83 // CHECK: #pragma omp declare simd
84 // CHECK-NEXT: #pragma omp declare simd
85 // CHECK-NEXT: int tadd(int b) {
86 // CHECK-NEXT: return this->x[b] + b;
87 // CHECK-NEXT: }
88 // CHECK: }
89 template <int X, typename T>
90 class TVV {
91 public:
92 // CHECK: template<> class TVV<16, float> {
93   #pragma omp declare simd simdlen(X)
94   int tadd(int a, int b) { return a + b; }
95 
96 // CHECK: #pragma omp declare simd simdlen(16)
97 // CHECK-NEXT: int tadd(int a, int b);
98 
99   #pragma omp declare simd aligned(a : X * 2) aligned(b) linear(ref(b): X)
100   float taddpf(float *a, T *&b) { return *a + *b; }
101 
102 // CHECK: #pragma omp declare simd aligned(a: 16 * 2) aligned(b) linear(ref(b): 16)
103 // CHECK-NEXT: float taddpf(float *a, float *&b) {
104 // CHECK-NEXT: return *a + *b;
105 // CHECK-NEXT: }
106 
107   #pragma omp declare simd
108   #pragma omp declare simd uniform(this, b)
109   int tadd(int b) { return x[b] + b; }
110 
111 // CHECK: #pragma omp declare simd uniform(this, b)
112 // CHECK-NEXT: #pragma omp declare simd
113 // CHECK-NEXT: int tadd(int b) {
114 // CHECK-NEXT: return this->x[b] + b;
115 // CHECK-NEXT: }
116 
117 private:
118   int x[X];
119 };
120 // CHECK: };
121 
122 // CHECK: #pragma omp declare simd simdlen(N) aligned(b: N * 2) linear(uval(c): N)
123 // CHECK: template <int N> void foo(int (&b)[N], float *&c)
124 // CHECK: #pragma omp declare simd simdlen(64) aligned(b: 64 * 2) linear(uval(c): 64)
125 // CHECK: template<> void foo<64>(int (&b)[64], float *&c)
126 #pragma omp declare simd simdlen(N) aligned(b : N * 2) linear(uval(c): N)
127 template <int N>
128 void foo(int (&b)[N], float *&c);
129 
130 // CHECK: TVV<16, float> t16;
131 TVV<16, float> t16;
132 
133 void f() {
134   float a = 1.0f, b = 2.0f;
135   float *p = &b;
136   float r = t16.taddpf(&a, p);
137   int res = t16.tadd(b);
138   int c[64];
139   foo(c, p);
140 }
141 
142 #endif
143