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

Merge pull request #148 from hlhr202/feature/fix-metal

Browse source 
Feature/fix metal
This commit is contained in:
Niko 2024-06-03 04:42:17 +00:00 • committed by GitHub
commit f1030ef762
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
5 changed files with 32 additions and 5 deletions
Download patch file Download diff file Expand all files Collapse all files

8
src/standalone.rs
View file

@ -81,7 +81,13 @@ pub unsafe fn set_log_callback(
log_callback: crate::WhisperLogCallback,
user_data: *mut std::ffi::c_void,
) {
unsafe { whisper_rs_sys::whisper_log_set(log_callback, user_data) }
unsafe {
whisper_rs_sys::whisper_log_set(log_callback, user_data);
#[cfg(feature = "metal")]
{
whisper_rs_sys::ggml_backend_metal_log_set_callback(log_callback, user_data);
}
}
}
/// Print system information.

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_backend_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_backend_metal_log_set_callback(
Some(whisper_cpp_tracing_trampoline),
std::ptr::null_mut(),
);
}
});
}