Check for default device

japlews · Sep 25, 2023 · f6977cf · f6977cf
1 parent 41253bd
commit f6977cf
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/core/src/Cuda/Kokkos_Cuda_Instance.cpp b/core/src/Cuda/Kokkos_Cuda_Instance.cpp
@@ -867,6 +867,11 @@ Cuda::Cuda(int device_id, cudaStream_t stream)
        << " Raised by Kokkos::Cuda::Cuda().\n";
     Kokkos::abort(ss.str().c_str());
   }
+  // FIXME_CUDA
+  if (device_id != Cuda().cuda_device())
+    Kokkos::abort(
+        "Currently, the device id must match the device id used when Kokkos "
+        "was initialized!");
   m_space_instance->initialize(device_id, stream, /*manage_stream*/ false);
 }