[AKS] az aks nodepool add/update --enable-managed-gpu option by runzhen · Pull Request #9704 · Azure/azure-cli-extensions

runzhen · 2026-03-19T17:43:24Z

Add managed GPU enablement option to node pool property in az aks nodepool add and az aks nodepool update

This checklist is used to make sure that common guidelines for a pull request are followed.

Related command

az aks nodepool add
az aks nodepool update

General Guidelines

Have you run azdev style <YOUR_EXT> locally? (pip install azdev required)
Have you run python scripts/ci/test_index.py -q locally? (pip install wheel==0.30.0 required)

For new extensions:

My extension description/summary conforms to the Extension Summary Guidelines.

About Extension Publish

There is a pipeline to automatically build, upload and publish extension wheels.
Once your pull request is merged into main branch, a new pull request will be created to update src/index.json automatically.
You only need to update the version information in file setup.py and historical information in file HISTORY.rst in your PR but do not modify src/index.json.

azure-client-tools-bot-prd · 2026-03-19T17:43:38Z

❌Azure CLI Extensions Breaking Change Test

❌aks-preview

rule cmd_name rule_message suggest_message

❌ 1010 - ParaPropUpdate aks nodepool add cmd aks nodepool add update parameter spot_max_price: updated property default from nan to nan please change property default from nan to nan for parameter spot_max_price of cmd aks nodepool add

⚠️ 1006 - ParaAdd aks nodepool add cmd aks nodepool add added parameter enable_managed_gpu

⚠️ 1006 - ParaAdd aks nodepool update cmd aks nodepool update added parameter enable_managed_gpu

azure-client-tools-bot-prd · 2026-03-19T17:43:41Z

Hi @runzhen,
Please write the description of changes which can be perceived by customers into HISTORY.rst.
If you want to release a new extension version, please update the version in setup.py as well.

Copilot

Pull request overview

Adds a preview --enable-managed-gpu flag to AKS nodepool add/update flows, wiring it through argument parsing, decorators, help text, and tests to set gpuProfile.nvidia.managementMode=Managed.

Changes:

Add --enable-managed-gpu to az aks nodepool add and az aks nodepool update argument sets and help.
Decorate AgentPool objects to set gpu_profile.nvidia.management_mode = "Managed" when enabled.
Add unit and live tests verifying the flag sets gpuProfile.nvidia.managementMode to Managed.

Reviewed changes

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

Show a summary per file

File	Description
src/aks-preview/azext_aks_preview/tests/latest/test_aks_commands.py	Adds live-only tests for nodepool add/update with `--enable-managed-gpu`.
src/aks-preview/azext_aks_preview/tests/latest/test_agentpool_decorator.py	Adds unit tests for context getter + add/update decorator behaviors for managed GPU.
src/aks-preview/azext_aks_preview/custom.py	Extends nodepool add/update custom function signatures with `enable_managed_gpu`.
src/aks-preview/azext_aks_preview/agentpool_decorator.py	Implements `get_enable_managed_gpu`, plus add/update mutation for managed GPU on AgentPool.
src/aks-preview/azext_aks_preview/_params.py	Registers `enable_managed_gpu` CLI argument for nodepool add/update.
src/aks-preview/azext_aks_preview/_help.py	Documents `--enable-managed-gpu` in command help.
src/aks-preview/azext_aks_preview/_consts.py	Adds constants for GPU management mode values.
src/aks-preview/HISTORY.rst	Adds changelog entry for the new managed GPU option.

You can also share your feedback on Copilot code review. Take the survey.

Co-authored-by: Copilot Autofix powered by AI <175728472+Copilot@users.noreply.github.com>

yonzhan · 2026-03-19T18:06:31Z

Thank you for your contribution! We will review the pull request and get back to you soon.

github-actions · 2026-03-19T18:07:17Z

The git hooks are available for azure-cli and azure-cli-extensions repos. They could help you run required checks before creating the PR.

Please sync the latest code with latest dev branch (for azure-cli) or main branch (for azure-cli-extensions).
After that please run the following commands to enable git hooks:

