1 // RUN: %libomp-compile-and-run
2
3 #include <stdio.h>
4 #include <omp.h>
5
main()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(0, f_a);
27 #pragma omp barrier
28 nptr[i] = omp_realloc(ptr[i], 1024 * 1024, 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 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 non-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 }
47