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]]
name = "breadman"
version = "0.1.0"
version = "0.2.0"
dependencies = [
"anyhow",
"breadpad-shared",
@ -331,7 +331,7 @@ dependencies = [
[[package]]
name = "breadpad"
version = "0.1.0"
version = "0.2.0"
dependencies = [
"anyhow",
"breadpad-shared",
@ -350,7 +350,7 @@ dependencies = [
[[package]]
name = "breadpad-shared"
version = "0.1.0"
version = "0.2.0"
dependencies = [
"anyhow",
"bread-theme",
@ -376,7 +376,7 @@ dependencies = [
[[package]]
name = "breadpad-test"
version = "0.1.0"
version = "0.2.0"
dependencies = [
"anyhow",
"breadpad-shared",

2
Cargo.toml
View file

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

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

@ -247,12 +247,18 @@ fn try_load_session(
path: &std::path::Path,
) -> (Option<ort::session::Session>, ExecutionProvider) {
// Try ROCm (iGPU) first, fall back to CPU.
match build_onnx_session(path, ort::ep::ROCm::default().build()) {
Ok(s) => {
tracing::info!("ONNX session loaded (ROCm iGPU)");
return (Some(s), ExecutionProvider::Gpu);
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()) {
Ok(s) => {
tracing::info!("ONNX session loaded (ROCm iGPU)");
return (Some(s), ExecutionProvider::Gpu);
}
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()) {
Ok(s) => {