blob: c0d71b22733b38ab60be89a0603f4da955a57444 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
|
// RUN: %libomptarget-compile-run-and-check-generic
// FIXME: https://github.com/llvm/llvm-project/issues/77841
// UNSUPPORTED: amdgcn-amd-amdhsa
// UNSUPPORTED: nvptx64-nvidia-cuda
// UNSUPPORTED: nvptx64-nvidia-cuda-LTO
#include <omp.h>
#include <stdio.h>
#define N 1024
int test_omp_aligned_alloc_on_device() {
int errors = 0;
omp_memspace_handle_t memspace = omp_default_mem_space;
omp_alloctrait_t traits[2] = {{omp_atk_alignment, 64}, {omp_atk_access, 64}};
omp_allocator_handle_t alloc =
omp_init_allocator(omp_default_mem_space, 1, traits);
#pragma omp target map(tofrom : errors) uses_allocators(alloc(traits))
{
int *x;
int not_correct_array_values = 0;
x = (int *)omp_aligned_alloc(64, N * sizeof(int), alloc);
if (x == NULL) {
errors++;
} else {
#pragma omp parallel for simd simdlen(16) aligned(x : 64)
for (int i = 0; i < N; i++) {
x[i] = i;
}
#pragma omp parallel for simd simdlen(16) aligned(x : 64)
for (int i = 0; i < N; i++) {
if (x[i] != i) {
#pragma omp atomic write
not_correct_array_values = 1;
}
}
if (not_correct_array_values) {
errors++;
}
omp_free(x, alloc);
}
}
omp_destroy_allocator(alloc);
return errors;
}
int main() {
int errors = 0;
if (test_omp_aligned_alloc_on_device())
printf("FAILED\n");
else
// CHECK: PASSED
printf("PASSED\n");
}
|
