79 lines · cpp
1// RUN: %libomp-cxx-compile-and-run2 3/*4 * This test aims to check whether hidden helper task can work with regular task5 * in terms of dependences. It is equivalent to the following code:6 *7 * #pragma omp parallel8 * for (int i = 0; i < N; ++i) {9 * int data1 = 0, data2 = 0;10 * #pragma omp taskgroup11 * {12 * #pragma omp hidden helper task shared(data1)13 * {14 * data1 = 1;15 * }16 * #pragma omp hidden helper task shared(data2)17 * {18 * data2 = 2;19 * }20 * }21 * assert(data1 == 1);22 * assert(data2 == 2);23 * }24 */25 26#include "common.h"27 28extern "C" {29struct kmp_task_t_with_privates {30 kmp_task_t task;31};32 33struct anon {34 int32_t *data;35};36}37 38template <int I>39kmp_int32 omp_task_entry(kmp_int32 gtid, kmp_task_t_with_privates *task) {40 auto shareds = reinterpret_cast<anon *>(task->task.shareds);41 auto p = shareds->data;42 *p = I;43 return 0;44}45 46int main(int argc, char *argv[]) {47 constexpr const int N = 1024;48#pragma omp parallel for49 for (int i = 0; i < N; ++i) {50 int32_t gtid = __kmpc_global_thread_num(nullptr);51 int32_t data1 = 0, data2 = 0;52 __kmpc_taskgroup(nullptr, gtid);53 54 auto task1 = __kmpc_omp_target_task_alloc(55 nullptr, gtid, 1, sizeof(kmp_task_t_with_privates), sizeof(anon),56 reinterpret_cast<kmp_routine_entry_t>(omp_task_entry<1>), -1);57 auto shareds = reinterpret_cast<anon *>(task1->shareds);58 shareds->data = &data1;59 __kmpc_omp_task(nullptr, gtid, task1);60 61 auto task2 = __kmpc_omp_target_task_alloc(62 nullptr, gtid, 1, sizeof(kmp_task_t_with_privates), sizeof(anon),63 reinterpret_cast<kmp_routine_entry_t>(omp_task_entry<2>), -1);64 shareds = reinterpret_cast<anon *>(task2->shareds);65 shareds->data = &data2;66 __kmpc_omp_task(nullptr, gtid, task2);67 68 __kmpc_end_taskgroup(nullptr, gtid);69 70 assert(data1 == 1);71 assert(data2 == 2);72 }73 74 std::cout << "PASS\n";75 return 0;76}77 78// CHECK: PASS79