Skip to content

fix: pull in additional PNs to GB200xBF3 mlxconfig profile #5544

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
chet merged 1 commit into from
Aug 29, 2026
Merged

fix: pull in additional PNs to GB200xBF3 mlxconfig profile #5544

Changes from all commits
Commits
Show all changes
1 commit
Select commit
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
15 changes: 10 additions & 5 deletions crates/libmlx-model/src/nvconfig.rs
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -43,10 +43,12 @@ const GB200_B3240_V1_PARAMETERS: [&str; 18] = [
"PCI_SWITCH0_UPSTREAM_PORT_PEX=0",
];

const GB200_B3240_PART_NUMBERS: [&str; 3] = [
const GB200_B3240_PART_NUMBERS: [&str; 5] = [
"900-9D3B6-00CN-AB0",
"900-9D3B6-00CN-PA0",
"900-9D3B6-00SN-AB0",
"900-9D3B6-00CN-PN0",
"900-9D3B6-00CN-P_Ax",
];

/// `DpuNvConfigProfile` identifies a fixed version of platform mlxconfig values.
Expand All @@ -59,9 +61,10 @@ pub enum DpuNvConfigProfile {
impl DpuNvConfigProfile {
/// Returns the GB200 profile for an exact supported B3240 part number.
///
/// Accepted part numbers are `900-9D3B6-00CN-AB0`, `900-9D3B6-00SN-AB0`,
/// and `900-9D3B6-00CN-PA0`. Surrounding whitespace and ASCII letter case
/// are ignored. The caller must separately verify that the DPU belongs to a
/// Accepted identities are `900-9D3B6-00CN-AB0`, `900-9D3B6-00SN-AB0`,
/// `900-9D3B6-00CN-PA0`, `900-9D3B6-00CN-PN0`, and
/// `900-9D3B6-00CN-P_Ax`. Surrounding whitespace and ASCII letter case are
/// ignored. The caller must separately verify that the DPU belongs to a
/// GB200 rack. Broader B3240 product prefixes are not accepted.
pub fn for_gb200_b3240_part_number(part_number: &str) -> Option<Self> {
let part_number = part_number.trim();
Expand Down Expand Up @@ -90,10 +93,12 @@ mod tests {
fn gb200_b3240_profile_requires_an_exact_supported_part_number() {
value_scenarios!(
run = DpuNvConfigProfile::for_gb200_b3240_part_number;
"supported part numbers" {
"supported identities" {
"900-9D3B6-00CN-AB0" => Some(DpuNvConfigProfile::Gb200B3240V1),
"900-9D3B6-00SN-AB0" => Some(DpuNvConfigProfile::Gb200B3240V1),
"900-9D3B6-00CN-PA0" => Some(DpuNvConfigProfile::Gb200B3240V1),
"900-9D3B6-00CN-PN0" => Some(DpuNvConfigProfile::Gb200B3240V1),
"900-9D3B6-00CN-P_Ax" => Some(DpuNvConfigProfile::Gb200B3240V1),
" 900-9d3b6-00cn-pa0\n" => Some(DpuNvConfigProfile::Gb200B3240V1),
}

Expand Down
Loading