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

fix: metal

Browse source
This commit is contained in:
hlhr202 2024-05-07 00:52:33 +08:00
parent 4dca14d5ec
commit 4bc5709e58
4 changed files with 38 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

9
src/whisper_sys_log.rs
View file

@ -37,6 +37,13 @@ unsafe extern "C" fn whisper_cpp_log_trampoline(
/// You should only call this once (subsequent calls have no ill effect).
pub fn install_whisper_log_trampoline() {
crate::LOG_TRAMPOLINE_INSTALL.call_once(|| unsafe {
whisper_rs_sys::whisper_log_set(Some(whisper_cpp_log_trampoline), std::ptr::null_mut())
whisper_rs_sys::whisper_log_set(Some(whisper_cpp_log_trampoline), std::ptr::null_mut());
#[cfg(feature = "metal")]
{
whisper_rs_sys::ggml_metal_log_set_callback(
Some(whisper_cpp_log_trampoline),
std::ptr::null_mut(),
);
}
});
}

9
src/whisper_sys_tracing.rs
View file

@ -37,6 +37,13 @@ unsafe extern "C" fn whisper_cpp_tracing_trampoline(
/// You should only call this once (subsequent calls have no effect).
pub fn install_whisper_tracing_trampoline() {
crate::LOG_TRAMPOLINE_INSTALL.call_once(|| unsafe {
whisper_rs_sys::whisper_log_set(Some(whisper_cpp_tracing_trampoline), std::ptr::null_mut())
whisper_rs_sys::whisper_log_set(Some(whisper_cpp_tracing_trampoline), std::ptr::null_mut());
#[cfg(feature = "metal")]
{
whisper_rs_sys::ggml_metal_log_set_callback(
Some(whisper_cpp_tracing_trampoline),
std::ptr::null_mut(),
);
}
});
}