diff --git a/providers/gpgpu/bin/gpu_passthrough.py b/providers/gpgpu/bin/gpu_passthrough.py index c4a3c3d0e8..4c821fdb31 100755 --- a/providers/gpgpu/bin/gpu_passthrough.py +++ b/providers/gpgpu/bin/gpu_passthrough.py @@ -42,8 +42,8 @@ "config_cmds": [ "apt-get -q update -y", "apt-get -q upgrade -y", - "apt-get -q install -y linux-generic ubuntu-drivers-common", - "ubuntu-drivers install --gpgpu", + "apt-get -q install -y ubuntu-drivers-common", + "ubuntu-drivers install --recommended", ], }, }, diff --git a/providers/gpgpu/units/jobs.pxu b/providers/gpgpu/units/jobs.pxu index 952bba9232..1dd2bb44be 100644 --- a/providers/gpgpu/units/jobs.pxu +++ b/providers/gpgpu/units/jobs.pxu @@ -82,7 +82,24 @@ requires: package.name == 'lxd' or package.name == 'lxd-installer' or snap.name == 'lxd' category_id: gpgpu plugin: shell +estimated_duration: 1m 45s command: gpu_passthrough.py --vendor=nvidia --pci={pci_device_name} lxd _purpose: Creates a LXD container and passes {pci_device_name} GPU through _summary: Test LXD GPU passthrough on NVIDIA GPU {pci_device_name} _template-summary: Test LXD GPU passthrough on NVIDIA GPU + +unit: template +template-resource: graphics_card +template-filter: graphics_card.vendor == 'NVIDIA Corporation' +template-id: gpgpu/lxdvm-nvidia-gpu-passthrough-pci-device-name +id: gpgpu/lxdvm-nvidia-gpu-passthrough-{pci_device_name} +requires: + executable.name == 'lxc' + package.name == 'lxd' or package.name == 'lxd-installer' or snap.name == 'lxd' +category_id: gpgpu +plugin: shell +estimated_duration: 12m +command: gpu_passthrough.py --vendor=nvidia --pci={pci_device_name} lxdvm +_purpose: Creates a LXD virtual machine and passes {pci_device_name} GPU through +_summary: Test LXD VM GPU passthrough on NVIDIA GPU {pci_device_name} +_template-summary: Test LXD VM GPU passthrough on NVIDIA GPU diff --git a/providers/gpgpu/units/test-plan.pxu b/providers/gpgpu/units/test-plan.pxu index efb6dcb7b3..2bf6828bed 100644 --- a/providers/gpgpu/units/test-plan.pxu +++ b/providers/gpgpu/units/test-plan.pxu @@ -38,7 +38,8 @@ _name: GPGPU Virtualization Passthrough Testing _description: Automated Tests for GPGPU Passthrough (non-graphical) include: - gpgpu/lxd-nvidia-gpu-passthrough-pci-device-name certification-status=blocker + gpgpu/lxd-nvidia-gpu-passthrough-pci-device-name certification-status=blocker + gpgpu/lxdvm-nvidia-gpu-passthrough-pci-device-name certification-status=blocker bootstrap_include: graphics_card executable