1 // RUN: %libomp-compile-and-run
2
3 #include <stdio.h>
4 #include <omp.h>
5
main()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 }
41