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