openmp/runtime/test/api/omp_realloc_def_fb.c

   1 // RUN: %libomp-compile-and-run
   2
   3 #include <stdio.h>
   4 #include <omp.h>
   5
   6 int main() {
   7   omp_alloctrait_t at[2];
   8   omp_allocator_handle_t a;
   9   omp_allocator_handle_t f_a;
  10   void *ptr[2];
  11   void *nptr[2];
  12   at[0].key = omp_atk_pool_size;
  13   at[0].value = 2 * 1024 * 1024;
  14   at[1].key = omp_atk_fallback;
  15   at[1].value = omp_atv_default_mem_fb;
  16
  17   a = omp_init_allocator(omp_large_cap_mem_space, 2, at);
  18   f_a = omp_init_allocator(omp_default_mem_space, 2, at);
  19   printf("allocator large created: %p\n", (void *)a);
  20   printf("allocator default created: %p\n", (void *)f_a);
  21
  22   #pragma omp parallel num_threads(2)
  23   {
  24     int i = omp_get_thread_num();
  25     ptr[i] = omp_alloc(1024 * 1024, f_a);
  26     #pragma omp barrier
  27     nptr[i] = omp_realloc(ptr[i], 1024 * 1024, a, f_a);
  28     #pragma omp barrier
  29     printf("th %d, nptr %p\n", i, nptr[i]);
  30     omp_free(nptr[i], a);
  31   }
  32   // Both pointers should be non-NULL
  33   if (nptr[0] != NULL && nptr[1] != NULL) {
  34     printf("passed\n");
  35     return 0;
  36   } else {
  37     printf("failed: pointers %p %p\n", nptr[0], nptr[1]);
  38     return 1;
  39   }
  40 }