diff --git a/tools/packaging/kata-deploy/local-build/kata-deploy-binaries.sh b/tools/packaging/kata-deploy/local-build/kata-deploy-binaries.sh index 682efb57d514..a7a1ec11b3d8 100755 --- a/tools/packaging/kata-deploy/local-build/kata-deploy-binaries.sh +++ b/tools/packaging/kata-deploy/local-build/kata-deploy-binaries.sh @@ -414,6 +414,39 @@ install_initrd_mariner() { install_initrd "mariner" } +#Instal NVIDIA GPU image +install_image_nvidia_gpu() { + export AGENT_POLICY="yes" + export AGENT_INIT="yes" + export EXTRA_PKGS="apt udev" + install_image "nvidia-gpu" +} + +#Install NVIDIA GPU initrd +install_initrd_nvidia_gpu() { + export AGENT_POLICY="yes" + export AGENT_INIT="yes" + export EXTRA_PKGS="apt udev" + install_initrd "nvidia-gpu" +} + +#Instal NVIDIA GPU confidential image +install_image_nvidia_gpu_confidential() { + export AGENT_POLICY="yes" + export AGENT_INIT="yes" + export EXTRA_PKGS="apt udev" + install_image "nvidia-gpu-confidential" +} + +#Install NVIDIA GPU confidential initrd +install_initrd_nvidia_gpu_confidential() { + export AGENT_POLICY="yes" + export AGENT_INIT="yes" + export EXTRA_PKGS="apt udev" + install_initrd "nvidia-gpu-confidential" +} + + install_se_image() { info "Create IBM SE image configured with AA_KBC=${AA_KBC}" "${se_image_builder}" --destdir="${destdir}" @@ -1024,6 +1057,14 @@ handle_build() { rootfs-initrd-mariner) install_initrd_mariner ;; + rootfs-nvidia-gpu-image) install_image_nvidia_gpu ;; + + rootfs-nvidia-gpu-initrd) install_initrd_nvidia_gpu ;; + + rootfs-nvidia-gpu-confidential-image) install_image_nvidia_gpu_confidential ;; + + rootfs-nvidia-gpu-confidential-initrd) install_initrd_nvidia_gpu_confidential ;; + runk) install_runk ;; shim-v2) install_shimv2 ;;