1 // RUN: %libomp-cxx-compile-and-run
2 // RUN: %libomp-cxx-compile && env OMP_NUM_THREADS=1 %libomp-run
3
4 /*
5 * This test aims to check whether hidden helper task can work with regular task
6 * in terms of dependences. It is equivalent to the following code:
7 *
8 * #pragma omp parallel
9 * for (int i = 0; i < N; ++i) {
10 * int data = -1;
11 * #pragma omp task shared(data) depend(out: data)
12 * {
13 * data = 1;
14 * }
15 * #pragma omp hidden helper task shared(data) depend(inout: data)
16 * {
17 * data += 2;
18 * }
19 * #pragma omp hidden helper task shared(data) depend(inout: data)
20 * {
21 * data += 4;
22 * }
23 * #pragma omp task shared(data) depend(inout: data)
24 * {
25 * data += 8;
26 * }
27 * #pragma omp taskwait
28 * assert(data == 15);
29 * }
30 */
31
32 #include "common.h"
33
34 extern "C" {
35 struct kmp_task_t_with_privates {
36 kmp_task_t task;
37 };
38
39 struct anon {
40 int32_t *data;
41 };
42 }
43
44 template <int I>
omp_task_entry(kmp_int32 gtid,kmp_task_t_with_privates * task)45 kmp_int32 omp_task_entry(kmp_int32 gtid, kmp_task_t_with_privates *task) {
46 auto shareds = reinterpret_cast<anon *>(task->task.shareds);
47 auto p = shareds->data;
48 *p += I;
49 return 0;
50 }
51
main(int argc,char * argv[])52 int main(int argc, char *argv[]) {
53 constexpr const int N = 1024;
54 #pragma omp parallel for
55 for (int i = 0; i < N; ++i) {
56 int32_t gtid = __kmpc_global_thread_num(nullptr);
57 int32_t data = 0;
58
59 // Task 1
60 auto task1 = __kmpc_omp_task_alloc(
61 nullptr, gtid, 1, sizeof(kmp_task_t_with_privates), sizeof(anon),
62 reinterpret_cast<kmp_routine_entry_t>(omp_task_entry<1>));
63
64 auto shareds = reinterpret_cast<anon *>(task1->shareds);
65 shareds->data = &data;
66
67 kmp_depend_info_t depinfo1;
68 depinfo1.base_addr = reinterpret_cast<intptr_t>(&data);
69 depinfo1.flag = 2; // OUT
70 depinfo1.len = 4;
71
72 __kmpc_omp_task_with_deps(nullptr, gtid, task1, 1, &depinfo1, 0, nullptr);
73
74 // Task 2
75 auto task2 = __kmpc_omp_target_task_alloc(
76 nullptr, gtid, 1, sizeof(kmp_task_t_with_privates), sizeof(anon),
77 reinterpret_cast<kmp_routine_entry_t>(omp_task_entry<2>), -1);
78
79 shareds = reinterpret_cast<anon *>(task2->shareds);
80 shareds->data = &data;
81
82 kmp_depend_info_t depinfo2;
83 depinfo2.base_addr = reinterpret_cast<intptr_t>(&data);
84 depinfo2.flag = 3; // INOUT
85 depinfo2.len = 4;
86
87 __kmpc_omp_task_with_deps(nullptr, gtid, task2, 1, &depinfo2, 0, nullptr);
88
89 // Task 3
90 auto task3 = __kmpc_omp_target_task_alloc(
91 nullptr, gtid, 1, sizeof(kmp_task_t_with_privates), sizeof(anon),
92 reinterpret_cast<kmp_routine_entry_t>(omp_task_entry<4>), -1);
93
94 shareds = reinterpret_cast<anon *>(task3->shareds);
95 shareds->data = &data;
96
97 kmp_depend_info_t depinfo3;
98 depinfo3.base_addr = reinterpret_cast<intptr_t>(&data);
99 depinfo3.flag = 3; // INOUT
100 depinfo3.len = 4;
101
102 __kmpc_omp_task_with_deps(nullptr, gtid, task3, 1, &depinfo3, 0, nullptr);
103
104 // Task 4
105 auto task4 = __kmpc_omp_task_alloc(
106 nullptr, gtid, 1, sizeof(kmp_task_t_with_privates), sizeof(anon),
107 reinterpret_cast<kmp_routine_entry_t>(omp_task_entry<8>));
108
109 shareds = reinterpret_cast<anon *>(task4->shareds);
110 shareds->data = &data;
111
112 kmp_depend_info_t depinfo4;
113 depinfo4.base_addr = reinterpret_cast<intptr_t>(&data);
114 depinfo4.flag = 3; // INOUT
115 depinfo4.len = 4;
116
117 __kmpc_omp_task_with_deps(nullptr, gtid, task4, 1, &depinfo4, 0, nullptr);
118
119 // Wait for all tasks
120 __kmpc_omp_taskwait(nullptr, gtid);
121
122 assert(data == 15);
123 }
124
125 std::cout << "PASS\n";
126 return 0;
127 }
128
129 // CHECK: PASS
130