libgomp/testsuite/libgomp.c/parallel-1.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

/* Trivial test of thread startup.  */

#include <omp.h>
#include <string.h>
#include <assert.h>
#include "libgomp_g.h"


static int nthr;
static int saw[4];

static void function(void *dummy)
{
  int iam = omp_get_thread_num ();

  if (iam == 0)
    nthr = omp_get_num_threads ();

  saw[iam] = 1;
}

int main()
{
  omp_set_dynamic (0);

  GOMP_parallel_start (function, NULL, 2);
  function (NULL);
  GOMP_parallel_end ();

  assert (nthr == 2);
  assert (saw[0] != 0);
  assert (saw[1] != 0);
  assert (saw[2] == 0);

  memset (saw, 0, sizeof (saw));
  
  GOMP_parallel_start (function, NULL, 3);
  function (NULL);
  GOMP_parallel_end ();

  assert (nthr == 3);
  assert (saw[0] != 0);
  assert (saw[1] != 0);
  assert (saw[2] != 0);
  assert (saw[3] == 0);

  return 0;
}