harald/wyoming-whisper-rs
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Add an OpenBLAS feature

Browse source
This commit is contained in:
Marcin Mielniczuk 2023-10-08 14:00:38 +02:00
parent f79bc283d3
commit 99ebf2bea5
4 changed files with 20 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

7
sys/build.rs
View file

@ -31,6 +31,10 @@ fn main() {
println!("cargo:rustc-link-lib=clblast");
println!("cargo:rustc-link-lib=OpenCL");
}
#[cfg(feature = "openblas")]
{
println!("cargo:rustc-link-lib=openblas");
}
#[cfg(feature = "cuda")]
{
println!("cargo:rustc-link-lib=cublas");
@ -109,6 +113,9 @@ fn main() {
#[cfg(feature = "cuda")]
cmd.arg("-DWHISPER_CUBLAS=ON");
#[cfg(feature = "openblas")]
cmd.arg("-DWHISPER_OPENBLAS=ON");
#[cfg(feature = "opencl")]
cmd.arg("-DWHISPER_CLBLAST=ON");