From 9dd66a3b4bc69e1670c7937ba305b880dc41e8ec Mon Sep 17 00:00:00 2001
From: Zero <zero@imaskeleton.me>
Date: Fri, 5 May 2023 22:45:08 -0600
Subject: [PATCH] add libraries to CUDA search path

---
 sys/build.rs | 9 +++++++++
 1 file changed, 9 insertions(+)

diff --git a/sys/build.rs b/sys/build.rs
index 9d3d658..e10e32f 100644
--- a/sys/build.rs
+++ b/sys/build.rs
@@ -31,6 +31,15 @@ fn main() {
         println!("cargo:rustc-link-lib=clblast");
         println!("cargo:rustc-link-lib=OpenCL");
     }
+    #[cfg(feature = "cuda")]
+    {
+        println!("cargo:rustc-link-lib=cublas");
+        println!("cargo:rustc-link-lib=culibos");
+        println!("cargo:rustc-link-lib=cudart");
+        println!("cargo:rustc-link-lib=cublasLt");
+        println!("cargo:rustc-link-search=/usr/local/cuda/lib64");
+        println!("cargo:rustc-link-search=/opt/cuda/lib64");
+    }
     println!("cargo:rerun-if-changed=wrapper.h");
 
     if env::var("WHISPER_DONT_GENERATE_BINDINGS").is_ok() {