libgomp/testsuite/libgomp.c-c++-common/taskloop-5.c

   1 /* { dg-do run } */
   2 /* { dg-options "-O2" } */
   3
   4 int u[64], v, w[64];
   5
   6 __attribute__((noinline, noclone)) int
   7 test (int a, int b, int c, int d, void (*fn) (int, int, int, int),
   8       int *num_tasks, int *min_iters, int *max_iters, int *sep)
   9 {
  10   int i, j, t = 0;
  11   __builtin_memset (u, 0, sizeof u);
  12   v = 0;
  13   fn (a, b, c, d);
  14   *min_iters = 0;
  15   *max_iters = 0;
  16   *num_tasks = v;
  17   *sep = v;
  18   if (v)
  19     {
  20       *min_iters = u[0];
  21       *max_iters = u[0];
  22       t = u[0];
  23       for (i = 1; i < v; i++)
  24         {
  25           if (*min_iters > u[i])
  26             *min_iters = u[i];
  27           if (*max_iters < u[i])
  28             *max_iters = u[i];
  29           t += u[i];
  30         }
  31       if (*min_iters != *max_iters)
  32         {
  33           for (i = 0; i < v - 1; i++)
  34             {
  35               int min_idx = i;
  36               for (j = i + 1; j < v; j++)
  37                 if (w[min_idx] > w[j])
  38                   min_idx = j;
  39               if (min_idx != i)
  40                 {
  41                   int tem = u[i];
  42                   u[i] = u[min_idx];
  43                   u[min_idx] = tem;
  44                   tem = w[i];
  45                   w[i] = w[min_idx];
  46                   w[min_idx] = tem;
  47                 }
  48             }
  49           if (u[0] != *max_iters)
  50             __builtin_abort ();
  51           for (i = 1; i < v; i++)
  52             if (u[i] != u[i - 1])
  53               {
  54                 if (*sep != v || u[i] != *min_iters)
  55                   __builtin_abort ();
  56                 *sep = i;
  57               }
  58         }
  59     }
  60   return t;
  61 }
  62
  63 void
  64 grainsize (int a, int b, int c, int d)
  65 {
  66   int i, j = 0, k = 0;
  67   #pragma omp taskloop firstprivate (j, k) grainsize(strict:d)
  68   for (i = a; i < b; i += c)
  69     {
  70       if (j == 0)
  71         {
  72           #pragma omp atomic capture
  73             k = v++;
  74           if (k >= 64)
  75             __builtin_abort ();
  76           w[k] = i;
  77         }
  78       u[k] = ++j;
  79     }
  80 }
  81
  82 void
  83 num_tasks (int a, int b, int c, int d)
  84 {
  85   int i, j = 0, k = 0;
  86   #pragma omp taskloop firstprivate (j, k) num_tasks(strict:d)
  87   for (i = a; i < b; i += c)
  88     {
  89       if (j == 0)
  90         {
  91           #pragma omp atomic capture
  92             k = v++;
  93           if (k >= 64)
  94             __builtin_abort ();
  95           w[k] = i;
  96         }
  97       u[k] = ++j;
  98     }
  99 }
 100
 101 int
 102 main ()
 103 {
 104   #pragma omp parallel
 105     #pragma omp single
 106       {
 107         int min_iters, max_iters, ntasks, sep;
 108         /* If grainsize is present and has strict modifier, # of task loop iters is == grainsize,
 109            except that it can be smaller on the last task.  */
 110         if (test (0, 79, 1, 17, grainsize, &ntasks, &min_iters, &max_iters, &sep) != 79
 111             || ntasks != 5 || min_iters != 11 || max_iters != 17 || sep != 4)
 112           __builtin_abort ();
 113         if (test (-49, 2541, 7, 28, grainsize, &ntasks, &min_iters, &max_iters, &sep) != 370
 114             || ntasks != 14 || min_iters != 6 || max_iters != 28 || sep != 13)
 115           __builtin_abort ();
 116         if (test (7, 21, 2, 15, grainsize, &ntasks, &min_iters, &max_iters, &sep) != 7
 117             || ntasks != 1 || min_iters != 7 || max_iters != 7 || sep != 1)
 118           __builtin_abort ();
 119         /* If num_tasks is present, # of tasks is min (# of loop iters, num_tasks)
 120            and each task has at least one iteration.  If strict modifier is present,
 121            first set of tasks has ceil (# of loop iters / num_tasks) iterations,
 122            followed by possibly empty set of tasks with floor (# of loop iters / num_tasks)
 123            iterations.  */
 124         if (test (-51, 2500, 48, 9, num_tasks, &ntasks, &min_iters, &max_iters, &sep) != 54
 125             || ntasks != 9 || min_iters != 6 || max_iters != 6 || sep != 9)
 126           __builtin_abort ();
 127         if (test (0, 57, 1, 9, num_tasks, &ntasks, &min_iters, &max_iters, &sep) != 57
 128             || ntasks != 9 || min_iters != 6 || max_iters != 7 || sep != 3)
 129           __builtin_abort ();
 130         if (test (0, 25, 2, 17, num_tasks, &ntasks, &min_iters, &max_iters, &sep) != 13
 131             || ntasks != 13 || min_iters != 1 || max_iters != 1 || sep != 13)
 132           __builtin_abort ();
 133       }
 134   return 0;
 135 }