libgomp/testsuite/libgomp.oacc-c++/template-reduction.C

   1 const int n = 100;
   2
   3 // Check explicit template copy map
   4
   5 template<typename T> T
   6 sum (T array[])
   7 {
   8    T s = 0;
   9
  10 #pragma acc parallel loop num_gangs (10) gang reduction (+:s) copy (array[0:n])
  11   for (int i = 0; i < n; i++)
  12     s += array[i];
  13
  14   return s;
  15 }
  16
  17 // Check implicit template copy map
  18
  19 template<typename T> T
  20 sum ()
  21 {
  22   T s = 0;
  23   T array[n];
  24
  25   for (int i = 0; i < n; i++)
  26     array[i] = i+1;
  27
  28 #pragma acc parallel loop num_gangs (10) gang reduction (+:s)
  29   for (int i = 0; i < n; i++)
  30     s += array[i];
  31
  32   return s;
  33 }
  34
  35 // Check template with default (present)
  36
  37 template<typename T> T
  38 sum_default_present ()
  39 {
  40   T s = 0;
  41   T array[n];
  42
  43   for (int i = 0; i < n; i++)
  44     array[i] = i+1;
  45
  46 #pragma acc enter data copyin (array)
  47
  48 #pragma acc parallel loop num_gangs (10) gang reduction (+:s) default (present)
  49   for (int i = 0; i < n; i++)
  50     s += array[i];
  51
  52 #pragma acc exit data delete (array)
  53
  54   return s;
  55 }
  56
  57 // Check present and async
  58
  59 template<typename T> T
  60 async_sum (T array[])
  61 {
  62    T s = 0;
  63
  64 #pragma acc parallel loop num_gangs (10) gang async (1) present (array[0:n])
  65    for (int i = 0; i < n; i++)
  66      array[i] = i+1;
  67
  68 #pragma acc parallel loop num_gangs (10) gang reduction (+:s) present (array[0:n]) async wait (1)
  69   for (int i = 0; i < n; i++)
  70     s += array[i];
  71
  72 #pragma acc wait
  73
  74   return s;
  75 }
  76
  77 // Check present and async and an explicit firstprivate
  78
  79 template<typename T> T
  80 async_sum (int c)
  81 {
  82    T s = 0;
  83
  84 #pragma acc parallel loop num_gangs (10) gang reduction (+:s) firstprivate (c) async wait (1)
  85   for (int i = 0; i < n; i++)
  86     s += i+c;
  87
  88 #pragma acc wait
  89
  90   return s;
  91 }
  92
  93 int
  94 main()
  95 {
  96   int a[n];
  97   int result = 0;
  98
  99   for (int i = 0; i < n; i++)
 100     {
 101       a[i] = i+1;
 102       result += i+1;
 103     }
 104
 105   if (sum (a) != result)
 106     __builtin_abort ();
 107
 108   if (sum<int> () != result)
 109     __builtin_abort ();
 110
 111   if (sum_default_present<int> () != result)
 112     __builtin_abort ();
 113
 114 #pragma acc enter data copyin (a)
 115   if (async_sum (a) != result)
 116     __builtin_abort ();
 117
 118   if (async_sum<int> (1) != result)
 119     __builtin_abort ();
 120 #pragma acc exit data delete (a)
 121
 122   return 0;
 123 }