openmp/libomptarget/deviceRTLs/nvptx/test/parallel/thread_limit.c

   1 // RUN: %compile-run-and-check
   2
   3 #include <stdio.h>
   4 #include <omp.h>
   5
   6 const int WarpSize = 32;
   7 const int ThreadLimit = 1 * WarpSize;
   8 const int NumThreads2 = 2 * WarpSize;
   9 const int NumThreads3 = 3 * WarpSize;
  10 const int MaxThreads = 1024;
  11
  12 int main(int argc, char *argv[]) {
  13   int check1[MaxThreads];
  14   int check2[MaxThreads];
  15   int check3[MaxThreads];
  16   for (int i = 0; i < MaxThreads; i++) {
  17     check1[i] = check2[i] = check3[i] = 0;
  18   }
  19
  20   int threadLimit = -1;
  21
  22   #pragma omp target teams num_teams(1) thread_limit(ThreadLimit) \
  23                            map(check1[:], check2[:], check3[:], threadLimit)
  24   {
  25     threadLimit = omp_get_thread_limit();
  26
  27     // All parallel regions should get as many threads as specified by the
  28     // thread_limit() clause.
  29     #pragma omp parallel
  30     {
  31       check1[omp_get_thread_num()] += omp_get_num_threads();
  32     }
  33
  34     omp_set_num_threads(NumThreads2);
  35     #pragma omp parallel
  36     {
  37       check2[omp_get_thread_num()] += omp_get_num_threads();
  38     }
  39
  40     #pragma omp parallel num_threads(NumThreads3)
  41     {
  42       check3[omp_get_thread_num()] += omp_get_num_threads();
  43     }
  44   }
  45
  46   // CHECK: threadLimit = 32
  47   printf("threadLimit = %d\n", threadLimit);
  48
  49   // CHECK-NOT: invalid
  50   for (int i = 0; i < MaxThreads; i++) {
  51     if (i < ThreadLimit) {
  52       if (check1[i] != ThreadLimit) {
  53         printf("invalid: check1[%d] should be %d, is %d\n", i, ThreadLimit, check1[i]);
  54       }
  55     } else if (check1[i] != 0) {
  56       printf("invalid: check1[%d] should be 0, is %d\n", i, check1[i]);
  57     }
  58
  59     if (i < ThreadLimit) {
  60       if (check2[i] != ThreadLimit) {
  61         printf("invalid: check2[%d] should be %d, is %d\n", i, ThreadLimit, check2[i]);
  62       }
  63     } else if (check2[i] != 0) {
  64       printf("invalid: check2[%d] should be 0, is %d\n", i, check2[i]);
  65     }
  66
  67     if (i < ThreadLimit) {
  68       if (check3[i] != ThreadLimit) {
  69         printf("invalid: check3[%d] should be %d, is %d\n", i, ThreadLimit, check3[i]);
  70       }
  71     } else if (check3[i] != 0) {
  72       printf("invalid: check3[%d] should be 0, is %d\n", i, check3[i]);
  73     }
  74   }
  75
  76   return 0;
  77 }