mirror of
https://github.com/llvm/llvm-project.git
synced 2025-04-16 17:56:32 +00:00
8823448807
Many tests in the `offload` project have requirements defined by which targets are not supported rather than which platforms are supported. This patch aims to streamline the requirement definitions by adding four new feature tags: `host`, `gpu`, `amdgpu`, and `nvidiagpu`.
72 lines
2.2 KiB
C
72 lines
2.2 KiB
C
// RUN: %libomptarget-compile-run-and-check-generic
|
|
// REQUIRES: ompt
|
|
// REQUIRES: gpu
|
|
|
|
/*
|
|
* Verify that for the target OpenMP APIs, the return address is non-null and
|
|
* distinct.
|
|
*/
|
|
|
|
#include <omp.h>
|
|
#include <stdlib.h>
|
|
|
|
#include "callbacks.h"
|
|
#include "register_non_emi.h"
|
|
|
|
int main() {
|
|
int dev = omp_get_default_device();
|
|
int host = omp_get_initial_device();
|
|
|
|
int host_var1 = 42;
|
|
int host_var2 = 0;
|
|
void *dev_ptr = NULL;
|
|
|
|
// Allocate space on the device
|
|
dev_ptr = omp_target_alloc(sizeof(int), dev);
|
|
if (dev_ptr == NULL)
|
|
abort();
|
|
|
|
// H2D transfer
|
|
if (omp_target_memcpy(dev_ptr, &host_var1, sizeof(int), 0, 0, dev, host))
|
|
abort();
|
|
|
|
// D2D transfer
|
|
if (omp_target_memcpy(dev_ptr, dev_ptr, sizeof(int), 0, 0, dev, dev))
|
|
abort();
|
|
|
|
// D2H transfer
|
|
if (omp_target_memcpy(&host_var2, dev_ptr, sizeof(int), 0, 0, host, dev))
|
|
abort();
|
|
|
|
// Free the device location
|
|
omp_target_free(dev_ptr, dev);
|
|
|
|
// Both host variables should have the same value.
|
|
return host_var1 != host_var2;
|
|
}
|
|
|
|
// clang-format off
|
|
/// CHECK: Callback DataOp: target_id=[[TARGET_ID:[0-9]+]] host_op_id=[[HOST_OP_ID:[0-9]+]] optype=1
|
|
/// CHECK-SAME: src_device_num=[[HOST:[0-9]+]]
|
|
/// CHECK-SAME: dest_device_num=[[DEVICE:[0-9]+]]
|
|
/// CHECK-NOT: code=(nil)
|
|
/// CHECK: code=[[CODE1:0x[0-f]+]]
|
|
/// CHECK: Callback DataOp: target_id=[[TARGET_ID:[0-9]+]] host_op_id=[[HOST_OP_ID:[0-9]+]] optype=2
|
|
/// CHECK-SAME: src_device_num=[[HOST]] {{.+}} dest_device_num=[[DEVICE]]
|
|
/// CHECK-NOT: code=(nil)
|
|
/// CHECK-NOT: code=[[CODE1]]
|
|
/// CHECK: code=[[CODE2:0x[0-f]+]]
|
|
/// CHECK: Callback DataOp: target_id=[[TARGET_ID:[0-9]+]] host_op_id=[[HOST_OP_ID:[0-9]+]] optype=3
|
|
/// CHECK-SAME: src_device_num=[[DEVICE]] {{.+}} dest_device_num=[[DEVICE]]
|
|
/// CHECK-NOT: code=(nil)
|
|
/// CHECK-NOT: code=[[CODE2]]
|
|
/// CHECK: code=[[CODE3:0x[0-f]+]]
|
|
/// CHECK: Callback DataOp: target_id=[[TARGET_ID:[0-9]+]] host_op_id=[[HOST_OP_ID:[0-9]+]] optype=3
|
|
/// CHECK-SAME: src_device_num=[[DEVICE]] {{.+}} dest_device_num=[[HOST]]
|
|
/// CHECK-NOT: code=(nil)
|
|
/// CHECK-NOT: code=[[CODE3]]
|
|
/// CHECK: code=[[CODE4:0x[0-f]+]]
|
|
/// CHECK: Callback DataOp: target_id=[[TARGET_ID:[0-9]+]] host_op_id=[[HOST_OP_ID:[0-9]+]] optype=4
|
|
/// CHECK-NOT: code=(nil)
|
|
/// CHECK-NOT: code=[[CODE4]]
|