test/OpenMP/target_teams_is_device_ptr_ast_print.cpp

   1 // RUN: %clang_cc1 -verify -fopenmp -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 -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 struct ST {
   9   int *a;
  10 };
  11 typedef int arr[10];
  12 typedef ST STarr[10];
  13 struct SA {
  14   const int da[5] = { 0 };
  15   ST g[10];
  16   STarr &rg = g;
  17   int i;
  18   int &j = i;
  19   int *k = &j;
  20   int *&z = k;
  21   int aa[10];
  22   arr &raa = aa;
  23   void func(int arg) {
  24 #pragma omp target teams is_device_ptr(k)
  25     {}
  26 #pragma omp target teams is_device_ptr(z)
  27     {}
  28 #pragma omp target teams is_device_ptr(aa) // OK
  29     {}
  30 #pragma omp target teams is_device_ptr(raa) // OK
  31     {}
  32 #pragma omp target teams is_device_ptr(g) // OK
  33     {}
  34 #pragma omp target teams is_device_ptr(rg) // OK
  35     {}
  36 #pragma omp target teams is_device_ptr(da) // OK
  37     {}
  38   return;
  39  }
  40 };
  41 // CHECK: struct SA
  42 // CHECK-NEXT: const int da[5] = {0};
  43 // CHECK-NEXT: ST g[10];
  44 // CHECK-NEXT: STarr &rg = this->g;
  45 // CHECK-NEXT: int i;
  46 // CHECK-NEXT: int &j = this->i;
  47 // CHECK-NEXT: int *k = &this->j;
  48 // CHECK-NEXT: int *&z = this->k;
  49 // CHECK-NEXT: int aa[10];
  50 // CHECK-NEXT: arr &raa = this->aa;
  51 // CHECK-NEXT: func(
  52 // CHECK-NEXT: #pragma omp target teams is_device_ptr(this->k)
  53 // CHECK-NEXT: {
  54 // CHECK-NEXT: }
  55 // CHECK-NEXT: #pragma omp target teams is_device_ptr(this->z)
  56 // CHECK-NEXT: {
  57 // CHECK-NEXT: }
  58 // CHECK-NEXT: #pragma omp target teams is_device_ptr(this->aa)
  59 // CHECK-NEXT: {
  60 // CHECK-NEXT: }
  61 // CHECK-NEXT: #pragma omp target teams is_device_ptr(this->raa)
  62 // CHECK-NEXT: {
  63 // CHECK-NEXT: }
  64 // CHECK-NEXT: #pragma omp target teams is_device_ptr(this->g)
  65 // CHECK-NEXT: {
  66 // CHECK-NEXT: }
  67 // CHECK-NEXT: #pragma omp target teams is_device_ptr(this->rg)
  68 // CHECK-NEXT: {
  69 // CHECK-NEXT: }
  70 // CHECK-NEXT: #pragma omp target teams is_device_ptr(this->da)
  71
  72 struct SB {
  73   unsigned A;
  74   unsigned B;
  75   float Arr[100];
  76   float *Ptr;
  77   float *foo() {
  78     return &Arr[0];
  79   }
  80 };
  81
  82 struct SC {
  83   unsigned A : 2;
  84   unsigned B : 3;
  85   unsigned C;
  86   unsigned D;
  87   float Arr[100];
  88   SB S;
  89   SB ArrS[100];
  90   SB *PtrS;
  91   SB *&RPtrS;
  92   float *Ptr;
  93
  94   SC(SB *&_RPtrS) : RPtrS(_RPtrS) {}
  95 };
  96
  97 union SD {
  98   unsigned A;
  99   float B;
 100 };
 101
 102 struct S1;
 103 extern S1 a;
 104 class S2 {
 105   mutable int a;
 106 public:
 107   S2():a(0) { }
 108   S2(S2 &s2):a(s2.a) { }
 109   static float S2s;
 110   static const float S2sc;
 111 };
 112 const float S2::S2sc = 0;
 113 const S2 b;
 114 const S2 ba[5];
 115 class S3 {
 116   int a;
 117 public:
 118   S3():a(0) { }
 119   S3(S3 &s3):a(s3.a) { }
 120 };
 121 const S3 c;
 122 const S3 ca[5];
 123 extern const int f;
 124 class S4 {
 125   int a;
 126   S4();
 127   S4(const S4 &s4);
 128 public:
 129   S4(int v):a(v) { }
 130 };
 131 class S5 {
 132   int a;
 133   S5():a(0) {}
 134   S5(const S5 &s5):a(s5.a) { }
 135 public:
 136   S5(int v):a(v) { }
 137 };
 138
 139 S3 h;
 140 #pragma omp threadprivate(h)
 141
 142 typedef struct {
 143   int a;
 144 } S6;
 145
 146 template <typename T>
 147 T tmain(T argc) {
 148   const T da[5] = { 0 };
 149   S6 h[10];
 150   auto &rh = h;
 151   T i;
 152   T &j = i;
 153   T *k = &j;
 154   T *&z = k;
 155   T aa[10];
 156   auto &raa = aa;
 157 #pragma omp target teams is_device_ptr(k)
 158   {}
 159 #pragma omp target teams is_device_ptr(z)
 160   {}
 161 #pragma omp target teams is_device_ptr(aa)
 162   {}
 163 #pragma omp target teams is_device_ptr(raa)
 164   {}
 165 #pragma omp target teams is_device_ptr(h)
 166   {}
 167 #pragma omp target teams is_device_ptr(rh)
 168   {}
 169 #pragma omp target teams is_device_ptr(da)
 170   {}
 171   return 0;
 172 }
 173
 174 // CHECK: template<> int tmain<int>(int argc) {
 175 // CHECK-NEXT: const int da[5] = {0};
 176 // CHECK-NEXT: S6 h[10];
 177 // CHECK-NEXT: auto &rh = h;
 178 // CHECK-NEXT: int i;
 179 // CHECK-NEXT: int &j = i;
 180 // CHECK-NEXT: int *k = &j;
 181 // CHECK-NEXT: int *&z = k;
 182 // CHECK-NEXT: int aa[10];
 183 // CHECK-NEXT: auto &raa = aa;
 184 // CHECK-NEXT: #pragma omp target teams is_device_ptr(k)
 185 // CHECK-NEXT: {
 186 // CHECK-NEXT: }
 187 // CHECK-NEXT: #pragma omp target teams is_device_ptr(z)
 188 // CHECK-NEXT: {
 189 // CHECK-NEXT: }
 190 // CHECK-NEXT: #pragma omp target teams is_device_ptr(aa)
 191 // CHECK-NEXT: {
 192 // CHECK-NEXT: }
 193 // CHECK-NEXT: #pragma omp target teams is_device_ptr(raa)
 194 // CHECK-NEXT: {
 195 // CHECK-NEXT: }
 196 // CHECK-NEXT: #pragma omp target teams is_device_ptr(h)
 197 // CHECK-NEXT: {
 198 // CHECK-NEXT: }
 199 // CHECK-NEXT: #pragma omp target teams is_device_ptr(rh)
 200 // CHECK-NEXT: {
 201 // CHECK-NEXT: }
 202 // CHECK-NEXT: #pragma omp target teams is_device_ptr(da)
 203
 204 // CHECK: template<> int *tmain<int *>(int *argc) {
 205 // CHECK-NEXT: int *const da[5] = {0};
 206 // CHECK-NEXT: S6 h[10];
 207 // CHECK-NEXT: auto &rh = h;
 208 // CHECK-NEXT: int *i;
 209 // CHECK-NEXT: int *&j = i;
 210 // CHECK-NEXT: int **k = &j;
 211 // CHECK-NEXT: int **&z = k;
 212 // CHECK-NEXT: int *aa[10];
 213 // CHECK-NEXT: auto &raa = aa;
 214 // CHECK-NEXT: #pragma omp target teams is_device_ptr(k)
 215 // CHECK-NEXT: {
 216 // CHECK-NEXT: }
 217 // CHECK-NEXT: #pragma omp target teams is_device_ptr(z)
 218 // CHECK-NEXT: {
 219 // CHECK-NEXT: }
 220 // CHECK-NEXT: #pragma omp target teams is_device_ptr(aa)
 221 // CHECK-NEXT: {
 222 // CHECK-NEXT: }
 223 // CHECK-NEXT: #pragma omp target teams is_device_ptr(raa)
 224 // CHECK-NEXT: {
 225 // CHECK-NEXT: }
 226 // CHECK-NEXT: #pragma omp target teams is_device_ptr(h)
 227 // CHECK-NEXT: {
 228 // CHECK-NEXT: }
 229 // CHECK-NEXT: #pragma omp target teams is_device_ptr(rh)
 230 // CHECK-NEXT: {
 231 // CHECK-NEXT: }
 232 // CHECK-NEXT: #pragma omp target teams is_device_ptr(da)
 233
 234 // CHECK-LABEL: int main(int argc, char **argv) {
 235 int main(int argc, char **argv) {
 236   const int da[5] = { 0 };
 237   S6 h[10];
 238   auto &rh = h;
 239   int i;
 240   int &j = i;
 241   int *k = &j;
 242   int *&z = k;
 243   int aa[10];
 244   auto &raa = aa;
 245 // CHECK-NEXT: const int da[5] = {0};
 246 // CHECK-NEXT: S6 h[10];
 247 // CHECK-NEXT: auto &rh = h;
 248 // CHECK-NEXT: int i;
 249 // CHECK-NEXT: int &j = i;
 250 // CHECK-NEXT: int *k = &j;
 251 // CHECK-NEXT: int *&z = k;
 252 // CHECK-NEXT: int aa[10];
 253 // CHECK-NEXT: auto &raa = aa;
 254 #pragma omp target teams is_device_ptr(k)
 255 // CHECK-NEXT: #pragma omp target teams is_device_ptr(k)
 256   {}
 257 // CHECK-NEXT: {
 258 // CHECK-NEXT: }
 259 #pragma omp target teams is_device_ptr(z)
 260 // CHECK-NEXT: #pragma omp target teams is_device_ptr(z)
 261   {}
 262 // CHECK-NEXT: {
 263 // CHECK-NEXT: }
 264 #pragma omp target teams is_device_ptr(aa)
 265 // CHECK-NEXT: #pragma omp target teams is_device_ptr(aa)
 266   {}
 267 // CHECK-NEXT: {
 268 // CHECK-NEXT: }
 269 #pragma omp target teams is_device_ptr(raa)
 270 // CHECK-NEXT: #pragma omp target teams is_device_ptr(raa)
 271   {}
 272 // CHECK-NEXT: {
 273 // CHECK-NEXT: }
 274 #pragma omp target teams is_device_ptr(h)
 275 // CHECK-NEXT: #pragma omp target teams is_device_ptr(h)
 276   {}
 277 // CHECK-NEXT: {
 278 // CHECK-NEXT: }
 279 #pragma omp target teams is_device_ptr(rh)
 280 // CHECK-NEXT: #pragma omp target teams is_device_ptr(rh)
 281   {}
 282 // CHECK-NEXT: {
 283 // CHECK-NEXT: }
 284 #pragma omp target teams is_device_ptr(da)
 285 // CHECK-NEXT: #pragma omp target teams is_device_ptr(da)
 286   {}
 287 // CHECK-NEXT: {
 288 // CHECK-NEXT: }
 289   return tmain<int>(argc) + *tmain<int *>(&argc);
 290 }
 291 #endif