From ab440a9eb7d10cf6e39455f7e6eb11a80ba0189a Mon Sep 17 00:00:00 2001 From: Andri Saar Date: Thu, 11 Jul 2024 18:04:09 +0000 Subject: [PATCH] Fix a typo in the orchestrator, and push the new version of the base image. With this, we now have GPUs in the container! Bug: 298933839 Change-Id: Ic0c12290e7f677a8eb72e819d0c578f2d74ea2e2 --- WORKSPACE | 2 +- oak_containers_orchestrator/src/container_runtime.rs | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/WORKSPACE b/WORKSPACE index c85e6699b3..68f6444f53 100644 --- a/WORKSPACE +++ b/WORKSPACE @@ -264,7 +264,7 @@ oci_pull( oci_pull( name = "oak_containers_nvidia_sysimage_base", - digest = "sha256:6c84579fbe5f0b8f9b311313f84d82065fa5922ce2ab134dfd98b515d673e7ca", + digest = "sha256:fdec1e1dff19a21899c40f5004fc2d9b8e1f047a4605657eea6119e387168611", image = "europe-west2-docker.pkg.dev/oak-ci/oak-containers-sysimage-base/oak-containers-nvidia-sysimage-base", ) diff --git a/oak_containers_orchestrator/src/container_runtime.rs b/oak_containers_orchestrator/src/container_runtime.rs index 5bb78d917a..498d4f57d9 100644 --- a/oak_containers_orchestrator/src/container_runtime.rs +++ b/oak_containers_orchestrator/src/container_runtime.rs @@ -57,7 +57,7 @@ pub async fn run( && entry.path().extension().unwrap_or_default() == OsStr::new("json") { let cdi_spec = crate::cdi::CdiSpec::new(entry.path())?; - if cdi_spec.kind() != "nvidia.com/gpu" { + if cdi_spec.kind() == "nvidia.com/gpu" { cdi_spec.inject("all", &mut spec)?; } }