xref: /llvm-project/offload/test/unified_shared_memory/close_member.c (revision 330d8983d25d08580fc1642fea48b2473f47a9da)
1*330d8983SJohannes Doerfert // RUN: %libomptarget-compile-run-and-check-generic
2*330d8983SJohannes Doerfert 
3*330d8983SJohannes Doerfert // REQUIRES: unified_shared_memory
4*330d8983SJohannes Doerfert // UNSUPPORTED: clang-6, clang-7, clang-8, clang-9
5*330d8983SJohannes Doerfert 
6*330d8983SJohannes Doerfert #include <omp.h>
7*330d8983SJohannes Doerfert #include <stdio.h>
8*330d8983SJohannes Doerfert 
9*330d8983SJohannes Doerfert #pragma omp requires unified_shared_memory
10*330d8983SJohannes Doerfert 
11*330d8983SJohannes Doerfert struct S {
12*330d8983SJohannes Doerfert   int x;
13*330d8983SJohannes Doerfert   int y;
14*330d8983SJohannes Doerfert };
15*330d8983SJohannes Doerfert 
main(int argc,char * argv[])16*330d8983SJohannes Doerfert int main(int argc, char *argv[]) {
17*330d8983SJohannes Doerfert   int dev = omp_get_default_device();
18*330d8983SJohannes Doerfert   struct S s = {10, 20};
19*330d8983SJohannes Doerfert 
20*330d8983SJohannes Doerfert #pragma omp target enter data map(close, to : s)
21*330d8983SJohannes Doerfert #pragma omp target map(alloc : s)
22*330d8983SJohannes Doerfert   {
23*330d8983SJohannes Doerfert     s.x = 11;
24*330d8983SJohannes Doerfert     s.y = 21;
25*330d8983SJohannes Doerfert   }
26*330d8983SJohannes Doerfert // To determine whether x needs to be transfered or deleted, the runtime
27*330d8983SJohannes Doerfert // cannot simply check whether unified shared memory is enabled and the
28*330d8983SJohannes Doerfert // 'close' modifier is specified.  It must check whether x was previously
29*330d8983SJohannes Doerfert // placed in device memory by, for example, a 'close' modifier that isn't
30*330d8983SJohannes Doerfert // specified here.  The following struct member case checks a special code
31*330d8983SJohannes Doerfert // path in the runtime implementation where members are transferred before
32*330d8983SJohannes Doerfert // deletion of the struct.
33*330d8983SJohannes Doerfert #pragma omp target exit data map(from : s.x, s.y)
34*330d8983SJohannes Doerfert 
35*330d8983SJohannes Doerfert   // CHECK: s.x=11, s.y=21
36*330d8983SJohannes Doerfert   printf("s.x=%d, s.y=%d\n", s.x, s.y);
37*330d8983SJohannes Doerfert   // CHECK: present: 0
38*330d8983SJohannes Doerfert   printf("present: %d\n", omp_target_is_present(&s, dev));
39*330d8983SJohannes Doerfert 
40*330d8983SJohannes Doerfert   return 0;
41*330d8983SJohannes Doerfert }
42