offload/test/api/omp_get_mapped_ptr.c


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

// RUN: %libomptarget-compile-and-run-generic
// XFAIL: intelgpu

#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;
}