36 lines · c
1// RUN: %libomp-compile-and-run2 3#include <stdio.h>4#include <omp.h>5 6int 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_default_mem_space, 2, at);15 printf("allocator created: %p\n", (void *)a);16#pragma omp parallel num_threads(2)17 {18 int i = omp_get_thread_num();19 #pragma omp barrier20 p[i] = omp_alloc(1024 * 1024, a);21 #pragma omp barrier22 printf("th %d, ptr %p\n", i, p[i]);23 omp_free(p[i], a);24 }25 // As an allocator has some small memory overhead26 // exactly one of the two pointers should be NULL27 // because of NULL fallback requested28 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