openmp/runtime/test/api/omp_realloc_size_0.c

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