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_null_fb; 14 a = omp_init_allocator(omp_large_cap_mem_space, 2, at); 15 printf("allocator large created: %p\n", a); 16 #pragma omp parallel num_threads(2) 17 { 18 int i = omp_get_thread_num(); 19 #pragma omp barrier 20 p[i] = omp_alloc(1024 * 1024, a); 21 #pragma omp barrier 22 printf("th %d, ptr %p\n", i, p[i]); 23 omp_free(p[i], a); 24 } 25 // As an allocator has some small memory overhead 26 // exactly one of the two pointers should be NULL 27 // because of NULL fallback requested 28 if ((p[0] == NULL && p[1] != NULL) || (p[0] != NULL && p[1] == NULL)) { 29 printf("passed\n"); 30 return 0; 31 } else { 32 printf("failed: pointers %p %p\n", p[0], p[1]); 33 return 1; 34 } 35 } 36