libgomp/testsuite/libgomp.c/loop-1.c

   1 /* Test that all loop iterations are touched.  This doesn't verify
   2    scheduling order, merely coverage.  */
   3
   4 /* { dg-require-effective-target sync_int_long } */
   5
   6 #include <omp.h>
   7 #include <string.h>
   8 #include <assert.h>
   9 #include "libgomp_g.h"
  10
  11
  12 #define N 10000
  13 static int S, E, INCR, CHUNK, NTHR;
  14 static int data[N];
  15
  16 static void clean_data (void)
  17 {
  18   memset (data, -1, sizeof (data));
  19 }
  20
  21 static void test_data (void)
  22 {
  23   int i, j;
  24
  25   for (i = 0; i < S; ++i)
  26     assert (data[i] == -1);
  27
  28   for (j = 0; i < E; ++i, j = (j + 1) % INCR)
  29     if (j == 0)
  30       assert (data[i] != -1);
  31     else
  32       assert (data[i] == -1);
  33
  34   for (; i < N; ++i)
  35     assert (data[i] == -1);
  36 }
  37
  38 static void set_data (long i, int val)
  39 {
  40   int old;
  41   assert (i >= 0 && i < N);
  42   old = __sync_lock_test_and_set (data+i, val);
  43   assert (old == -1);
  44 }
  45
  46
  47 #define TMPL_1(sched)                                           \
  48 static void f_##sched##_1 (void *dummy)                         \
  49 {                                                               \
  50   int iam = omp_get_thread_num ();                              \
  51   long s0, e0, i;                                               \
  52   if (GOMP_loop_##sched##_start (S, E, INCR, CHUNK, &s0, &e0))  \
  53     do                                                          \
  54       {                                                         \
  55         for (i = s0; i < e0; i += INCR)                         \
  56           set_data (i, iam);                                    \
  57       }                                                         \
  58     while (GOMP_loop_##sched##_next (&s0, &e0));                \
  59   GOMP_loop_end ();                                             \
  60 }                                                               \
  61 static void t_##sched##_1 (void)                                \
  62 {                                                               \
  63   clean_data ();                                                \
  64   GOMP_parallel_start (f_##sched##_1, NULL, NTHR);              \
  65   f_##sched##_1 (NULL);                                         \
  66   GOMP_parallel_end ();                                         \
  67   test_data ();                                                 \
  68 }
  69
  70 TMPL_1(static)
  71 TMPL_1(dynamic)
  72 TMPL_1(guided)
  73
  74 #define TMPL_2(sched)                                   \
  75 static void f_##sched##_2 (void *dummy)                 \
  76 {                                                       \
  77   int iam = omp_get_thread_num ();                      \
  78   long s0, e0, i;                                       \
  79   while (GOMP_loop_##sched##_next (&s0, &e0))           \
  80     {                                                   \
  81       for (i = s0; i < e0; i += INCR)                   \
  82         set_data (i, iam);                              \
  83     }                                                   \
  84   GOMP_loop_end_nowait ();                              \
  85 }                                                       \
  86 static void t_##sched##_2 (void)                        \
  87 {                                                       \
  88   clean_data ();                                        \
  89   GOMP_parallel_loop_##sched##_start                    \
  90     (f_##sched##_2, NULL, NTHR, S, E, INCR, CHUNK);     \
  91   f_##sched##_2 (NULL);                                 \
  92   GOMP_parallel_end ();                                 \
  93   test_data ();                                         \
  94 }
  95
  96 TMPL_2(static)
  97 TMPL_2(dynamic)
  98 TMPL_2(guided)
  99
 100 static void test (void)
 101 {
 102   t_static_1 ();
 103   t_dynamic_1 ();
 104   t_guided_1 ();
 105   t_static_2 ();
 106   t_dynamic_2 ();
 107   t_guided_2 ();
 108 }
 109
 110 int main()
 111 {
 112   omp_set_dynamic (0);
 113
 114   NTHR = 4;
 115
 116   S = 0, E = N, INCR = 1, CHUNK = 4;
 117   test ();
 118
 119   S = 0, E = N, INCR = 2, CHUNK = 4;
 120   test ();
 121
 122   S = 1, E = N-1, INCR = 1, CHUNK = 5;
 123   test ();
 124
 125   S = 1, E = N-1, INCR = 2, CHUNK = 5;
 126   test ();
 127
 128   S = 2, E = 4, INCR = 1, CHUNK = 1;
 129   test ();
 130
 131   S = 0, E = N, INCR = 1, CHUNK = 0;
 132   t_static_1 ();
 133   t_static_2 ();
 134
 135   S = 1, E = N-1, INCR = 1, CHUNK = 0;
 136   t_static_1 ();
 137   t_static_2 ();
 138
 139   return 0;
 140 }