[AKS] add --gpu-mig-strategy option for aks nodepool add/update by runzhen · Pull Request #9722 · Azure/azure-cli-extensions

runzhen · 2026-03-25T22:23:40Z

add --gpu-mig-strategy option for aks nodepool add/update

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)
My extension version conforms to the Extension version schema

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-25T22:23:43Z

❌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 gpu_mig_strategy

⚠️ 1008 - ParaPropAdd aks nodepool add cmd aks nodepool add update parameter enable_managed_gpu: added property choices=['false', 'true']

⚠️ 1008 - ParaPropAdd aks nodepool add cmd aks nodepool add update parameter enable_managed_gpu: added property nargs=?

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

⚠️ 1008 - ParaPropAdd aks nodepool update cmd aks nodepool update update parameter enable_managed_gpu: added property choices=['false', 'true']

⚠️ 1008 - ParaPropAdd aks nodepool update cmd aks nodepool update update parameter enable_managed_gpu: added property nargs=?

azure-client-tools-bot-prd · 2026-03-25T22:23:46Z

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.

github-actions · 2026-03-25T22:23:54Z

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>

yonzhan · 2026-03-25T22:23:55Z

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

github-actions · 2026-03-25T22:25:35Z

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-25T22:27:53Z

Hi @runzhen

Release Suggestions

Module: aks-preview

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

Notes

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

Copilot

Pull request overview

Adds support in the aks-preview extension for specifying a GPU MIG (Multi-Instance GPU) strategy when adding or updating AKS nodepools, and wires that option through the agentpool decorator into the SDK model.

Changes:

Introduces --gpu-mig-strategy argument (enum) for az aks nodepool add and az aks nodepool update.
Decorates/updates agentpool.gpu_profile.nvidia.mig_strategy during nodepool create/update flows.
Adds unit coverage for the context getter of gpu_mig_strategy and CLI help text for nodepool add.

Reviewed changes

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

Show a summary per file

File	Description
src/aks-preview/azext_aks_preview/agentpool_decorator.py	Adds getter + create/update wiring to set `gpu_profile.nvidia.mig_strategy`; adjusts managed GPU update behavior.
src/aks-preview/azext_aks_preview/_params.py	Adds the new CLI argument + enum list; modifies `enable_managed_gpu` argument registration.
src/aks-preview/azext_aks_preview/_help.py	Documents `--gpu-mig-strategy` in `aks nodepool add` help.
src/aks-preview/azext_aks_preview/_consts.py	Adds constants for MIG strategies (`Single`, `Mixed`).
src/aks-preview/azext_aks_preview/custom.py	Adds `gpu_mig_strategy` parameter to nodepool add/update custom handlers.
src/aks-preview/azext_aks_preview/tests/latest/test_agentpool_decorator.py	Adds unit tests for `get_gpu_mig_strategy()` behavior.

src/aks-preview/azext_aks_preview/agentpool_decorator.py

src/aks-preview/azext_aks_preview/_params.py

src/aks-preview/azext_aks_preview/_help.py

src/aks-preview/azext_aks_preview/agentpool_decorator.py

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

runzhen · 2026-03-25T23:01:52Z

Hello @FumingZhang , do you know why the pipeline fails? I have merge the latest code from main branch

FumingZhang · 2026-03-25T23:42:07Z

The CI failed with this error @runzhen

-  FAIL - HIGH severity: missing_parameter_help
    Parameter: aks nodepool update, `gpu_mig_strategy` - Missing help

https://dev.azure.com/azclitools/public/_build/results?buildId=304989&view=logs&j=e2e497a5-ba27-5eb9-f39e-d929d56974a6&t=61cc2d02-f6f4-5007-71e9-5651933c3fe7&l=188

runzhen · 2026-03-26T04:53:43Z

Hi @FumingZhang , I added the live test case, and excluded it in the json file, the test passed in my local.

src/aks-preview/HISTORY.rst

yanzhudd · 2026-03-26T23:37:14Z

/azp run

yanzhudd · 2026-03-27T02:39:21Z

/azp run

azure-pipelines · 2026-03-27T02:39:33Z

Azure Pipelines successfully started running 2 pipeline(s).

FumingZhang · 2026-03-27T02:49:55Z

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-27T03:37:27Z

Hi @FumingZhang @yanzhudd , I rebased the code, can you review again? thanks

yanzhudd · 2026-03-27T11:41:59Z

/azp run

azure-pipelines · 2026-03-27T11:42:14Z

Azure Pipelines successfully started running 2 pipeline(s).

