47 lines · c
1// RUN: %libomp-compile-and-run2 3#include <stdio.h>4#include <omp.h>5 6int 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(0, f_a);27 #pragma omp barrier28 nptr[i] = omp_realloc(ptr[i], 1024 * 1024, a, f_a);29 #pragma omp barrier30 printf("th %d, nptr %p\n", i, nptr[i]);31 omp_free(nptr[i], a);32 }33 34 // Both ptr pointers should be NULL35 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 non-NULL40 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}47