1// RUN: %libomptarget-compile-run-and-check-generic
2
3// REQUIRES: unified_shared_memory
4// UNSUPPORTED: clang-6, clang-7, clang-8, clang-9
5
6#include <assert.h>
7#include <omp.h>
8#include <stdio.h>
9
10#pragma omp requires unified_shared_memory
11
12int main(int argc, char *argv[]) {
13 int dev = omp_get_default_device();
14 int x = 10;
15 int *x_dev = (int *)omp_target_alloc(sizeof x, dev);
16 assert(x_dev && "expected omp_target_alloc to succeed");
17 int rc = omp_target_associate_ptr(&x, x_dev, sizeof x, 0, dev);
18 assert(!rc && "expected omp_target_associate_ptr to succeed");
19
20// To determine whether x needs to be transfered, the runtime cannot simply
21// check whether unified shared memory is enabled and the 'close' modifier is
22// specified. It must check whether x was previously placed in device memory
23// by, for example, omp_target_associate_ptr.
24#pragma omp target map(always, tofrom : x)
25 x += 1;
26
27 // CHECK: x=11
28 printf(format: "x=%d\n", x);
29 // CHECK: present: 1
30 printf(format: "present: %d\n", omp_target_is_present(&x, dev));
31
32 return 0;
33}
34

source code of offload/test/unified_shared_memory/associate_ptr.c