Fixed missing cuda driver dependency error. (#195)

ROCm · Feb 18, 2025 · 669fc3e · 669fc3e
1 parent 0debd16
commit 669fc3e
Showing 1 changed file with 9 additions and 0 deletions.
diff --git a/HIP-Basic/runtime_compilation/CMakeLists.txt b/HIP-Basic/runtime_compilation/CMakeLists.txt
@@ -76,6 +76,15 @@ if(GPU_RUNTIME STREQUAL "CUDA")
     # In this example we also need to link nvrtc CUDA library
     find_package(CUDAToolkit)
     list(APPEND link_libs "CUDA::nvrtc")
+
+    find_library(CUDA_DRIVER_LIB cuda)
+    if(NOT CUDA_DRIVER_LIB)
+        message(
+            FATAL_ERROR
+            "Could not find CUDA Driver library (libcuda.so). Please ensure that the NVIDIA drivers are installed."
+        )
+    endif()
+    list(APPEND link_libs "${CUDA_DRIVER_LIB}")
 endif()
 
 target_link_libraries(${example_name} ${link_libs})