#1849: tests: Skip the CUDA test if CUDA driver is not present

DARMA-tasking · Jun 19, 2022 · e674cff · e674cff
1 parent edb20e9
commit e674cff
Showing 1 changed file with 11 additions and 0 deletions.
diff --git a/tests/unit/active/test_async_op_cuda.cc b/tests/unit/active/test_async_op_cuda.cc
@@ -180,6 +180,17 @@ struct CUDAGroup {
 };
 
 TEST_F(TestAsyncOp, test_async_op_cuda) {
+  int driverVer;
+  if (
+    (cudaDriverGetVersion(&driverVer) == cudaErrorInvalidValue) or
+    (driverVer == 0)) {
+    vtWarn(
+      "Trying to run test_async_op_cuda but CUDA driver is not present!\n"
+    );
+
+    return;
+  }
+
   auto const this_node = theContext()->getNode();
   auto p = theObjGroup()->makeCollective<CUDAGroup>("test_async_op_cuda");
   auto ep = theTerm()->makeEpochRooted(term::UseDS{true});