Skip to content

feat: azurelinux add nvidia vgpu driver installation selection#7986

Draft
miz060 wants to merge 1 commit intomainfrom
mitchzhu/azl-grid-gpu_driver-pr
Draft

feat: azurelinux add nvidia vgpu driver installation selection#7986
miz060 wants to merge 1 commit intomainfrom
mitchzhu/azl-grid-gpu_driver-pr

Conversation

@miz060
Copy link
Member

@miz060 miz060 commented Feb 27, 2026

What this PR does / why we need it:
Converged GPU sizes (NVads_A10_v5, NCads_A10_v4) require NVIDIA GRID vGPU guest drivers instead of standard CUDA drivers. Previously, AzureLinux 3.0 had no GRID driver support and cannot support these sizes. This PR adds azurelinux GRID driver installation logic, routing converged sizes to the GRID driver path based on NVIDIA_GPU_DRIVER_TYPE while leaving the existing cuda/cuda-open selection unchanged for all other GPU SKUs.

Which issue(s) this PR fixes:

Fixes #

Validation:

Copilot AI review requested due to automatic review settings February 27, 2026 20:59
@miz060
Copy link
Member Author

miz060 commented Feb 27, 2026

Waiting for nvidia-vgpu-guest-driver to land in PMC.

Copy link
Contributor

Copilot AI left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Pull request overview

Adds NVIDIA GRID (vGPU guest) driver selection for converged A10 SKUs on Azure Linux/Mariner by branching GPU driver installation based on NVIDIA_GPU_DRIVER_TYPE, and extends ShellSpec coverage for the routing logic.

Changes:

  • Add downloadGridDrivers() and route converged SKUs (NVIDIA_GPU_DRIVER_TYPE=grid) to GRID installation in downloadGPUDrivers().
  • Add ShellSpec tests validating GRID vs CUDA vs CUDA-open routing behavior.

Reviewed changes

Copilot reviewed 2 out of 2 changed files in this pull request and generated 6 comments.

File Description
parts/linux/cloud-init/artifacts/mariner/cse_install_mariner.sh Introduces GRID driver install function and selects GRID vs CUDA driver flow based on NVIDIA_GPU_DRIVER_TYPE.
spec/parts/linux/cloud-init/artifacts/cse_install_mariner_spec.sh Adds tests that validate the new selection/routing behavior without performing real downloads.

Add GRID(vGPU) driver installation logic for azurelinux with A10 GPU VM
sizes

Signed-off-by: Mitch Zhu <mitchzhu@microsoft.com>
@miz060 miz060 force-pushed the mitchzhu/azl-grid-gpu_driver-pr branch from f869d83 to 9730e58 Compare March 6, 2026 01:27
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants