offload/test/ompt/target_memcpy.c

   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
  16 int 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]]