fix: azurelinux configure GRID vGPU licensing#8106
Merged
Conversation
Set gridd config and restart its systemd service so that the daemon can acquire a license. Signed-off-by: Mitch Zhu <mitchzhu@microsoft.com>
Contributor
There was a problem hiding this comment.
Pull request overview
Updates Linux provisioning to configure NVIDIA GRID vGPU licensing on Mariner/AzureLinux and regenerates the corresponding CustomData snapshot test fixtures used by pkg/agent.
Changes:
- Add Mariner/AzureLinux-only logic in
configGPUDrivers()to setFeatureType=1in/etc/nvidia/gridd.confand (re)enable/(re)startnvidia-griddwhenNVIDIA_GPU_DRIVER_TYPE=grid. - Regenerate multiple
pkg/agent/testdata/**/CustomDatafiles to reflect the updated provisioning payload. - No functional code changes outside the GPU provisioning path; remaining diffs are snapshot updates.
Reviewed changes
Copilot reviewed 32 out of 72 changed files in this pull request and generated 1 comment.
Show a summary per file
| File | Description |
|---|---|
| parts/linux/cloud-init/artifacts/cse_config.sh | Adds GRID vGPU licensing configuration/restart for nvidia-gridd on Mariner/AzureLinux during GPU driver setup. |
| pkg/agent/testdata/MarinerV2+Kata/CustomData | Regenerated CustomData fixture. |
| pkg/agent/testdata/MarinerV2+CustomCloud/CustomData | Regenerated CustomData fixture. |
| pkg/agent/testdata/CustomizedImage/CustomData | Regenerated CustomData fixture. |
| pkg/agent/testdata/AzureLinuxV3+Kata/CustomData | Regenerated CustomData fixture. |
You can also share your feedback on Copilot code review. Take the survey.
miz060
requested review from
AbelHu,
Devinwong,
YaoC,
awesomenix,
calvin197,
cameronmeissner,
djsly,
ganeshkumarashok,
junjiezhang1997,
lilypan26,
mxj220,
pdamianov-dev,
phealy,
r2k1,
sulixu,
surajssd,
timmy-wright,
yewmsft and
zachary-bailey
as code owners
awesomenix
reviewed
Mar 17, 2026
| # GRID vGPU licensing: configure and restart nvidia-gridd after device nodes exist | ||
| if [ "$NVIDIA_GPU_DRIVER_TYPE" = "grid" ]; then | ||
| sed -i -e '/^FeatureType=/d' -e '$ a FeatureType=1' /etc/nvidia/gridd.conf | ||
| systemctl enable nvidia-gridd.service |
Contributor
There was a problem hiding this comment.
we have systemctlEnableAndStart which you can use
Member
Author
There was a problem hiding this comment.
Got it. Updated to use systemctlEnableAndStart as suggested.
awesomenix
approved these changes
Mar 17, 2026
lilypan26
approved these changes
Mar 17, 2026
henryli001
reviewed
Mar 17, 2026
Contributor
There was a problem hiding this comment.
Pull request overview
This PR updates the Linux CSE GPU driver configuration to properly bring up NVIDIA GRID vGPU licensing on Azure Linux/Mariner by starting the nvidia-gridd systemd service after GPU device nodes are available, and refreshes the generated CustomData snapshots accordingly.
Changes:
- Start/enable
nvidia-griddduring GPU driver configuration for Mariner/Azure Linux whenNVIDIA_GPU_DRIVER_TYPE=grid. - Regenerate/update multiple
pkg/agent/testdata/**/CustomDatasnapshot files to reflect the script change.
Reviewed changes
Copilot reviewed 32 out of 72 changed files in this pull request and generated 1 comment.
Show a summary per file
| File | Description |
|---|---|
| parts/linux/cloud-init/artifacts/cse_config.sh | Adds a nvidia-gridd start/enable step for GRID driver type on Mariner/Azure Linux. |
| pkg/agent/testdata/MarinerV2+Kata/CustomData | Updated generated CustomData snapshot (gzip/base64 payload changed). |
| pkg/agent/testdata/MarinerV2+CustomCloud/CustomData | Updated generated CustomData snapshot (gzip/base64 payload changed). |
| pkg/agent/testdata/CustomizedImage/CustomData | Updated generated CustomData snapshot (gzip/base64 payload changed). |
| pkg/agent/testdata/AzureLinuxV3+Kata/CustomData | Updated generated CustomData snapshot (gzip/base64 payload changed). |
You can also share your feedback on Copilot code review. Take the survey.
henryli001
approved these changes
Mar 17, 2026
awesomenix
approved these changes
Mar 17, 2026
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
Add this suggestion to a batch that can be applied as a single commit.This suggestion is invalid because no changes were made to the code.Suggestions cannot be applied while the pull request is closed.Suggestions cannot be applied while viewing a subset of changes.Only one suggestion per line can be applied in a batch.Add this suggestion to a batch that can be applied as a single commit.Applying suggestions on deleted lines is not supported.You must change the existing code in this line in order to create a valid suggestion.Outdated suggestions cannot be applied.This suggestion has been applied or marked resolved.Suggestions cannot be applied from pending reviews.Suggestions cannot be applied on multi-line comments.Suggestions cannot be applied while the pull request is queued to merge.Suggestion cannot be applied right now. Please check back later.
What this PR does / why we need it:
I’ve added the Azure Linux NVIDIA vGPU driver installation path to AgentBaker (#7986). The driver installs and works as expected, but vGPU licensing was not being configured. This change adds the missing setup so vGPU licensing is properly configured. Verified via extra GRID test scenario using AgentBaker E2E, similar to ScenarioUbuntu2404GRID: https://msazure.visualstudio.com/CloudNativeCompute/_build/results?buildId=157022962&view=results
Which issue(s) this PR fixes:
Fixes #