| // This test is adapted from test_parallel_for_allocate.c in SOLLVE V&V. |
| // https://github.com/SOLLVE/sollve_vv/blob/master/tests/5.0/parallel_for/test_parallel_for_allocate.c |
| // RUN: %libomp-compile-and-run |
| |
| // Support for allocate was added in GCC 11 |
| // UNSUPPORTED: gcc-4, gcc-5, gcc-6, gcc-7, gcc-8, gcc-9, gcc-10 |
| |
| #include <omp.h> |
| |
| #include <assert.h> |
| #include <stdlib.h> |
| |
| #define N 128 |
| |
| int main(int argc, char *argv[]) { |
| int errors = 0; |
| int *x; |
| int result[N][N]; |
| int successful_alloc = 0; |
| |
| omp_memspace_handle_t x_memspace = omp_default_mem_space; |
| omp_alloctrait_t x_traits[1] = {omp_atk_alignment, 64}; |
| omp_allocator_handle_t x_alloc = omp_init_allocator(x_memspace, 1, x_traits); |
| |
| for (int i = 0; i < N; i++) { |
| for (int j = 0; j < N; j++) { |
| result[i][j] = -1; |
| } |
| } |
| |
| #pragma omp parallel for allocate(x_alloc: x) private(x) shared(result) |
| for (int i = 0; i < N; i++) { |
| x = (int *)malloc(N * sizeof(int)); |
| if (x != NULL) { |
| #pragma omp simd simdlen(16) aligned(x : 64) |
| for (int j = 0; j < N; j++) { |
| x[j] = j * i; |
| } |
| for (int j = 0; j < N; j++) { |
| result[i][j] = x[j]; |
| } |
| free(x); |
| successful_alloc++; |
| } |
| } |
| |
| errors += successful_alloc < 1; |
| |
| for (int i = 0; i < N; i++) { |
| for (int j = 0; j < N; j++) { |
| errors += result[i][j] != i * j; |
| } |
| } |
| |
| omp_destroy_allocator(x_alloc); |
| |
| return errors; |
| } |