Merge pull request #122 from XUJiahua/win-cuda

fix: LNK2019: unresolved external symbol cuGetErrorString
This commit is contained in:
Niko 2024-03-20 16:37:18 +00:00 committed by GitHub
commit 6226530f2b
No known key found for this signature in database
GPG key ID: B5690EEEBB952194

View file

@ -44,6 +44,7 @@ fn main() {
println!("cargo:rustc-link-lib=cublas"); println!("cargo:rustc-link-lib=cublas");
println!("cargo:rustc-link-lib=cudart"); println!("cargo:rustc-link-lib=cudart");
println!("cargo:rustc-link-lib=cublasLt"); println!("cargo:rustc-link-lib=cublasLt");
println!("cargo:rustc-link-lib=cuda");
cfg_if::cfg_if! { cfg_if::cfg_if! {
if #[cfg(target_os = "windows")] { if #[cfg(target_os = "windows")] {
let cuda_path = PathBuf::from(env::var("CUDA_PATH").unwrap()).join("lib/x64"); let cuda_path = PathBuf::from(env::var("CUDA_PATH").unwrap()).join("lib/x64");