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