Skip to content

feat(nvidia): support RTX PRO 6000 Blackwell dynamic MIG - #2135

Merged
hami-robot[bot] merged 1 commit into
Project-HAMi:masterfrom
shkatara:include-mig-support-for-rtx-6000-blackwell
Jul 27, 2026
Merged

feat(nvidia): support RTX PRO 6000 Blackwell dynamic MIG#2135
hami-robot[bot] merged 1 commit into
Project-HAMi:masterfrom
shkatara:include-mig-support-for-rtx-6000-blackwell

Conversation

@shkatara

@shkatara shkatara commented Jul 26, 2026

Copy link
Copy Markdown
Contributor

What type of PR is this?

/kind feature

What this PR does / why we need it:

Why this is needed:

  1. HAMi currently bundles nvidia-mig-parted v0.12.2, which uses go-nvlib v0.7.3 and go-nvml v0.12.9.
  2. That go-nvlib version does not recognize the newer Blackwell MIG profile IDs. PCI device ID 2bb5 (vendor ID 10DE) belongs to the NVIDIA Corporation [RTX PRO 6000 Blackwell / GB202GL](1.2.1, 1.2.3) professional graphics card.
  3. Parsing profiles such as 1g.24gb and 2g.48gb therefore fails. No MIG devices are created, but nvidia-mig-parted still returns successfully and prints MIG configuration applied successfully.

This PR:

  • Adds dynamic MIG geometries for the NVIDIA RTX PRO 6000 Blackwell Server Edition:
    • 1g.24gb × 4
    • 2g.48gb × 2
    • 4g.96gb × 1
  • Adds the model and geometries to the default Helm device configuration.
  • Documents the supported profiles and covers matching the full NVML model name.
  • Updates the bundled nvidia-mig-parted to v0.12.3, which includes native MIG profile support for this GPU.

Which issue(s) this PR fixes:

Fixes #2131

Special notes for your reviewer:

Validated with:

go test -count=1 ./pkg/device-plugin/nvidiadevice/nvinternal/plugin

The native MIG configuration was also exercised on RTX PRO 6000 Blackwell Server Edition hardware.

Cursor AI assisted with investigation, implementation, tests, documentation, and drafting this pull request. The resulting changes were reviewed and validated locally.

Does this PR introduce a user-facing change?:

HAMi dynamic MIG now supports NVIDIA RTX PRO 6000 Blackwell Server Edition GPUs with their native 1g.24gb, 2g.48gb, and 4g.96gb profiles.

Add the supported 1g.24gb, 2g.48gb, and 4g.96gb geometries for the
RTX PRO 6000 Blackwell Server Edition, including chart defaults,
documentation, and model-matching coverage.

Update the bundled nvidia-mig-parted to v0.12.3, which includes native
MIG profile support for this GPU.

Signed-off-by: Shubham Katara <shubham.katara@trivago.com>
@coderabbitai

coderabbitai Bot commented Jul 26, 2026

Copy link
Copy Markdown

Review Change Stack

📝 Walkthrough

Walkthrough

The bundled nvidia-mig-parted version is upgraded to v0.12.3, and scheduler configuration, documentation, and tests add MIG geometry support for the RTX PRO 6000 Blackwell Server Edition.

Changes

Blackwell MIG support

Layer / File(s) Summary
Update bundled MIG tooling
docker/Dockerfile, docker/Dockerfile.hamimaster, docker/Dockerfile.withlib
Docker builds now install nvidia-mig-parted v0.12.3 instead of v0.12.2.
Add Blackwell MIG geometries
charts/hami/templates/scheduler/device-configmap.yaml, pkg/device-plugin/nvidiadevice/nvinternal/plugin/util_test.go, docs/develop/dynamic-mig.md
Scheduler configuration, documentation, and tests define and validate the 1g.24gb, 2g.48gb, and 4g.96gb profiles for the RTX PRO 6000 Blackwell Server Edition.

Estimated code review effort: 2 (Simple) | ~10 minutes

Possibly related PRs

Suggested labels: enhancement

Suggested reviewers: archlitchi

Poem

A rabbit hops through MIGs so bright,
Three Blackwell shapes now fit just right.
The builder brings new tools to play,
Tests guard each profile through the day.
“Hare-brained configs,” we cheer with delight!

🚥 Pre-merge checks | ✅ 5
✅ Passed checks (5 passed)
Check name Status Explanation
Linked Issues check ✅ Passed The changes add RTX PRO 6000 Blackwell MIG geometries and upgrade bundled nvidia-mig-parted to v0.12.3, matching the issue's requirements.
Out of Scope Changes check ✅ Passed The PR stays focused on MIG support, config, docs, and validation; no unrelated changes are evident.
Docstring Coverage ✅ Passed No functions found in the changed files to evaluate docstring coverage. Skipping docstring coverage check.
Description Check ✅ Passed Check skipped - CodeRabbit’s high-level summary is enabled.
Title check ✅ Passed The title clearly summarizes the main change: adding dynamic MIG support for the RTX PRO 6000 Blackwell.
✨ Finishing Touches
🧪 Generate unit tests (beta)
  • Create PR with unit tests

Thanks for using CodeRabbit! It's free for OSS, and your support helps us grow. If you like it, consider giving us a shout-out.

❤️ Share

Comment @coderabbitai help to get the list of available commands.

@FouoF

FouoF commented Jul 27, 2026

Copy link
Copy Markdown
Contributor

/lgtm

@hami-robot hami-robot Bot added the lgtm label Jul 27, 2026
@codecov

codecov Bot commented Jul 27, 2026

Copy link
Copy Markdown

Codecov Report

✅ All modified and coverable lines are covered by tests.

Flag Coverage Δ
unittests 61.96% <ø> (-0.02%) ⬇️

Flags with carried forward coverage won't be shown. Click here to find out more.
see 1 file with indirect coverage changes

🚀 New features to boost your workflow:
  • ❄️ Test Analytics: Detect flaky tests, report on failures, and find test suite problems.

@archlitchi archlitchi left a comment

Copy link
Copy Markdown
Member

Choose a reason for hiding this comment

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

/lgtm

@hami-robot

hami-robot Bot commented Jul 27, 2026

Copy link
Copy Markdown
Contributor

[APPROVALNOTIFIER] This PR is APPROVED

This pull-request has been approved by: archlitchi, shkatara

The full list of commands accepted by this bot can be found here.

The pull request process is described here

Details Needs approval from an approver in each of these files:

Approvers can indicate their approval by writing /approve in a comment
Approvers can cancel approval by writing /approve cancel in a comment

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Projects

None yet

Development

Successfully merging this pull request may close these issues.

Bundled nvidia-mig-parted does not support RTX PRO 6000 Blackwell Server Edition

3 participants