reference, declarationdefinition
definition → references, declarations, derived classes, virtual overrides
reference to multiple definitions → definitions
unreferenced
    1
    2
    3
    4
    5
    6
    7
    8
    9
   10
   11
   12
   13
   14
   15
   16
   17
   18
   19
   20
   21
   22
   23
   24
   25
   26
   27
   28
   29
   30
   31
   32
   33
   34
   35
   36
   37
   38
   39
   40
   41
   42
   43
   44
   45
   46
   47
   48
   49
   50
   51
   52
   53
   54
   55
   56
   57
   58
   59
   60
   61
   62
   63
   64
   65
   66
   67
   68
   69
   70
   71
   72
   73
   74
   75
   76
   77
   78
   79
   80
   81
   82
   83
   84
   85
   86
   87
   88
   89
   90
   91
   92
   93
   94
   95
   96
   97
   98
   99
  100
  101
  102
  103
  104
  105
  106
  107
  108
  109
  110
  111
  112
  113
  114
  115
  116
  117
  118
  119
  120
  121
  122
  123
  124
  125
  126
  127
  128
  129
  130
  131
  132
  133
  134
  135
  136
// RUN: %compile-run-and-check

#include <omp.h>
#include <stdio.h>

const int MaxThreads = 1024;
const int NumThreads = 64;
const int NumThreads1 = 1;

int main(int argc, char *argv[]) {
  int inParallel = -1, numThreads = -1, threadNum = -1;
  int check1[MaxThreads];
  int check2[MaxThreads];
  for (int i = 0; i < MaxThreads; i++) {
    check1[i] = check2[i] = 0;
  }

#pragma omp target map(inParallel, numThreads, threadNum, check1[:], check2[:])
  {
    inParallel = omp_in_parallel();
    numThreads = omp_get_num_threads();
    threadNum = omp_get_thread_num();

// Expecting active parallel region.
#pragma omp parallel num_threads(NumThreads)
    {
      int id = omp_get_thread_num();
      check1[id] += omp_get_num_threads() + omp_in_parallel();

// Expecting serialized parallel region.
#pragma omp parallel
      {
        // Expected to be 1.
        int nestedInParallel = omp_in_parallel();
        // Expected to be 1.
        int nestedNumThreads = omp_get_num_threads();
        // Expected to be 0.
        int nestedThreadNum = omp_get_thread_num();
#pragma omp atomic
        check2[id] += nestedInParallel + nestedNumThreads + nestedThreadNum;
      }
    }
  }

  // CHECK: target: inParallel = 0, numThreads = 1, threadNum = 0
  printf("target: inParallel = %d, numThreads = %d, threadNum = %d\n",
         inParallel, numThreads, threadNum);

  // CHECK-NOT: invalid
  for (int i = 0; i < MaxThreads; i++) {
    // Check that all threads reported
    // omp_get_num_threads() = 64, omp_in_parallel() = 1.
    int Expected = NumThreads + 1;
    if (i < NumThreads) {
      if (check1[i] != Expected) {
        printf("invalid: check1[%d] should be %d, is %d\n", i, Expected,
               check1[i]);
      }
    } else if (check1[i] != 0) {
      printf("invalid: check1[%d] should be 0, is %d\n", i, check1[i]);
    }

    // Check serialized parallel region.
    if (i < NumThreads) {
      if (check2[i] != 2) {
        printf("invalid: check2[%d] should be 2, is %d\n", i, check2[i]);
      }
    } else if (check2[i] != 0) {
      printf("invalid: check2[%d] should be 0, is %d\n", i, check2[i]);
    }
  }

  inParallel = -1;
  numThreads = -1;
  threadNum = -1;
  for (int i = 0; i < MaxThreads; i++) {
    check1[i] = check2[i] = 0;
  }

#pragma omp target map(inParallel, numThreads, threadNum, check1[:], check2[:])
  {
    inParallel = omp_in_parallel();
    numThreads = omp_get_num_threads();
    threadNum = omp_get_thread_num();

// Expecting active parallel region.
#pragma omp parallel num_threads(NumThreads1)
    {
      int id = omp_get_thread_num();
      check1[id] += omp_get_num_threads() + omp_in_parallel();

// Expecting serialized parallel region.
#pragma omp parallel
      {
        // Expected to be 0.
        int nestedInParallel = omp_in_parallel();
        // Expected to be 1.
        int nestedNumThreads = omp_get_num_threads();
        // Expected to be 0.
        int nestedThreadNum = omp_get_thread_num();
#pragma omp atomic
        check2[id] += nestedInParallel + nestedNumThreads + nestedThreadNum;
      }
    }
  }

  // CHECK: target: inParallel = 0, numThreads = 1, threadNum = 0
  printf("target: inParallel = %d, numThreads = %d, threadNum = %d\n",
         inParallel, numThreads, threadNum);

  // CHECK-NOT: invalid
  for (int i = 0; i < MaxThreads; i++) {
    // Check that all threads reported
    // omp_get_num_threads() = 1, omp_in_parallel() = 0.
    int Expected = 1;
    if (i < NumThreads1) {
      if (check1[i] != Expected) {
        printf("invalid: check1[%d] should be %d, is %d\n", i, Expected,
               check1[i]);
      }
    } else if (check1[i] != 0) {
      printf("invalid: check1[%d] should be 0, is %d\n", i, check1[i]);
    }

    // Check serialized parallel region.
    if (i < NumThreads1) {
      if (check2[i] != 1) {
        printf("invalid: check2[%d] should be 1, is %d\n", i, check2[i]);
      }
    } else if (check2[i] != 0) {
      printf("invalid: check2[%d] should be 0, is %d\n", i, check2[i]);
    }
  }

  return 0;
}