openmp/runtime/test/tasking/hidden_helper_task/taskgroup.cpp

   1 // RUN: %libomp-cxx-compile-and-run
   2
   3 /*
   4  * This test aims to check whether hidden helper task can work with regular task
   5  * in terms of dependences. It is equivalent to the following code:
   6  *
   7  * #pragma omp parallel
   8  * for (int i = 0; i < N; ++i) {
   9  *   int data1 = 0, data2 = 0;
  10  * #pragma omp taskgroup
  11  *   {
  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
  28 extern "C" {
  29 struct kmp_task_t_with_privates {
  30   kmp_task_t task;
  31 };
  32
  33 struct anon {
  34   int32_t *data;
  35 };
  36 }
  37
  38 template <int I>
  39 kmp_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
  46 int main(int argc, char *argv[]) {
  47   constexpr const int N = 1024;
  48 #pragma omp parallel for
  49   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: PASS