103 lines · c
1// RUN: %libomptarget-compile-generic -fopenmp-version=512// RUN: %libomptarget-run-generic 2>&1 \3// RUN: | %fcheck-generic4 5// UNSUPPORTED: amdgcn-amd-amdhsa6 7#include <omp.h>8#include <stdio.h>9#include <stdlib.h>10 11#define N 102412#define FROM 6413#define LENGTH 12814 15void foo() {16 const int device_id = omp_get_default_device();17 float *A;18 A = (float *)omp_target_alloc((FROM + LENGTH) * sizeof(float), device_id);19 20 float *A_dev = NULL;21#pragma omp target has_device_addr(A[FROM : LENGTH]) map(A_dev)22 { A_dev = A; }23 // CHECK: Success24 if (A_dev == NULL || A_dev != A)25 fprintf(stderr, "Failure %p %p \n", A_dev, A);26 else27 fprintf(stderr, "Success\n");28}29 30void bar() {31 short x[10];32 short *xp = &x[0];33 34 x[1] = 111;35#pragma omp target data map(tofrom : xp[0 : 2]) use_device_addr(xp[0 : 2])36#pragma omp target has_device_addr(xp[0 : 2])37 {38 xp[1] = 222;39 // CHECK: 22240 printf("%d %p\n", xp[1], &xp[1]);41 }42 // CHECK: 22243 printf("%d %p\n", xp[1], &xp[1]);44}45 46void moo() {47 short *b = malloc(sizeof(short));48 b = b - 1;49 50 b[1] = 111;51#pragma omp target data map(tofrom : b[1]) use_device_addr(b[1])52#pragma omp target has_device_addr(b[1])53 {54 b[1] = 222;55 // CHECK: 22256 printf("%hd %p %p %p\n", b[1], b, &b[1], &b);57 }58 // CHECK: 22259 printf("%hd %p %p %p\n", b[1], b, &b[1], &b);60}61 62void zoo() {63 short x[10];64 short *(xp[10]);65 xp[1] = &x[0];66 short **xpp = &xp[0];67 68 x[1] = 111;69#pragma omp target data map(tofrom : xpp[1][1]) map(xpp[1]) \70 use_device_addr(xpp[1])71#pragma omp target has_device_addr(xpp[1])72 {73 xpp[1][1] = 222;74 // CHECK: 22275 printf("%d %p %p\n", xpp[1][1], xpp[1], &xpp[1][1]);76 }77 // CHECK: 22278 printf("%d %p %p\n", xpp[1][1], xpp[1], &xpp[1][1]);79}80void xoo() {81 short a[10], b[10];82 a[1] = 111;83 b[1] = 111;84#pragma omp target data map(to : a[0 : 2], b[0 : 2]) use_device_addr(a, b)85#pragma omp target has_device_addr(a) has_device_addr(b[0])86 {87 a[1] = 222;88 b[1] = 222;89 // CHECK: 222 22290 printf("%hd %hd %p %p %p\n", a[1], b[1], &a, b, &b);91 }92 // CHECK:11193 printf("%hd %hd %p %p %p\n", a[1], b[1], &a, b, &b); // 111 111 p1d p2d p3d94}95int main() {96 foo();97 bar();98 moo();99 zoo();100 xoo();101 return 0;102}103