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

Merge pull request #90 from marmistrz/openblas

Browse source 
Add an OpenBLAS feature
This commit is contained in:
Niko 2023-10-08 14:40:46 +00:00 • committed by GitHub
commit 174e111ebc
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
4 changed files with 20 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

1
sys/Cargo.toml
View file

@ -36,6 +36,7 @@ include = [
coreml = []
cuda = []
opencl = []
openblas = []
[dependencies]

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");