offload/test/offloading/thread_state_2.c

   1 // This fails when optimized for now.
   2 // RUN: %libomptarget-compile-run-and-check-generic
   3 // XUN: %libomptarget-compileopt-run-and-check-generic
   4
   5 #include <omp.h>
   6 #include <stdio.h>
   7
   8 int main() {
   9   // TODO: Test all ICVs on all levels
  10   int o_lvl = 111, i_lvl = 222, o_tid = 333, i_tid = 333, o_nt = 444,
  11       i_nt = 555;
  12 #pragma omp target teams map(tofrom : o_lvl, i_lvl, o_tid, i_tid, o_nt, i_nt)  \
  13     num_teams(2) thread_limit(64)
  14   {
  15     omp_set_max_active_levels(1);
  16     if (omp_get_team_num() == 0) {
  17 #pragma omp parallel num_threads(64)
  18       if (omp_get_thread_num() == omp_get_num_threads() - 1) {
  19         o_lvl = omp_get_level();
  20         o_tid = omp_get_thread_num();
  21         o_nt = omp_get_num_threads();
  22 #pragma omp parallel num_threads(64)
  23         if (omp_get_thread_num() == omp_get_num_threads() - 1) {
  24           i_lvl = omp_get_level();
  25           i_tid = omp_get_thread_num();
  26           i_nt = omp_get_num_threads();
  27         }
  28       }
  29     }
  30   }
  31   if (o_lvl == 1 && o_tid == o_nt - 1 && o_nt >= 1 && i_lvl == 2 &&
  32       i_tid == 0 && i_nt == 1) {
  33     // CHECK: Success
  34     printf("Success\n");
  35     return 0;
  36   }
  37   printf("outer: lvl: %i, tid: %i, nt: %i\n", o_lvl, o_tid, o_nt);
  38   printf("inner: lvl: %i, tid: %i, nt: %i\n", i_lvl, i_tid, i_nt);
  39   return 1;
  40 }