libgomp/testsuite/libgomp.c-c++-common/alloc-2.c

   1 #include <omp.h>
   2 #include <stdint.h>
   3 #include <stdlib.h>
   4
   5 int
   6 main ()
   7 {
   8   omp_alloctrait_t traits[3]
   9     = { { omp_atk_alignment, 64 },
  10         { omp_atk_fallback, omp_atv_null_fb },
  11         { omp_atk_pool_size, 4096 } };
  12   omp_allocator_handle_t a
  13     = omp_init_allocator (omp_default_mem_space, 3, traits);
  14   if (a == omp_null_allocator)
  15     abort ();
  16
  17   #pragma omp parallel num_threads(4)
  18   {
  19     int n = omp_get_thread_num ();
  20     double *volatile p, *volatile q;
  21     omp_set_default_allocator ((n & 1) ? a : omp_default_mem_alloc);
  22     p = (double *) omp_alloc (1696, omp_null_allocator);
  23     if (p == NULL)
  24       abort ();
  25     p[0] = 1.0;
  26     p[1695 / sizeof (double)] = 2.0;
  27     #pragma omp barrier
  28     omp_set_default_allocator ((n & 1) ? omp_default_mem_alloc : a);
  29     q = (double *) omp_alloc (1696, omp_null_allocator);
  30     if (n & 1)
  31       {
  32         if (q == NULL)
  33           abort ();
  34         q[0] = 3.0;
  35         q[1695 / sizeof (double)] = 4.0;
  36       }
  37     else if (q != NULL)
  38       abort ();
  39     #pragma omp barrier
  40     omp_free (p, omp_null_allocator);
  41     omp_free (q, omp_null_allocator);
  42     omp_set_default_allocator (omp_default_mem_alloc);
  43   }
  44   omp_destroy_allocator (a);
  45   return 0;
  46 }