diff --git a/mlir/tools/mlir-shlib/CMakeLists.txt b/mlir/tools/mlir-shlib/CMakeLists.txt --- a/mlir/tools/mlir-shlib/CMakeLists.txt +++ b/mlir/tools/mlir-shlib/CMakeLists.txt @@ -31,6 +31,11 @@ endif() if(LLVM_BUILD_LLVM_DYLIB) + if(MLIR_ENABLE_CUDA_RUNNER) + find_library(CUDA_DRIVER_LIBRARY cuda REQUIRED) + list(APPEND _DEPS ${CUDA_DRIVER_LIBRARY}) + endif() + add_mlir_library( MLIR SHARED