final/runtime/test/parallel/omp_parallel_num_threads.c


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

// RUN: %libomp-compile-and-run
#include <stdio.h>
#include "omp_testsuite.h"

int test_omp_parallel_num_threads()
{
  int num_failed;
  int threads;
  int nthreads;
  int max_threads = 0;

  num_failed = 0;

  /* first we check how many threads are available */
  #pragma omp parallel
  {
    #pragma omp master
    max_threads = omp_get_num_threads ();
  }

  /* we increase the number of threads from one to maximum:*/
  for(threads = 1; threads <= max_threads; threads++) {
    nthreads = 0;
    #pragma omp parallel reduction(+:num_failed) num_threads(threads)
    {
      num_failed = num_failed + !(threads == omp_get_num_threads());
      #pragma omp atomic
      nthreads += 1;
    }
    num_failed = num_failed + !(nthreads == threads);
  }
  return (!num_failed);
}

int main()
{
  int i;
  int num_failed=0;

  for(i = 0; i < REPETITIONS; i++) {
    if(!test_omp_parallel_num_threads()) {
      num_failed++;
    }
  }
  return num_failed;
}