| 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 |
| 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: &loc, gtid, TIED, sz: sizeof(kmp_task_t), shar: 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: &loc, gtid, task: A, nd: 2, dep_lst: deps, nd_noalias: 0, noalias_dep_lst: 0); |
| 40 | |
| 41 | // B - in(x, y) |
| 42 | B = __kmpc_omp_task_alloc(loc: &loc, gtid, TIED, sz: sizeof(kmp_task_t), shar: 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: &loc, gtid, task: B, nd: 2, dep_lst: deps, nd_noalias: 0, noalias_dep_lst: 0); |
| 52 | |
| 53 | // Retrieve TDG nodes |
| 54 | A_succ = __kmpc_task_get_successors(task: A); |
| 55 | B_node = __kmpc_task_get_depnode(task: 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 | |