Breadway/breadpad
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

fix: skip ROCm EP registration when not available in ORT build
Some checks failed
release / build (push) Failing after 3s
Details

Browse source 
Eliminates the spurious ERROR log from ORT when ROCm isn't compiled in.
Checks is_available() before attempting registration so the session
correctly falls back to CPU without noise.
This commit is contained in:
Breadway 2026-06-07 15:53:06 +08:00
parent 708eb8f3b4
commit 478d06a5d5
3 changed files with 16 additions and 10 deletions
Download patch file Download diff file Expand all files Collapse all files

8
Cargo.lock generated
View file

@ -314,7 +314,7 @@ dependencies = [
[[package]] [[package]]
name = "breadman" name = "breadman"
version = "0.1.0" version = "0.2.0"
dependencies = [ dependencies = [
"anyhow", "anyhow",
"breadpad-shared", "breadpad-shared",
@ -331,7 +331,7 @@ dependencies = [
[[package]] [[package]]
name = "breadpad" name = "breadpad"
version = "0.1.0" version = "0.2.0"
dependencies = [ dependencies = [
"anyhow", "anyhow",
"breadpad-shared", "breadpad-shared",
@ -350,7 +350,7 @@ dependencies = [
[[package]] [[package]]
name = "breadpad-shared" name = "breadpad-shared"
version = "0.1.0" version = "0.2.0"
dependencies = [ dependencies = [
"anyhow", "anyhow",
"bread-theme", "bread-theme",
@ -376,7 +376,7 @@ dependencies = [
[[package]] [[package]]
name = "breadpad-test" name = "breadpad-test"
version = "0.1.0" version = "0.2.0"
dependencies = [ dependencies = [
"anyhow", "anyhow",
"breadpad-shared", "breadpad-shared",

2
Cargo.toml
View file

@ -8,7 +8,7 @@ members = [
resolver = "2" resolver = "2"
[workspace.package] [workspace.package]
version = "0.1.0" version = "0.2.0"
edition = "2021" edition = "2021"
license = "MIT" license = "MIT"
authors = ["Breadway"] authors = ["Breadway"]

6
breadpad-shared/src/classifier.rs
View file

@ -247,6 +247,11 @@ fn try_load_session(
path: &std::path::Path, path: &std::path::Path,
) -> (Option<ort::session::Session>, ExecutionProvider) { ) -> (Option<ort::session::Session>, ExecutionProvider) {
// Try ROCm (iGPU) first, fall back to CPU. // Try ROCm (iGPU) first, fall back to CPU.
let rocm_available = {
use ort::execution_providers::ExecutionProvider as _;
ort::ep::ROCm::default().is_available().unwrap_or(false)
};
if rocm_available {
match build_onnx_session(path, ort::ep::ROCm::default().build()) { match build_onnx_session(path, ort::ep::ROCm::default().build()) {
Ok(s) => { Ok(s) => {
tracing::info!("ONNX session loaded (ROCm iGPU)"); tracing::info!("ONNX session loaded (ROCm iGPU)");
@ -254,6 +259,7 @@ fn try_load_session(
} }
Err(e) => tracing::debug!("ROCm EP unavailable: {}; trying CPU", e), Err(e) => tracing::debug!("ROCm EP unavailable: {}; trying CPU", e),
} }
}
match build_onnx_session(path, ort::ep::CPU::default().build()) { match build_onnx_session(path, ort::ep::CPU::default().build()) {
Ok(s) => { Ok(s) => {
tracing::info!("ONNX session loaded (CPU)"); tracing::info!("ONNX session loaded (CPU)");