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