set device

AMReX-Codes · Sep 21, 2024 · f9810e5 · f9810e5
1 parent 95449d3
commit f9810e5
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 4 deletions.
diff --git a/Tests/GPU/main.cpp b/Tests/GPU/main.cpp
@@ -4,14 +4,15 @@
 
 using namespace amrex;
 
-#if defined(AMREX_USE_GPU)
-#endif
-
 int main (int argc, char* argv[])
 {
     amrex::ignore_unused(argc, argv);
 
-#ifdef AMREX_USE_SYCL
+#if defined(AMREX_USE_CUDA)
+    AMREX_CUDA_SAFE_CALL(cudaSetDevice(0));
+#elif defined(AMREX_USE_HIP)
+    AMREX_HIP_SAFE_CALL(hipSetDevice(0));
+#elif defined(AMREX_USE_SYCL)
     sycl::device sycl_device(sycl::gpu_selector_v);
     sycl::context sycl_context(sycl_device);
     sycl::queue sycl_queue(sycl_context, sycl_device,

diff --git a/Tools/GNUMake/Make.defs b/Tools/GNUMake/Make.defs
@@ -350,6 +350,10 @@ else ifeq ($(USE_HIP),TRUE)
       AMD_ARCH = $(AMREX_AMD_ARCH)
     endif
 
+    ifeq ($(AMD_ARCH),)
+      $(error Must set AMD_ARCH)
+    endif
+
     ifeq ($(HIP_SAVE_TEMPS),TRUE)
       ifeq ($(USE_GPU_RDC),TRUE)
         $(warning *** HIP_SAVE_TEMPS requires USE_GPU_RDC=FALSE to obtain the assembly files for AMD GPU kernels.)