Home | History | Annotate | Download | only in OpenMP
      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 = 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)
     41 template <>
     42 void 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 
     50 class 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 
     74 private:
     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: }
     91 template <int X, typename T>
     92 class TVV {
     93 public:
     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 
    121 private:
    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)
    131 template <int N>
    132 void foo(int (&b)[N], float *&c);
    133 
    134 // CHECK: TVV<16, float> t16;
    135 TVV<16, float> t16;
    136 
    137 void 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