927af4b3c5
This patch sets the def-allocator-var ICV based on the environment variables provided in OMP_ALLOCATOR. Previously, only allowed value for OMP_ALLOCATOR was a predefined memory allocator. OpenMP 5.1 specification allows predefined memory allocator, predefined mem space, or predefined mem space with traits in OMP_ALLOCATOR. If an allocator can not be created using the provided environment variables, the def-allocator-var is set to omp_default_mem_alloc. Differential Revision: https://reviews.llvm.org/D94985
32 lines
916 B
C
32 lines
916 B
C
// RUN: %libomp-compile
|
|
// RUN: env OMP_ALLOCATOR=omp_high_bw_mem_alloc %libomp-run
|
|
// RUN: env OMP_ALLOCATOR=omp_default_mem_space %libomp-run
|
|
// RUN: env OMP_ALLOCATOR=omp_large_cap_mem_space:alignment=16,pinned=true \
|
|
// RUN: %libomp-run
|
|
// RUN: env \
|
|
// RUN: OMP_ALLOCATOR=omp_high_bw_mem_space:pool_size=1048576,fallback=allocator_fb,fb_data=omp_low_lat_mem_alloc \
|
|
// RUN: %libomp-run
|
|
|
|
#include <stdio.h>
|
|
#include <omp.h>
|
|
|
|
int main() {
|
|
void *p[2];
|
|
#pragma omp parallel num_threads(2)
|
|
{
|
|
int i = omp_get_thread_num();
|
|
p[i] = omp_alloc(1024 * 1024, omp_get_default_allocator());
|
|
#pragma omp barrier
|
|
printf("th %d, ptr %p\n", i, p[i]);
|
|
omp_free(p[i], omp_get_default_allocator());
|
|
}
|
|
// Both pointers should be non-NULL
|
|
if (p[0] != NULL && p[1] != NULL) {
|
|
printf("passed\n");
|
|
return 0;
|
|
} else {
|
|
printf("failed: pointers %p %p\n", p[0], p[1]);
|
|
return 1;
|
|
}
|
|
}
|