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 }