diff options
-rw-r--r-- | openmp/libomptarget/plugins/amdgpu/src/rtl.cpp | 1 | ||||
-rw-r--r-- | openmp/libomptarget/test/api/omp_device_memory.c | 3 |
2 files changed, 2 insertions, 2 deletions
diff --git a/openmp/libomptarget/plugins/amdgpu/src/rtl.cpp b/openmp/libomptarget/plugins/amdgpu/src/rtl.cpp index 7e042c5..4f4881f 100644 --- a/openmp/libomptarget/plugins/amdgpu/src/rtl.cpp +++ b/openmp/libomptarget/plugins/amdgpu/src/rtl.cpp @@ -2623,6 +2623,7 @@ void *__tgt_rtl_data_alloc(int DeviceId, int64_t Size, void *, int32_t Kind) { hsa_amd_memory_pool_t MemoryPool; switch (Kind) { case TARGET_ALLOC_DEFAULT: + case TARGET_ALLOC_DEVICE: // GPU memory MemoryPool = DeviceInfo().getDeviceMemoryPool(DeviceId); break; diff --git a/openmp/libomptarget/test/api/omp_device_memory.c b/openmp/libomptarget/test/api/omp_device_memory.c index 6bb02be..816ae35 100644 --- a/openmp/libomptarget/test/api/omp_device_memory.c +++ b/openmp/libomptarget/test/api/omp_device_memory.c @@ -1,5 +1,4 @@ -// RUN: %libomptarget-compile-run-and-check-nvptx64-nvidia-cuda -// REQUIRES: nvptx64-nvidia-cuda +// RUN: %libomptarget-compile-run-and-check-generic #include <omp.h> #include <stdio.h> |