xref: /llvm-project/openmp/runtime/test/tasking/omp_taskloop_num_tasks.c (revision ed5fe64581f4f076c00d5e57caff1ff746d57c79)
1 // This test is known to be fragile on NetBSD kernel at the moment.
2 // UNSUPPORTED: netbsd
3 // RUN: %libomp-compile-and-run
4 // RUN: %libomp-compile && env KMP_TASKLOOP_MIN_TASKS=1 %libomp-run
5 
6 // These compilers don't support the taskloop construct
7 // UNSUPPORTED: gcc-4, gcc-5, icc-16
8 
9 // This test is known to be fragile on NetBSD kernel at the moment,
10 // https://bugs.llvm.org/show_bug.cgi?id=42020.
11 // UNSUPPORTED: netbsd
12 
13 /*
14  * Test for taskloop
15  * Method: calculate how many times the iteration space is dispatched
16  *     and judge if each dispatch has the requested grainsize
17  * It is possible for two adjacent chunks are executed by the same thread
18  */
19 #include <stdio.h>
20 #include <omp.h>
21 #include <stdlib.h>
22 #include "omp_testsuite.h"
23 
24 #define CFDMAX_SIZE 1120
25 
test_omp_taskloop_num_tasks()26 int test_omp_taskloop_num_tasks()
27 {
28   int i;
29   int *tids;
30   int *tidsArray;
31   int count;
32   int result = 0;
33   int num_tasks;
34 
35   for (num_tasks = 1; num_tasks < 120; ++num_tasks) {
36     count = 0;
37     tidsArray = (int *)malloc(sizeof(int) * CFDMAX_SIZE);
38     tids = tidsArray;
39 
40     #pragma omp parallel shared(tids)
41     {
42       int i;
43       #pragma omp master
44       #pragma omp taskloop num_tasks(num_tasks)
45       for (i = 0; i < CFDMAX_SIZE; i++) {
46         tids[i] = omp_get_thread_num();
47       }
48     }
49 
50     for (i = 0; i < CFDMAX_SIZE - 1; ++i) {
51       if (tids[i] != tids[i + 1]) {
52         count++;
53       }
54     }
55 
56     if (count > num_tasks) {
57       fprintf(stderr, "counted too many tasks: (wanted %d, got %d)\n",
58               num_tasks, count);
59       result++;
60     }
61   }
62 
63   return (result==0);
64 }
65 
main()66 int main()
67 {
68   int i;
69   int num_failed=0;
70 
71   for (i = 0; i < REPETITIONS; i++) {
72     if (!test_omp_taskloop_num_tasks()) {
73       num_failed++;
74     }
75   }
76   return num_failed;
77 }
78