dtourolle/DarkRoom
1
0
Fork 0
Code Issues 39 Pull Requests Actions Packages Projects Releases 20 Wiki Activity

Let ep_probe name the OpenVINO GPU

Browse Source 
On the hybrid laptop OpenVINO's `GPU` was the RTX 3050 through NVIDIA's
OpenCL, not the Iris Xe; DARKROOM_OV_GPU picks GPU.0, GPU.1 and so on.
This commit is contained in:
dtourolle
2026-10-04 21:00:15 -04:00
parent 0291b80672
commit 555ec0efb3
1 changed files with 5 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
core/dr-inference-engine/examples/ep_probe.rs
+5 -2
View File
@@ -69,6 +69,9 @@ fn build(ep: Ep, bytes: &[u8], threads: usize, cache: &Path) -> ort::Result<ort:
let _ = std::fs::create_dir_all(&d); let _ = std::fs::create_dir_all(&d);
d.to_string_lossy().into_owned() d.to_string_lossy().into_owned()
}; };
// `GPU` is OpenVINO's first OpenCL GPU, which on a hybrid laptop can be
// the discrete NVIDIA one; DARKROOM_OV_GPU=GPU.1 names another.
let gpu = std::env::var("DARKROOM_OV_GPU").unwrap_or_else(|_| "GPU".into());
match ep { match ep {
Ep::Cpu => {} Ep::Cpu => {}
Ep::MiGraphX => migraphx(&mut b, false, &cache.join("f32"))?, Ep::MiGraphX => migraphx(&mut b, false, &cache.join("f32"))?,
@@ -79,7 +82,7 @@ fn build(ep: Ep, bytes: &[u8], threads: usize, cache: &Path) -> ort::Result<ort:
&mut b, &mut b,
c"OpenVINO", c"OpenVINO",
&[ &[
("device_type", "GPU".into()), ("device_type", gpu.clone()),
("precision", "FP32".into()), ("precision", "FP32".into()),
("cache_dir", dir("ov-gpu-f32")), ("cache_dir", dir("ov-gpu-f32")),
], ],
@@ -88,7 +91,7 @@ fn build(ep: Ep, bytes: &[u8], threads: usize, cache: &Path) -> ort::Result<ort:
&mut b, &mut b,
c"OpenVINO", c"OpenVINO",
&[ &[
("device_type", "GPU".into()), ("device_type", gpu.clone()),
("precision", "FP16".into()), ("precision", "FP16".into()),
("cache_dir", dir("ov-gpu-fp16")), ("cache_dir", dir("ov-gpu-fp16")),
], ],