gpu: Add build targets for GPU rootfs initrd/image
This commit is contained in:
Zvonko Kaiser 2024-05-15 13:30:42 +02:00 committed by GitHub
commit 117e2f2ecc
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -413,6 +413,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}"
@ -1028,6 +1061,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 ;;