llvm-project/offload/test/api/omp_get_mapped_ptr.c
fineg74 1611a23a5b
[OFFLOAD] Add spirv implementation for named barrier (#180393)
This change adds implementation for named barriers for SPIRV backend.
Since there is no built in API/intrinsics for named barrier in SPIRV,
the implementation loosely follows implementation for AMD
2026-03-27 20:14:09 +01:00

40 lines
927 B
C

// RUN: %libomptarget-compile-and-run-generic
#include <assert.h>
#include <omp.h>
#include <stdlib.h>
#define N 1024
#define OFFSET 16
int main(int argc, char *argv[]) {
int *host_data = (int *)malloc(sizeof(int) * N);
void *device_ptr = omp_get_mapped_ptr(host_data, 0);
assert(device_ptr == NULL && "the pointer should not be mapped right now");
#pragma omp target enter data map(to: host_data[:N])
device_ptr = omp_get_mapped_ptr(host_data, 0);
assert(device_ptr && "the pointer should be mapped now");
void *ptr = NULL;
#pragma omp target map(from: ptr)
{ ptr = host_data; }
assert(ptr == device_ptr && "wrong pointer mapping");
device_ptr = omp_get_mapped_ptr(host_data + OFFSET, 0);
assert(device_ptr && "the pointer with offset should be mapped");
#pragma omp target map(from: ptr)
{ ptr = host_data + OFFSET; }
assert(ptr == device_ptr && "wrong pointer mapping");
return 0;
}