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

Merge pull request #169 from arizhih/whisper-cpp-latest

Browse source 
feat: Add support for new whisper.cpp project structure
This commit is contained in:
Niko 2024-09-01 23:15:18 +00:00 • committed by GitHub
commit 8596d2d98c
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
10 changed files with 305 additions and 396 deletions
Download patch file Download diff file Expand all files Collapse all files

60
sys/build.rs
View file

@ -30,11 +30,6 @@ fn main() {
#[cfg(feature = "coreml")]
println!("cargo:rustc-link-lib=static=whisper.coreml");
#[cfg(feature = "opencl")]
{
println!("cargo:rustc-link-lib=clblast");
println!("cargo:rustc-link-lib=OpenCL");
}
#[cfg(feature = "openblas")]
{
if let Ok(openblas_path) = env::var("OPENBLAS_PATH") {
@ -91,6 +86,13 @@ fn main() {
}
}
#[cfg(feature = "openmp")]
{
if target.contains("gnu") {
println!("cargo:rustc-link-lib=gomp");
}
}
println!("cargo:rerun-if-changed=wrapper.h");
let out = PathBuf::from(env::var("OUT_DIR").unwrap());
@ -114,10 +116,11 @@ fn main() {
let bindings = bindgen::Builder::default().header("wrapper.h");
#[cfg(feature = "metal")]
let bindings = bindings.header("whisper.cpp/ggml-metal.h");
let bindings = bindings.header("whisper.cpp/ggml/include/ggml-metal.h");
let bindings = bindings
.clang_arg("-I./whisper.cpp")
.clang_arg("-I./whisper.cpp/include")
.clang_arg("-I./whisper.cpp/ggml/include")
.parse_callbacks(Box::new(bindgen::CargoCallbacks::new()))
.generate();
@ -160,11 +163,11 @@ fn main() {
}
if cfg!(feature = "cuda") {
config.define("WHISPER_CUDA", "ON");
config.define("GGML_CUDA", "ON");
}
if cfg!(feature = "hipblas") {
config.define("WHISPER_HIPBLAS", "ON");
config.define("GGML_HIPBLAS", "ON");
config.define("CMAKE_C_COMPILER", "hipcc");
config.define("CMAKE_CXX_COMPILER", "hipcc");
println!("cargo:rerun-if-env-changed=AMDGPU_TARGETS");
@ -174,20 +177,16 @@ fn main() {
}
if cfg!(feature = "openblas") {
config.define("WHISPER_OPENBLAS", "ON");
}
if cfg!(feature = "opencl") {
config.define("WHISPER_CLBLAST", "ON");
config.define("GGML_BLAS", "ON");
}
if cfg!(feature = "metal") {
config.define("WHISPER_METAL", "ON");
config.define("WHISPER_METAL_NDEBUG", "ON");
config.define("WHISPER_METAL_EMBED_LIBRARY", "ON");
config.define("GGML_METAL", "ON");
config.define("GGML_METAL_NDEBUG", "ON");
config.define("GGML_METAL_EMBED_LIBRARY", "ON");
} else {
// Metal is enabled by default, so we need to explicitly disable it
config.define("WHISPER_METAL", "OFF");
config.define("GGML_METAL", "OFF");
}
if cfg!(debug_assertions) || cfg!(feature = "force-debug") {
@ -207,18 +206,17 @@ fn main() {
}
}
if cfg!(not(feature = "openmp")) {
config.define("GGML_OPENMP", "OFF");
}
let destination = config.build();
if target.contains("window") && !target.contains("gnu") {
println!(
"cargo:rustc-link-search={}",
out.join("build").join("Release").display()
);
} else {
println!("cargo:rustc-link-search={}", out.join("build").display());
}
add_link_search_path(&out.join("lib")).unwrap();
println!("cargo:rustc-link-search=native={}", destination.display());
println!("cargo:rustc-link-lib=static=whisper");
println!("cargo:rustc-link-lib=static=ggml");
// for whatever reason this file is generated during build and triggers cargo complaining
_ = std::fs::remove_file("bindings/javascript/package.json");
@ -236,3 +234,13 @@ fn get_cpp_link_stdlib(target: &str) -> Option<&'static str> {
Some("stdc++")
}
}
fn add_link_search_path(dir: &std::path::Path) -> std::io::Result<()> {
if dir.is_dir() {
println!("cargo:rustc-link-search={}", dir.display());
for entry in std::fs::read_dir(dir)? {
add_link_search_path(&entry?.path())?;
}
}
Ok(())
}