pip install azdev --upgrade
azdev setup -c <your azure-cli repo path> -r <your azure-cli-extensions repo path>

github-actions · 2026-03-19T18:24:17Z

CodeGen Tools Feedback Collection

Thank you for using our CodeGen tool. We value your feedback, and we would like to know how we can improve our product. Please take a few minutes to fill our codegen survey

github-actions · 2026-03-19T18:37:48Z

Hi @runzhen

Release Suggestions

Module: aks-preview

Update VERSION to 19.0.0b28 in src/aks-preview/setup.py

Notes

For more info about extension versioning, please refer to Extension version schema

github-actions · 2026-03-19T18:37:48Z

Hi @runzhen

Release Suggestions

Module: aks-preview

Update VERSION to 19.0.0b28 in src/aks-preview/setup.py

Notes

For more info about extension versioning, please refer to Extension version schema

FumingZhang · 2026-03-23T04:46:17Z

/azp run

azure-pipelines · 2026-03-23T04:46:30Z

Azure Pipelines successfully started running 2 pipeline(s).

FumingZhang · 2026-03-23T23:40:10Z

Queued live test to validate the change.

Re-queued live test

FumingZhang · 2026-03-24T00:13:17Z

/azp run

azure-pipelines · 2026-03-24T00:13:31Z

Azure Pipelines successfully started running 2 pipeline(s).

FumingZhang · 2026-03-24T00:44:58Z

/azp run

azure-pipelines · 2026-03-24T00:45:12Z

Azure Pipelines successfully started running 2 pipeline(s).

FumingZhang · 2026-03-24T02:55:59Z

/azp run

azure-pipelines · 2026-03-24T02:56:13Z

Azure Pipelines successfully started running 2 pipeline(s).

FumingZhang · 2026-03-24T04:27:11Z

