trending/OpenShell
1
0
Fork 0
mirror of https://github.com/NVIDIA/OpenShell.git synced 2026-10-26 06:52:57 +08:00
Code Issues Packages Projects Releases Wiki Activity

test(e2e): default GPU probe image (#1450)

Browse Source 
Signed-off-by: Evan Lezar <elezar@nvidia.com>
This commit is contained in:
Evan Lezar
2026-05-20 13:25:09 +02:00
committed by GitHub
parent 2a065a5ff2
commit cade0bb0ac
1 changed files with 2 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files
e2e/rust/tests/gpu_device_selection.rs
+2 -7
View File
@@ -21,6 +21,7 @@ const SANDBOX_CREATE_TIMEOUT: Duration = Duration::from_secs(600);
const CDI_GPU_DEVICE_ALL: &str = "nvidia.com/gpu=all";
const CDI_GPU_DEVICE_PREFIX: &str = "nvidia.com/gpu=";
const GPU_PROBE_IMAGE_ENV: &str = "OPENSHELL_E2E_GPU_PROBE_IMAGE";
const DEFAULT_GPU_PROBE_IMAGE: &str = "nvcr.io/nvidia/base/ubuntu:noble-20251013";
fn gpu_lines(output: &str) -> Vec<String> {
strip_ansi(output)
@@ -36,13 +37,7 @@ fn gpu_probe_image() -> String {
.ok()
.map(|value| value.trim().to_string())
.filter(|value| !value.is_empty())
.unwrap_or_else(|| {
panic!(
"{GPU_PROBE_IMAGE_ENV} must be set to a container image that supports \
NVIDIA Container Toolkit CDI injection (set by \
.github/workflows/e2e-gpu-test.yaml in CI)"
)
})
.unwrap_or_else(|| DEFAULT_GPU_PROBE_IMAGE.to_string())
}
fn object_string<'a>(object: &'a Map<String, Value>, key: &str) -> Option<&'a str> {