libgomp/testsuite/libgomp.c/loop-2.c

   1 /* Validate static scheduling iteration dispatch.  We only test with
   2    even thread distributions here; there are multiple valid solutions
   3    for uneven thread distributions.  */
   4
   5 /* { dg-require-effective-target sync_int_long } */
   6
   7 #include <omp.h>
   8 #include <string.h>
   9 #include <assert.h>
  10 #include "libgomp_g.h"
  11
  12
  13 #define N 360
  14 static int data[N][2];
  15 static int INCR, NTHR, CHUNK;
  16
  17 static void clean_data (void)
  18 {
  19   memset (data, -1, sizeof (data));
  20 }
  21
  22 static void test_data (void)
  23 {
  24   int n, i, c, thr, iter, chunk;
  25
  26   chunk = CHUNK;
  27   if (chunk == 0)
  28     chunk = N / INCR / NTHR;
  29
  30   thr = iter = c = i = 0;
  31
  32   for (n = 0; n < N; ++n)
  33     {
  34       if (i == 0)
  35         {
  36           assert (data[n][0] == thr);
  37           assert (data[n][1] == iter);
  38         }
  39       else
  40         {
  41           assert (data[n][0] == -1);
  42           assert (data[n][1] == -1);
  43         }
  44
  45       if (++i == INCR)
  46         {
  47           i = 0;
  48           if (++c == chunk)
  49             {
  50               c = 0;
  51               if (++thr == NTHR)
  52                 {
  53                   thr = 0;
  54                   ++iter;
  55                 }
  56             }
  57         }
  58     }
  59 }
  60
  61 static void set_data (long i, int thr, int iter)
  62 {
  63   int old;
  64   assert (i >= 0 && i < N);
  65   old = __sync_lock_test_and_set (&data[i][0], thr);
  66   assert (old == -1);
  67   old = __sync_lock_test_and_set (&data[i][1], iter);
  68   assert (old == -1);
  69 }
  70
  71 static void f_static_1 (void *dummy)
  72 {
  73   int iam = omp_get_thread_num ();
  74   long s0, e0, i, count = 0;
  75   if (GOMP_loop_static_start (0, N, INCR, CHUNK, &s0, &e0))
  76     do
  77       {
  78         for (i = s0; i < e0; i += INCR)
  79           set_data (i, iam, count);
  80         ++count;
  81       }
  82     while (GOMP_loop_static_next (&s0, &e0));
  83   GOMP_loop_end ();
  84 }
  85
  86 static void test (void)
  87 {
  88   clean_data ();
  89   GOMP_parallel_start (f_static_1, NULL, NTHR);
  90   f_static_1 (NULL);
  91   GOMP_parallel_end ();
  92   test_data ();
  93 }
  94
  95 int main()
  96 {
  97   omp_set_dynamic (0);
  98
  99   NTHR = 5;
 100
 101   INCR = 1, CHUNK = 0;  /* chunk = 360 / 5 = 72 */
 102   test ();
 103
 104   INCR = 4, CHUNK = 0;  /* chunk = 360 / 4 / 5 = 18 */
 105   test ();
 106
 107   INCR = 1, CHUNK = 4;  /* 1 * 4 * 5 = 20 -> 360 / 20 = 18 iterations.  */
 108   test ();
 109
 110   INCR = 3, CHUNK = 4;  /* 3 * 4 * 5 = 60 -> 360 / 60 = 6 iterations.  */
 111   test ();
 112
 113   return 0;
 114 }