1 // REQUIRES: linux
2 // RUN: %libomp-compile && env OMP_NUM_THREADS='2' %libomp-run
3
4 #include <assert.h>
5 #include <omp.h>
6
7 #include "kmp_task_deps.h"
8
9 // the test
main(void)10 int main(void) {
11 volatile int done = 0;
12
13 #pragma omp parallel num_threads(2)
14 {
15 while (omp_get_thread_num() != 0 && !done)
16 ;
17
18 #pragma omp single
19 {
20 kmp_task_t *A, *B;
21 kmp_depnode_list_t *A_succ;
22 kmp_base_depnode_t *B_node;
23 dep deps[2];
24 int gtid;
25 int x, y;
26
27 gtid = __kmpc_global_thread_num(&loc);
28
29 // A - out(x, y)
30 A = __kmpc_omp_task_alloc(&loc, gtid, TIED, sizeof(kmp_task_t), 0, NULL);
31 deps[0].addr = (size_t)&x;
32 deps[0].len = 0;
33 deps[0].flags = 2; // OUT
34
35 deps[1].addr = (size_t)&y;
36 deps[1].len = 0;
37 deps[1].flags = 2; // OUT
38
39 __kmpc_omp_task_with_deps(&loc, gtid, A, 2, deps, 0, 0);
40
41 // B - in(x, y)
42 B = __kmpc_omp_task_alloc(&loc, gtid, TIED, sizeof(kmp_task_t), 0, NULL);
43 deps[0].addr = (size_t)&x;
44 deps[0].len = 0;
45 deps[0].flags = 1; // IN
46
47 deps[1].addr = (size_t)&y;
48 deps[1].len = 0;
49 deps[1].flags = 1; // IN
50
51 __kmpc_omp_task_with_deps(&loc, gtid, B, 2, deps, 0, 0);
52
53 // Retrieve TDG nodes
54 A_succ = __kmpc_task_get_successors(A);
55 B_node = __kmpc_task_get_depnode(B);
56
57 // 'B' should only be added once to 'A' successors list
58 assert(A_succ->node == B_node);
59 assert(A_succ->next == NULL);
60
61 #pragma omp taskwait
62
63 done = 1;
64 }
65 }
66 return 0;
67 }
68