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
13static int S, E, INCR, CHUNK, NTHR;
14static int data[N];
15
16static void clean_data (void)
17{
18  memset (data, -1, sizeof (data));
19}
20
21static 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
38static 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)						\
48static 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}								\
61static 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
70TMPL_1(static)
71TMPL_1(dynamic)
72TMPL_1(guided)
73
74#define TMPL_2(sched)					\
75static 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}							\
86static 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
96TMPL_2(static)
97TMPL_2(dynamic)
98TMPL_2(guided)
99
100static 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
110int 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}
141