1// RUN: %libomp-compile && %libomp-run 7
2// RUN: %libomp-run 0 && %libomp-run -1
3// RUN: %libomp-run 1 && %libomp-run 2 && %libomp-run 5
4// RUN: %libomp-compile -DMY_SCHEDULE=guided && %libomp-run 7
5// RUN: %libomp-run 1 && %libomp-run 2 && %libomp-run 5
6// UNSUPPORTED: clang-11, clang-12
7#include <stdio.h>
8#include <omp.h>
9#include <stdlib.h>
10#include <limits.h>
11#include "omp_testsuite.h"
12
13#define INCR 7
14#define MY_MAX 200
15#define MY_MIN -200
16#ifndef MY_SCHEDULE
17# define MY_SCHEDULE dynamic
18#endif
19
20int num_disp_buffers, num_loops;
21int a, b, a_known_value, b_known_value;
22
23int test_kmp_set_disp_num_buffers()
24{
25 int success = 1;
26 a = 0;
27 b = 0;
28 // run many small dynamic loops to stress the dispatch buffer system
29 #pragma omp parallel
30 {
31 int i,j;
32 for (j = 0; j < num_loops; j++) {
33 #pragma omp for schedule(MY_SCHEDULE) nowait
34 for (i = MY_MIN; i < MY_MAX; i+=INCR) {
35 #pragma omp atomic
36 a++;
37 }
38 #pragma omp for schedule(MY_SCHEDULE) nowait
39 for (i = MY_MAX; i >= MY_MIN; i-=INCR) {
40 #pragma omp atomic
41 b++;
42 }
43 }
44 }
45 // detect failure
46 if (a != a_known_value || b != b_known_value) {
47 success = 0;
48 printf(format: "a = %d (should be %d), b = %d (should be %d)\n", a, a_known_value,
49 b, b_known_value);
50 }
51 return success;
52}
53
54int main(int argc, char** argv)
55{
56 int i,j;
57 int num_failed=0;
58
59 if (argc != 2) {
60 fprintf(stderr, format: "usage: %s num_disp_buffers\n", argv[0]);
61 exit(status: 1);
62 }
63
64 // set the number of dispatch buffers
65 num_disp_buffers = atoi(nptr: argv[1]);
66 kmp_set_disp_num_buffers(num_disp_buffers);
67
68 // figure out the known values to compare with calculated result
69 a_known_value = 0;
70 b_known_value = 0;
71
72 // if specified to use bad num_disp_buffers set num_loops
73 // to something reasonable
74 if (num_disp_buffers <= 0)
75 num_loops = 10;
76 else
77 num_loops = num_disp_buffers*10;
78
79 for (j = 0; j < num_loops; j++) {
80 for (i = MY_MIN; i < MY_MAX; i+=INCR)
81 a_known_value++;
82 for (i = MY_MAX; i >= MY_MIN; i-=INCR)
83 b_known_value++;
84 }
85
86 for(i = 0; i < REPETITIONS; i++) {
87 if(!test_kmp_set_disp_num_buffers()) {
88 num_failed++;
89 }
90 }
91 if (num_failed == 0)
92 printf(format: "passed\n");
93 else
94 printf(format: "failed %d\n", num_failed);
95 return num_failed;
96}
97

source code of openmp/runtime/test/worksharing/for/kmp_set_dispatch_buf.c