Skip to content

Commit

Permalink
gpu: Add build targets for GPU rootfs initrd/image
Browse files Browse the repository at this point in the history
Preparation for complete GPU rootfs build step #1/#N

Signed-off-by: Zvonko Kaiser <zkaiser@nvidia.com>
  • Loading branch information
zvonkok committed May 10, 2024
1 parent 20515fe commit a8a17fc
Showing 1 changed file with 42 additions and 0 deletions.
42 changes: 42 additions & 0 deletions tools/packaging/kata-deploy/local-build/kata-deploy-binaries.sh
Original file line number Diff line number Diff line change
Expand Up @@ -414,6 +414,40 @@ 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}"
Expand Down Expand Up @@ -1024,6 +1058,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 ;;
Expand Down

0 comments on commit a8a17fc

Please sign in to comment.