feat(nvidia): support RTX PRO 6000 Blackwell dynamic MIG - #2135
Conversation
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>
📝 WalkthroughWalkthroughThe bundled ChangesBlackwell MIG support
Estimated code review effort: 2 (Simple) | ~10 minutes Possibly related PRs
Suggested labels: Suggested reviewers: Poem
🚥 Pre-merge checks | ✅ 5✅ Passed checks (5 passed)
✨ Finishing Touches🧪 Generate unit tests (beta)
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. Comment |
|
/lgtm |
Codecov Report✅ All modified and coverable lines are covered by tests.
Flags with carried forward coverage won't be shown. Click here to find out more. 🚀 New features to boost your workflow:
|
|
[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 DetailsNeeds approval from an approver in each of these files:
Approvers can indicate their approval by writing |
What type of PR is this?
/kind feature
What this PR does / why we need it:
Why this is needed:
nvidia-mig-partedv0.12.2, which usesgo-nvlibv0.7.3 andgo-nvmlv0.12.9.go-nvlibversion 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.1g.24gband2g.48gbtherefore fails. No MIG devices are created, butnvidia-mig-partedstill returns successfully and printsMIG configuration applied successfully.This PR:
1g.24gb× 42g.48gb× 24g.96gb× 1nvidia-mig-partedto 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:
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, and4g.96gbprofiles.