runzhen · 2026-03-27T19:33:55Z

Hi @FumingZhang @yanzhudd I merged the main branch, but the pipeline still fails

runzhen · 2026-03-30T03:30:47Z

Hi @yanzhudd can you review it again?

yanzhudd · 2026-03-30T23:55:01Z

/azp run

azure-pipelines · 2026-03-30T23:55:15Z

Azure Pipelines successfully started running 2 pipeline(s).

runzhen added 2 commits March 25, 2026 22:13

mig-strategy

147cf5e

mig

8e1713e

runzhen requested review from FumingZhang and andyliuliming as code owners March 25, 2026 22:23

Copilot AI review requested due to automatic review settings March 25, 2026 22:23

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

microsoft-github-policy-service bot requested review from yanzhudd, yonzhan and zhoxing-ms March 25, 2026 22:24

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

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

Copilot started reviewing on behalf of runzhen March 25, 2026 22:24 View session

Copilot AI reviewed Mar 25, 2026

View reviewed changes

runzhen and others added 5 commits March 25, 2026 15:35

Update src/aks-preview/azext_aks_preview/_help.py

887984b

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

Update src/aks-preview/azext_aks_preview/agentpool_decorator.py

cb62719

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

Update src/aks-preview/azext_aks_preview/_params.py

af3cab6

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

Update src/aks-preview/azext_aks_preview/_params.py

265543e

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

mig

677b06c

yonzhan unassigned zhoxing-ms Mar 25, 2026

gpu

18e32fc

runzhen added 2 commits March 26, 2026 04:29

mig

4b4f6d0

Merge branch 'main' into managed-gpu2

36f0ea9

FumingZhang reviewed Mar 26, 2026

View reviewed changes

src/aks-preview/HISTORY.rst Outdated Show resolved Hide resolved

Add MIG strategy option to node pool properties

611e240

runzhen requested a review from FumingZhang March 26, 2026 05:36

FumingZhang approved these changes Mar 26, 2026

View reviewed changes

Merge branch 'main' into managed-gpu2

5af0e11

FumingZhang previously approved these changes Mar 27, 2026

View reviewed changes

Merge branch 'main' into managed-gpu2

c6132f6

Merge branch 'main' into managed-gpu2

1f27edd

runzhen dismissed FumingZhang’s stale review via 1f27edd March 30, 2026 03:02

FumingZhang previously approved these changes Mar 30, 2026

View reviewed changes

Merge branch 'main' into managed-gpu2

f7874a3

runzhen dismissed FumingZhang’s stale review via f7874a3 March 30, 2026 08:25

FumingZhang approved these changes Mar 31, 2026

View reviewed changes

yanzhudd approved these changes Mar 31, 2026

View reviewed changes

yanzhudd merged commit 085b87d into Azure:main Mar 31, 2026
32 checks passed

runzhen deleted the managed-gpu2 branch March 31, 2026 00:32

Conversation

runzhen commented Mar 25, 2026

Related command

General Guidelines

About Extension Publish

Uh oh!

azure-client-tools-bot-prd bot commented Mar 25, 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 25, 2026

Uh oh!

github-actions bot commented Mar 25, 2026

Uh oh!

yonzhan commented Mar 25, 2026

Uh oh!

github-actions bot commented Mar 25, 2026

CodeGen Tools Feedback Collection

Uh oh!

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

Uh oh!

Release Suggestions

Module: aks-preview

Notes

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!

runzhen commented Mar 25, 2026

Uh oh!

FumingZhang commented Mar 25, 2026

Uh oh!

runzhen commented Mar 26, 2026

Uh oh!

Uh oh!

yanzhudd commented Mar 26, 2026

Uh oh!

yanzhudd commented Mar 27, 2026

Uh oh!

azure-pipelines bot commented Mar 27, 2026

Uh oh!

FumingZhang commented Mar 27, 2026

Uh oh!

runzhen commented Mar 27, 2026

Uh oh!

yanzhudd commented Mar 27, 2026

Uh oh!

azure-pipelines bot commented Mar 27, 2026

Uh oh!

runzhen commented Mar 27, 2026

Uh oh!

runzhen commented Mar 30, 2026

Uh oh!

yanzhudd commented Mar 30, 2026

Uh oh!

azure-pipelines bot commented Mar 30, 2026

Uh oh!

Uh oh!

Reviewers

Assignees

Labels

Projects

Milestone

Development

Uh oh!

6 participants

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

edited

Loading

github-actions bot commented Mar 25, 2026 •

edited

Loading