Skip to content

Add GB300 recipe overlays for EKS #2381

Description

@yuanchen8911

Goal

Add service-bound recipe overlays for GB300 on EKS so that aicr recipe --accelerator gb300 --service eks --os ubuntu --intent <training|inference> ... resolves end to end, with a bundle that deploys and validates on a real GB300 NVL72 EKS cluster.

gb300 is already a declared accelerator in pkg/recipe/criteria.go, but recipes/overlays/ still has zero GB300 overlays, so every GB300 query fails to resolve.

Motivation

A GB300 EKS cluster is now available for development and qualification (nhensley-gb300, us-east-2):

  • 2 x p6e-gb300r.36xlarge GPU nodes, Ubuntu 24.04, kernel 6.17.0-1019-aws, containerd 2.2.1
  • 3 x m7i.xlarge system nodes (Amazon Linux 2023)
  • Kubernetes v1.35.6
  • Scale-out fabric is EFA (efa-only ENI on the GPU nodes), not RoCE — same fabric family as the existing GB200 EKS overlays
  • topology.k8s.aws/network-node-layer-{1,2,3} labels present for topology-aware scheduling

This unblocks the overlay work that #1318 was closed as not-planned for (no hardware at the time), and it is the platform half of the ask in #1883 (GB300 NVL72 inference recipe referenced by the DSX NCP docs).

Scope

  1. gb300-any.yaml cross-cutting overlay carrying the deployment-phase floor and component version pins, modeled on gb200-any.yaml.
  2. Service-bound overlays for EKS, modeled on the gb200-eks-* family:
    • gb300-eks-training.yaml / gb300-eks-ubuntu-training.yaml
    • gb300-eks-inference.yaml / gb300-eks-ubuntu-inference.yaml
    • platform variants as warranted (-kubeflow, -inference-dynamo)
  3. Machine-type qualification comments naming p6e-gb300r.36xlarge, consistent with the convention added for the other accelerators.
  4. EFA + MNNVL/IMEX configuration: EFA device plugin (or networking DRA driver, per EFA on AWS: legacy device plugin vs networking DRA driver — should we migrate? #1326), NVLink compute-domain handling for GB300 NVL72.
  5. Performance-phase thresholds tuned per leaf for the GB300 EFA fabric rather than inherited from GB200.
  6. Cover the new overlays in recipes/overlays_validation_floor_test.go.
  7. Run make bom-docs and commit the regenerated docs/user/container-images.md.

Success criteria

  1. aicr recipe --accelerator gb300 --service eks --os ubuntu --intent training and the inference equivalent both resolve without manual overlay authoring.
  2. aicr bundle output deploys cleanly on the GB300 EKS cluster.
  3. aicr validate passes the deployment, conformance, and performance phases on that cluster.
  4. New overlays are covered by the validation-floor test and make qualify is green.

Notes and open items

  • GB300 inference for the DSX NCP use case is tracked separately in Add a GB300 NVL72 inference recipe #1883; this issue covers the EKS overlay family that inference variant would build on. Close or re-scope Add a GB300 NVL72 inference recipe #1883 once the inference leaf lands.
  • The GPU operator and EFA device plugin are not yet installed on the cluster — no node advertises nvidia.com/gpu or vpc.amazonaws.com/efa today. Bringing those up is part of the first deploy.
  • Each GPU node currently exposes a single EFA ENI on network card 0. Confirm whether that is the intended provisioning for p6e-gb300r.36xlarge before setting performance thresholds, since it caps scale-out bandwidth.
  • Internal-only GB300 variants live in the internal recipes repo; keep the public overlays aligned rather than duplicating.

Related: #1883, #1318 (closed not-planned), #1326, #1410.

Metadata

Metadata

Assignees

Labels

area/recipestheme/recipesRecipe expansion, overlays, mixins, and component registry

Projects

No projects

Milestone

No milestone

Relationships

None yet

Development

No branches or pull requests

Issue actions