Adding functionality to set the device id for HIP memory space

ornladios · May 4, 2023 · a17eda2 · a17eda2
1 parent b9a811c
commit a17eda2
Showing 1 changed file with 9 additions and 0 deletions.
diff --git a/source/adios2/helper/adiosKokkos.cpp b/source/adios2/helper/adiosKokkos.cpp
@@ -111,6 +111,15 @@ void KokkosInit()
     int device_id;
     cudaGetDevice(&device_id);
     settings.set_device_id(device_id);
+#endif
+#ifdef ADIOS2_HAVE_KOKKOS_HIP
+    int device_id;
+    hipError_t ret;
+    ret = hipGetDevice(&device_id);
+    if (ret == hipSuccess)
+    {
+        settings.set_device_id(device_id);
+    }
 #endif
     Kokkos::initialize(settings);
 }