0
|
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 }
|