| // RUN: %libomptarget-compile-run-and-check-generic |
| // REQUIRES: ompt |
| // UNSUPPORTED: aarch64-unknown-linux-gnu |
| // UNSUPPORTED: aarch64-unknown-linux-gnu-LTO |
| // UNSUPPORTED: x86_64-pc-linux-gnu |
| // UNSUPPORTED: x86_64-pc-linux-gnu-LTO |
| // UNSUPPORTED: s390x-ibm-linux-gnu |
| // UNSUPPORTED: s390x-ibm-linux-gnu-LTO |
| |
| /* |
| * Verify all three data transfer directions: H2D, D2D and D2H |
| */ |
| |
| #include <omp.h> |
| #include <stdio.h> |
| #include <stdlib.h> |
| |
| #include "callbacks.h" |
| #include "register_emi.h" |
| |
| int main(void) { |
| int NumDevices = omp_get_num_devices(); |
| assert(NumDevices > 0 && "No device(s) present."); |
| int Device = omp_get_default_device(); |
| int Host = omp_get_initial_device(); |
| // Note: Zero value depicts an OFFLOAD_SUCCESS |
| int Status; |
| |
| printf("Allocating Memory on Device\n"); |
| int *DevPtr = (int *)omp_target_alloc(sizeof(int), Device); |
| assert(DevPtr && "Could not allocate memory on device."); |
| int *HstPtr = (int *)malloc(sizeof(int)); |
| *HstPtr = 42; |
| |
| printf("Testing: Host to Device\n"); |
| Status = omp_target_memcpy(DevPtr, HstPtr, sizeof(int), 0, 0, Device, Host); |
| assert(Status == 0 && "H2D memory copy operation failed.\n"); |
| |
| printf("Testing: Device to Device\n"); |
| Status = omp_target_memcpy(DevPtr, DevPtr, sizeof(int), 0, 0, Device, Device); |
| assert(Status == 0 && "D2D memory copy operation failed.\n"); |
| |
| printf("Testing: Device to Host\n"); |
| Status = omp_target_memcpy(HstPtr, DevPtr, sizeof(int), 0, 0, Host, Device); |
| assert(Status == 0 && "D2H memory copy operation failed.\n"); |
| |
| printf("Checking Correctness\n"); |
| assert(*HstPtr == 42); |
| |
| printf("Freeing Memory on Device\n"); |
| free(HstPtr); |
| omp_target_free(DevPtr, Device); |
| |
| return 0; |
| } |
| |
| // clang-format off |
| |
| /// CHECK: Callback Init: |
| |
| /// CHECK: Allocating Memory on Device |
| /// CHECK: Callback DataOp EMI: endpoint=1 optype=1 |
| /// CHECK-SAME: src_device_num=[[HOST:[0-9]+]] |
| /// CHECK-SAME: dest_device_num=[[DEVICE:[0-9]+]] |
| /// CHECK: Callback DataOp EMI: endpoint=2 optype=1 {{.+}} src_device_num=[[HOST]] {{.+}} dest_device_num=[[DEVICE]] |
| |
| /// CHECK: Testing: Host to Device |
| /// CHECK: Callback DataOp EMI: endpoint=1 optype=2 {{.+}} src_device_num=[[HOST]] {{.+}} dest_device_num=[[DEVICE]] |
| /// CHECK: Callback DataOp EMI: endpoint=2 optype=2 {{.+}} src_device_num=[[HOST]] {{.+}} dest_device_num=[[DEVICE]] |
| |
| /// CHECK: Testing: Device to Device |
| /// CHECK: Callback DataOp EMI: endpoint=1 optype=3 {{.+}} src_device_num=[[DEVICE]] {{.+}} dest_device_num=[[DEVICE]] |
| /// CHECK: Callback DataOp EMI: endpoint=2 optype=3 {{.+}} src_device_num=[[DEVICE]] {{.+}} dest_device_num=[[DEVICE]] |
| |
| /// CHECK: Testing: Device to Host |
| /// CHECK: Callback DataOp EMI: endpoint=1 optype=3 {{.+}} src_device_num=[[DEVICE]] {{.+}} dest_device_num=[[HOST]] |
| /// CHECK: Callback DataOp EMI: endpoint=2 optype=3 {{.+}} src_device_num=[[DEVICE]] {{.+}} dest_device_num=[[HOST]] |
| |
| /// CHECK: Checking Correctness |
| |
| /// CHECK: Freeing Memory on Device |
| /// CHECK: Callback DataOp EMI: endpoint=1 optype=4 {{.+}} src_device_num=[[DEVICE]] |
| /// CHECK: Callback DataOp EMI: endpoint=2 optype=4 {{.+}} src_device_num=[[DEVICE]] |
| |
| /// CHECK: Callback Fini: |
| |
| // clang-format on |