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
clean_data(void)16 static void clean_data (void)
17 {
18 memset (data, -1, sizeof (data));
19 }
20
test_data(void)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
set_data(long i,int val)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)
TMPL_1(dynamic)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
main()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 }
141