From 3d33a734581d8cc8bb018489622398bac4384603 Mon Sep 17 00:00:00 2001 From: hyiip Date: Wed, 19 Feb 2025 02:55:46 +0800 Subject: [PATCH] link cuda when cuda feature is on --- sys/build.rs | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/sys/build.rs b/sys/build.rs index 8181d24..ecea43e 100644 --- a/sys/build.rs +++ b/sys/build.rs @@ -265,6 +265,10 @@ fn main() { if cfg!(feature = "metal") { println!("cargo:rustc-link-lib=static=ggml-metal"); } + + if cfg!(feature = "cuda") { + println!("cargo:rustc-link-lib=static=ggml-cuda"); + } println!( "cargo:WHISPER_CPP_VERSION={}",