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   void *p[2];
10   at[0].key = omp_atk_pool_size;
11   at[0].value = 2 * 1024 * 1024;
12   at[1].key = omp_atk_fallback;
13   at[1].value = omp_atv_default_mem_fb;
14   a = omp_init_allocator(omp_large_cap_mem_space, 2, at);
15   printf("allocator large created: %p\n", (void *)a);
16 #pragma omp parallel num_threads(2)
17   {
18     int i = omp_get_thread_num();
19     p[i] = omp_alloc(1024 * 1024, a);
20     #pragma omp barrier
21     printf("th %d, ptr %p\n", i, p[i]);
22     omp_free(p[i], a);
23   }
24   // Both pointers should be non-NULL
25   if (p[0] != NULL && p[1] != NULL) {
26     printf("passed\n");
27     return 0;
28   } else {
29     printf("failed: pointers %p %p\n", p[0], p[1]);
30     return 1;
31   }
32 }
33