libgomp/testsuite/libgomp.c++/ref-mapping-1.C

   1 /* { dg-do run } */
   2
   3 #include <cassert>
   4
   5 void test_ref ()
   6 {
   7   int c_orig = 5;
   8   int &c = c_orig;
   9
  10 #pragma omp target map(tofrom: c)
  11   {
  12     c++;
  13   }
  14
  15   assert (c == 6);
  16 }
  17
  18 void test_ref_to_ptr ()
  19 {
  20   int val = 5;
  21   int *ptr_orig = &val;
  22   int *&ptr_ref = ptr_orig;
  23
  24 #pragma omp target map(tofrom: ptr_ref[0])
  25   {
  26     (*ptr_ref)++;
  27   }
  28
  29   assert (val == 6);
  30 }
  31
  32 void test_ref_to_array ()
  33 {
  34   int arr[1];
  35   int (&arr_ref)[1] = arr;
  36
  37   arr_ref[0] = 5;
  38
  39 #pragma omp target map(tofrom: arr_ref[0:1])
  40   {
  41     arr_ref[0]++;
  42   }
  43
  44   assert (arr_ref[0] == 6);
  45
  46 #pragma omp target map(tofrom: arr_ref[0])
  47   {
  48     arr_ref[0]++;
  49   }
  50
  51   assert (arr_ref[0] == 7);
  52 }
  53
  54 void test_ref_to_ptr_array ()
  55 {
  56   int *arr[1];
  57   int *(&arr_ref)[1] = arr;
  58   int val = 5;
  59
  60   arr_ref[0] = &val;
  61
  62 #pragma omp target data map(alloc: arr_ref, arr_ref[0])
  63   {
  64 #pragma omp target map(tofrom: arr_ref[0][0:1])
  65     {
  66       arr_ref[0][0]++;
  67     }
  68   }
  69
  70   assert (arr_ref[0][0] == 6);
  71 }
  72
  73 int main ()
  74 {
  75   test_ref ();
  76   test_ref_to_ptr ();
  77   test_ref_to_array ();
  78   test_ref_to_ptr_array ();
  79   return 0;
  80 }