1// RUN: %libomptarget-compile-run-and-check-generic
2// REQUIRES: ompt
3// REQUIRES: gpu
4
5/*
6 * Verify that for the target OpenMP APIs, the return address is non-null and
7 * distinct.
8 */
9
10#include <omp.h>
11#include <stdlib.h>
12
13#include "callbacks.h"
14#include "register_non_emi.h"
15
16int main() {
17 int dev = omp_get_default_device();
18 int host = omp_get_initial_device();
19
20 int host_var1 = 42;
21 int host_var2 = 0;
22 void *dev_ptr = NULL;
23
24 // Allocate space on the device
25 dev_ptr = omp_target_alloc(sizeof(int), dev);
26 if (dev_ptr == NULL)
27 abort();
28
29 // H2D transfer
30 if (omp_target_memcpy(dev_ptr, &host_var1, sizeof(int), 0, 0, dev, host))
31 abort();
32
33 // D2D transfer
34 if (omp_target_memcpy(dev_ptr, dev_ptr, sizeof(int), 0, 0, dev, dev))
35 abort();
36
37 // D2H transfer
38 if (omp_target_memcpy(&host_var2, dev_ptr, sizeof(int), 0, 0, host, dev))
39 abort();
40
41 // Free the device location
42 omp_target_free(dev_ptr, dev);
43
44 // Both host variables should have the same value.
45 return host_var1 != host_var2;
46}
47
48// clang-format off
49/// CHECK: Callback DataOp: target_id=[[TARGET_ID:[0-9]+]] host_op_id=[[HOST_OP_ID:[0-9]+]] optype=1
50/// CHECK-SAME: src_device_num=[[HOST:[0-9]+]]
51/// CHECK-SAME: dest_device_num=[[DEVICE:[0-9]+]]
52/// CHECK-NOT: code=(nil)
53/// CHECK: code=[[CODE1:0x[0-f]+]]
54/// CHECK: Callback DataOp: target_id=[[TARGET_ID:[0-9]+]] host_op_id=[[HOST_OP_ID:[0-9]+]] optype=2
55/// CHECK-SAME: src_device_num=[[HOST]] {{.+}} dest_device_num=[[DEVICE]]
56/// CHECK-NOT: code=(nil)
57/// CHECK-NOT: code=[[CODE1]]
58/// CHECK: code=[[CODE2:0x[0-f]+]]
59/// CHECK: Callback DataOp: target_id=[[TARGET_ID:[0-9]+]] host_op_id=[[HOST_OP_ID:[0-9]+]] optype=3
60/// CHECK-SAME: src_device_num=[[DEVICE]] {{.+}} dest_device_num=[[DEVICE]]
61/// CHECK-NOT: code=(nil)
62/// CHECK-NOT: code=[[CODE2]]
63/// CHECK: code=[[CODE3:0x[0-f]+]]
64/// CHECK: Callback DataOp: target_id=[[TARGET_ID:[0-9]+]] host_op_id=[[HOST_OP_ID:[0-9]+]] optype=3
65/// CHECK-SAME: src_device_num=[[DEVICE]] {{.+}} dest_device_num=[[HOST]]
66/// CHECK-NOT: code=(nil)
67/// CHECK-NOT: code=[[CODE3]]
68/// CHECK: code=[[CODE4:0x[0-f]+]]
69/// CHECK: Callback DataOp: target_id=[[TARGET_ID:[0-9]+]] host_op_id=[[HOST_OP_ID:[0-9]+]] optype=4
70/// CHECK-NOT: code=(nil)
71/// CHECK-NOT: code=[[CODE4]]
72

source code of offload/test/ompt/target_memcpy.c