c0e4a0c3a4
The hidden helper task is only enabled on Linux (kmp_runtime.cpp initializes __kmp_enable_hidden_helper to TRUE for linux but to FALSE for any other OS), and the __kmp_stg_parse_use_hidden_helper function always makes it disabled on non-Linux OSes too. Add a lit test feature for hidden helper tasks (only made available on Linux) and mark two tests as requiring this feature. Disable hidden helper tasks in the test that doesn't really involve them, for consistent behaviour across platforms. Differential Revision: https://reviews.llvm.org/D137749
132 lines
3.8 KiB
C++
132 lines
3.8 KiB
C++
// RUN: %libomp-cxx-compile-and-run
|
|
// RUN: %libomp-cxx-compile && env OMP_NUM_THREADS=1 %libomp-run
|
|
// REQUIRES: hidden-helper
|
|
|
|
/*
|
|
* This test aims to check whether hidden helper thread has right gtid. We also
|
|
* test if there is mixed dependences between regular tasks and hidden helper
|
|
* tasks, the tasks are executed by right set of threads. It is equivalent to
|
|
* the following code:
|
|
*
|
|
* #pragma omp parallel for
|
|
* for (int i = 0; i < N; ++i) {
|
|
* int data1 = -1, data2 = -1, data3 = -1;
|
|
* int depvar;
|
|
* #pragma omp task shared(data1) depend(inout: depvar)
|
|
* {
|
|
* data1 = omp_get_global_thread_id();
|
|
* }
|
|
* #pragma omp task hidden helper shared(data2) depend(inout: depvar)
|
|
* {
|
|
* data2 = omp_get_global_thread_id();
|
|
* }
|
|
* #pragma omp task shared(data3) depend(inout: depvar)
|
|
* {
|
|
* data3 = omp_get_global_thread_id();
|
|
* }
|
|
* #pragma omp taskwait
|
|
* assert(data1 == 0 || data1 > __kmp_num_hidden_helper_threads);
|
|
* assert(data2 > 0 && data2 <= __kmp_num_hidden_helper_threads);
|
|
* assert(data3 == 0 || data3 > __kmp_num_hidden_helper_threads);
|
|
* }
|
|
*/
|
|
|
|
#include "common.h"
|
|
|
|
extern "C" {
|
|
struct kmp_task_t_with_privates {
|
|
kmp_task_t task;
|
|
};
|
|
|
|
struct anon {
|
|
int32_t *data;
|
|
};
|
|
}
|
|
|
|
kmp_int32 __kmp_hidden_helper_threads_num;
|
|
|
|
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 = __kmpc_global_thread_num(nullptr);
|
|
return 0;
|
|
}
|
|
|
|
template <bool hidden_helper_task> void assert_gtid(int v) {
|
|
if (__kmp_hidden_helper_threads_num) {
|
|
if (hidden_helper_task) {
|
|
assert(v > 0 && v <= __kmp_hidden_helper_threads_num);
|
|
} else {
|
|
assert(v == 0 || v > __kmp_hidden_helper_threads_num);
|
|
}
|
|
} else {
|
|
assert(v >= 0);
|
|
}
|
|
}
|
|
|
|
int main(int argc, char *argv[]) {
|
|
__kmp_hidden_helper_threads_num = get_num_hidden_helper_threads();
|
|
|
|
constexpr const int N = 1024;
|
|
#pragma omp parallel for
|
|
for (int i = 0; i < N; ++i) {
|
|
int32_t data1 = -1, data2 = -1, data3 = -1;
|
|
int depvar;
|
|
int32_t gtid = __kmpc_global_thread_num(nullptr);
|
|
|
|
// Task 1, regular task
|
|
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));
|
|
auto shareds = reinterpret_cast<anon *>(task1->shareds);
|
|
shareds->data = &data1;
|
|
|
|
kmp_depend_info_t depinfo1;
|
|
depinfo1.base_addr = reinterpret_cast<intptr_t>(&depvar);
|
|
depinfo1.flag = 3; // INOUT
|
|
depinfo1.len = 4;
|
|
|
|
__kmpc_omp_task_with_deps(nullptr, gtid, task1, 1, &depinfo1, 0, nullptr);
|
|
|
|
// Task 2, hidden helper task
|
|
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), -1);
|
|
shareds = reinterpret_cast<anon *>(task2->shareds);
|
|
shareds->data = &data2;
|
|
|
|
kmp_depend_info_t depinfo2;
|
|
depinfo2.base_addr = reinterpret_cast<intptr_t>(&depvar);
|
|
depinfo2.flag = 3; // INOUT
|
|
depinfo2.len = 4;
|
|
|
|
__kmpc_omp_task_with_deps(nullptr, gtid, task2, 1, &depinfo2, 0, nullptr);
|
|
|
|
// Task 3, regular task
|
|
auto task3 = __kmpc_omp_task_alloc(
|
|
nullptr, gtid, 1, sizeof(kmp_task_t_with_privates), sizeof(anon),
|
|
reinterpret_cast<kmp_routine_entry_t>(omp_task_entry));
|
|
shareds = reinterpret_cast<anon *>(task3->shareds);
|
|
shareds->data = &data3;
|
|
|
|
kmp_depend_info_t depinfo3;
|
|
depinfo3.base_addr = reinterpret_cast<intptr_t>(&depvar);
|
|
depinfo3.flag = 3; // INOUT
|
|
depinfo3.len = 4;
|
|
|
|
__kmpc_omp_task_with_deps(nullptr, gtid, task3, 1, &depinfo3, 0, nullptr);
|
|
|
|
__kmpc_omp_taskwait(nullptr, gtid);
|
|
|
|
// FIXME: 8 here is not accurate
|
|
assert_gtid<false>(data1);
|
|
assert_gtid<true>(data2);
|
|
assert_gtid<false>(data3);
|
|
}
|
|
|
|
std::cout << "PASS\n";
|
|
return 0;
|
|
}
|
|
|
|
// CHECK: PASS
|