| 1 | // RUN: %libomp-compile-and-run |
|---|
| 2 | // RUN: env OMP_ALLOCATOR=omp_default_mem_space:pinned=true %libomp-run |
|---|
| 3 | |
|---|
| 4 | #include <omp.h> |
|---|
| 5 | |
|---|
| 6 | int main() { |
|---|
| 7 | omp_alloctrait_t pinned_trait[1] = {{omp_atk_pinned, omp_atv_true}}; |
|---|
| 8 | omp_allocator_handle_t pinned_alloc = |
|---|
| 9 | omp_init_allocator(omp_default_mem_space, 1, pinned_trait); |
|---|
| 10 | omp_allocator_handle_t default_alloc = omp_get_default_allocator(); |
|---|
| 11 | double *a = (double *)omp_alloc(10 * sizeof(double), pinned_alloc); |
|---|
| 12 | double *b = (double *)omp_alloc(10 * sizeof(double), default_alloc); |
|---|
| 13 | |
|---|
| 14 | if (!a || !b) return 1; |
|---|
| 15 | |
|---|
| 16 | #pragma omp parallel for |
|---|
| 17 | for (int i = 0; i < 10; i++) { |
|---|
| 18 | a[i] = 0; |
|---|
| 19 | b[i] = 1; |
|---|
| 20 | } |
|---|
| 21 | |
|---|
| 22 | for (int i = 0; i < 10; i++) |
|---|
| 23 | if (a[i] != 0 || b[i] != 1) return 1; |
|---|
| 24 | |
|---|
| 25 | omp_free(a, pinned_alloc); |
|---|
| 26 | omp_free(b, default_alloc); |
|---|
| 27 | |
|---|
| 28 | return 0; |
|---|
| 29 | } |
|---|
| 30 | |
|---|
