From e5f44fae30728c4fd13ce9d2a740fb7818c00eff Mon Sep 17 00:00:00 2001 From: "alex.lyn" Date: Thu, 26 Jun 2025 16:10:13 +0800 Subject: [PATCH] runtime-rs: Add GPU annotations during remote hypervisor preparation Add GPU specific annotations used by remote hypervisor for instance selection during `prepare_vm`. Fixes #10484 Signed-off-by: alex.lyn --- src/runtime-rs/crates/hypervisor/src/remote/inner.rs | 9 +++++++++ 1 file changed, 9 insertions(+) diff --git a/src/runtime-rs/crates/hypervisor/src/remote/inner.rs b/src/runtime-rs/crates/hypervisor/src/remote/inner.rs index 2c4e7caf8..47eec7e2b 100644 --- a/src/runtime-rs/crates/hypervisor/src/remote/inner.rs +++ b/src/runtime-rs/crates/hypervisor/src/remote/inner.rs @@ -12,6 +12,7 @@ use async_trait::async_trait; use kata_types::{ annotations::{ cri_containerd::{SANDBOX_NAMESPACE_LABEL_KEY, SANDBOX_NAME_LABEL_KEY}, + KATA_ANNO_CFG_HYPERVISOR_DEFAULT_GPUS, KATA_ANNO_CFG_HYPERVISOR_DEFAULT_GPU_MODEL, KATA_ANNO_CFG_HYPERVISOR_DEFAULT_MEMORY, KATA_ANNO_CFG_HYPERVISOR_DEFAULT_VCPUS, KATA_ANNO_CFG_HYPERVISOR_IMAGE_PATH, KATA_ANNO_CFG_HYPERVISOR_MACHINE_TYPE, }, @@ -124,6 +125,14 @@ impl RemoteInner { KATA_ANNO_CFG_HYPERVISOR_IMAGE_PATH.to_string(), config.boot_info.image.to_string(), ); + annotations.insert( + KATA_ANNO_CFG_HYPERVISOR_DEFAULT_GPUS.to_string(), + config.remote_info.default_gpus.to_string(), + ); + annotations.insert( + KATA_ANNO_CFG_HYPERVISOR_DEFAULT_GPU_MODEL.to_string(), + config.remote_info.default_gpu_model.to_string(), + ); annotations }