File: associate_ptr.c

package info (click to toggle)
llvm-toolchain-15 1%3A15.0.6-4
  • links: PTS, VCS
  • area: main
  • in suites: bookworm
  • size: 1,554,644 kB
  • sloc: cpp: 5,922,452; ansic: 1,012,136; asm: 674,362; python: 191,568; objc: 73,855; f90: 42,327; lisp: 31,913; pascal: 11,973; javascript: 10,144; sh: 9,421; perl: 7,447; ml: 5,527; awk: 3,523; makefile: 2,520; xml: 885; cs: 573; fortran: 567
file content (33 lines) | stat: -rw-r--r-- 1,052 bytes parent folder | download | duplicates (4) 
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
 
// RUN: %libomptarget-compile-run-and-check-generic

// REQUIRES: unified_shared_memory
// UNSUPPORTED: clang-6, clang-7, clang-8, clang-9

#include <assert.h>
#include <omp.h>
#include <stdio.h>

#pragma omp requires unified_shared_memory

int main(int argc, char *argv[]) {
  int dev = omp_get_default_device();
  int x = 10;
  int *x_dev = (int *)omp_target_alloc(sizeof x, dev);
  assert(x_dev && "expected omp_target_alloc to succeed");
  int rc = omp_target_associate_ptr(&x, x_dev, sizeof x, 0, dev);
  assert(!rc && "expected omp_target_associate_ptr to succeed");

  // To determine whether x needs to be transfered, the runtime cannot simply
  // check whether unified shared memory is enabled and the 'close' modifier is
  // specified.  It must check whether x was previously placed in device memory
  // by, for example, omp_target_associate_ptr.
  #pragma omp target map(always, tofrom: x)
  x += 1;

  // CHECK: x=11
  printf("x=%d\n", x);
  // CHECK: present: 1
  printf("present: %d\n", omp_target_is_present(&x, dev));

  return 0;
}