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

// RUN: %libomp-cxx-compile-and-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 data1 = 0, data2 = 0;
 * #pragma omp taskgroup
 *   {
 * #pragma omp hidden helper task shared(data1)
 *    {
 *      data1 = 1;
 *    }
 * #pragma omp hidden helper task shared(data2)
 *    {
 *      data2 = 2;
 *    }
 *   }
 *   assert(data1 == 1);
 *   assert(data2 == 2);
 * }
 */

#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 data1 = 0, data2 = 0;
    __kmpc_taskgroup(nullptr, gtid);

    auto task1 = __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>), -1);
    auto shareds = reinterpret_cast<anon *>(task1->shareds);
    shareds->data = &data1;
    __kmpc_omp_task(nullptr, gtid, task1);

    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 = &data2;
    __kmpc_omp_task(nullptr, gtid, task2);

    __kmpc_end_taskgroup(nullptr, gtid);

    assert(data1 == 1);
    assert(data2 == 2);
  }

  std::cout << "PASS\n";
  return 0;
}

// CHECK: PASS