accelerator/cuda: Check for cuda devices

Adds a check during component initialization for cuda capable devices. Does not select the component if no cuda capable devices are detected. Signed-off-by: William Zhang <[email protected]>
open-mpi · Apr 25, 2023 · ae98e04 · ae98e04
1 parent da6d715
commit ae98e04
Showing 1 changed file with 9 additions and 1 deletion.
diff --git a/opal/mca/accelerator/cuda/accelerator_cuda_component.c b/opal/mca/accelerator/cuda/accelerator_cuda_component.c
@@ -188,7 +188,15 @@ static opal_accelerator_base_module_t* accelerator_cuda_init(void)
     if (!opal_cuda_support) {
         return NULL;
     }
-
+    int count = 0;
+    /* If cuInit fails or there are no cuda capable devices, return NULL. */
+    if (cuInit(0)) {
+        return NULL;
+    }
+    CUresult ret = cuDeviceGetCount(&count);
+    if (ret || count == 0) {
+        return NULL;
+    }
     opal_accelerator_cuda_delayed_init();
     return &opal_accelerator_cuda_module;
 }