There is a bug affecting the CI, and I have reported it as a GitHub issue (#9714). As a temporary solution, please rebase or merge from main to ensure the CI passes. @runzhen

runzhen · 2026-03-24T04:39:50Z

@FumingZhang , can you run the pipeline again?

FumingZhang · 2026-03-24T06:32:54Z

/azp run

azure-pipelines · 2026-03-24T06:33:06Z

Azure Pipelines successfully started running 2 pipeline(s).

FumingZhang · 2026-03-24T08:04:04Z

The CI failed because of bugs #9714 and #9716. Until the CLI team resolves the issue, the current workaround is to rebase or merge your branch with main. This ensures that no other changes from the main branch are mistakenly included in your PR, which could cause validation to fail. @runzhen

runzhen · 2026-03-24T17:55:35Z

@FumingZhang can you run pipeline again ?

FumingZhang · 2026-03-24T23:04:34Z

/azp run

azure-pipelines · 2026-03-24T23:04:48Z

Azure Pipelines successfully started running 2 pipeline(s).

runzhen · 2026-03-25T00:57:02Z

Hi @yanzhudd @FumingZhang , anything else need I to do? can you merge this PR now?

runzhen added 2 commits March 19, 2026 04:50

add managed gpu

dd07270

managed gpu

374e8d3

runzhen requested review from FumingZhang and andyliuliming as code owners March 19, 2026 17:43

Copilot AI review requested due to automatic review settings March 19, 2026 17:43

azure-client-tools-bot-prd bot added this to the April 2026 (2026-04-07) milestone Mar 19, 2026

microsoft-github-policy-service bot requested review from yanzhudd, yonzhan and zhoxing-ms March 19, 2026 17:44

microsoft-github-policy-service bot assigned yanzhudd and zhoxing-ms Mar 19, 2026

microsoft-github-policy-service bot added Auto-Assign Auto assign by bot AKS labels Mar 19, 2026

Copilot AI reviewed Mar 19, 2026

View reviewed changes

Potential fix for pull request finding

0dfbc07

Co-authored-by: Copilot Autofix powered by AI <175728472+Copilot@users.noreply.github.com>

Copilot started reviewing on behalf of runzhen March 19, 2026 18:27 View session

runzhen and others added 3 commits March 19, 2026 13:23

Merge branch 'main' into managed-gpu

3db9dd7

managed gpu

1938f19

managed gpu

6a49051

runzhen changed the title ~~add --enable-managed-gpu option~~ az aks nodepool add/update --enable-managed-gpu option Mar 19, 2026

runzhen added 3 commits March 19, 2026 20:55

managed gpu

1abc43c

managed gpu

719d15f

managed gpu

0499eb0

Merge branch 'main' into managed-gpu

9404ba0

managed gpu

23b7f62

gpu

5a2e3a2

FumingZhang mentioned this pull request Mar 24, 2026

CI: Linter/style checks pick up unrelated files due to missing git fetch and no merge-base #9714

Open

Merge branch 'main' into managed-gpu

941dcde

Merge branch 'main' into managed-gpu

cd40e91

runzhen changed the title ~~az aks nodepool add/update --enable-managed-gpu option~~ [AKS] az aks nodepool add/update --enable-managed-gpu option Mar 24, 2026

gpu

0fb2982

FumingZhang approved these changes Mar 25, 2026

View reviewed changes

yanzhudd approved these changes Mar 25, 2026

View reviewed changes

yanzhudd merged commit 3aa7aa4 into Azure:main Mar 25, 2026
24 checks passed

runzhen deleted the managed-gpu branch March 25, 2026 05:17

Conversation

runzhen commented Mar 19, 2026

Related command

General Guidelines

About Extension Publish

Uh oh!

azure-client-tools-bot-prd bot commented Mar 19, 2026 • edited Loading Uh oh! There was an error while loading. Please reload this page.

Uh oh!

Uh oh!

azure-client-tools-bot-prd bot commented Mar 19, 2026

Uh oh!

Copilot AI left a comment

Choose a reason for hiding this comment

Pull request overview

Reviewed changes

Uh oh!

Uh oh!

Uh oh!

Uh oh!

Uh oh!

Uh oh!

Uh oh!

Uh oh!

Uh oh!

yonzhan commented Mar 19, 2026

Uh oh!

github-actions bot commented Mar 19, 2026

Uh oh!

github-actions bot commented Mar 19, 2026

CodeGen Tools Feedback Collection

Uh oh!

github-actions bot commented Mar 19, 2026

Release Suggestions

Module: aks-preview

Notes

Uh oh!

github-actions bot commented Mar 19, 2026

Release Suggestions

Module: aks-preview

Notes

Uh oh!

FumingZhang commented Mar 23, 2026

Uh oh!

azure-pipelines bot commented Mar 23, 2026

Uh oh!

FumingZhang commented Mar 23, 2026

Uh oh!

FumingZhang commented Mar 24, 2026

Uh oh!

azure-pipelines bot commented Mar 24, 2026

Uh oh!

FumingZhang commented Mar 24, 2026

Uh oh!

azure-pipelines bot commented Mar 24, 2026

Uh oh!

FumingZhang commented Mar 24, 2026

Uh oh!

azure-pipelines bot commented Mar 24, 2026

Uh oh!

FumingZhang commented Mar 24, 2026

Uh oh!

runzhen commented Mar 24, 2026

Uh oh!

FumingZhang commented Mar 24, 2026

Uh oh!

azure-pipelines bot commented Mar 24, 2026

Uh oh!

FumingZhang commented Mar 24, 2026

Uh oh!

runzhen commented Mar 24, 2026

Uh oh!

FumingZhang commented Mar 24, 2026

Uh oh!

azure-pipelines bot commented Mar 24, 2026

Uh oh!

runzhen commented Mar 25, 2026 • edited Loading Uh oh! There was an error while loading. Please reload this page.

Uh oh!

Uh oh!

Uh oh!

Reviewers

azure-client-tools-bot-prd bot commented Mar 19, 2026 •

edited

Loading

runzhen commented Mar 25, 2026 •

edited

Loading