openmp/runtime/test/api/omp_pause_resource.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
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62

// RUN: %libomp-compile-and-run

// Linking fails for icc 18/19
// UNSUPPORTED: icc-18, icc-19

#include <stdio.h>
#include "omp_testsuite.h"

int test_omp_pause_resource() {
  int fails, nthreads, my_dev;

  fails = 0;
  nthreads = 0;
  my_dev = omp_get_initial_device();

#pragma omp parallel
#pragma omp single
  nthreads = omp_get_num_threads();

  if (omp_pause_resource(omp_pause_soft, my_dev))
    fails++;

#pragma omp parallel shared(nthreads)
#pragma omp single
  nthreads = omp_get_num_threads();

  if (nthreads == 0)
    fails++;
  if (omp_pause_resource(omp_pause_hard, my_dev))
    fails++;
  nthreads = 0;

#pragma omp parallel shared(nthreads)
#pragma omp single
  nthreads = omp_get_num_threads();

  if (nthreads == 0)
    fails++;
  if (omp_pause_resource_all(omp_pause_soft))
    fails++;
  nthreads = 0;

#pragma omp parallel shared(nthreads)
#pragma omp single
  nthreads = omp_get_num_threads();

  if (nthreads == 0)
    fails++;
  return fails == 0;
}

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

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