Skip to content

[air] Add GPU_8xB300 support to the experimental AIR CLI - #6649

Open
caroline-db wants to merge 2 commits into
mainfrom
air/b300-support
Open

[air] Add GPU_8xB300 support to the experimental AIR CLI#6649
caroline-db wants to merge 2 commits into
mainfrom
air/b300-support

Conversation

@caroline-db

@caroline-db caroline-db commented Sep 11, 2026

Copy link
Copy Markdown
Contributor

Changes

  • Add GPU_8xB300 support to the experimental AIR CLI with eight GPUs per node and positive accelerator counts that are multiples of eight.
  • Carry B300 through submission, display, filtering, and node-count calculation, with coverage for numeric v6 and databricks_ai_v6 environment routing.
  • Reject B300 conversion through convert-to-dabs until DAB support lands in a separate PR.

Why

Enable direct B300 workloads without coupling the experimental AIR change to Bundle-owned schema and validation changes. Jobs remains authoritative for supported environment versions, including the B300 minimum, for now. A follow-up may incorporate that validation into --dry-run once its final behavior is defined.

Tests

  • Added unit and acceptance coverage for parsing, casing, accelerator counts, v6 submission payloads, display, filtering, and conversion rejection.
  • Passed full Go lint, 10,024 unit tests, and 5,134 acceptance tests.
  • Successfully completed a live B300 smoke workload; streamed device output reported NVIDIA B300 SXM6 AC.

This PR was written with Codex.

@eng-dev-ecosystem-bot

eng-dev-ecosystem-bot commented Sep 11, 2026

Copy link
Copy Markdown
Collaborator

Integration test report

Commit: c51a90b

Run: 34634677430

Env 💚​RECOVERED ✅​pass 🙈​skip Time
💚​ aws linux 1 275 16 9:55
💚​ aws windows 1 277 14 8:05
💚​ azure linux 1 274 16 9:33
💚​ azure windows 1 276 14 5:20
💚​ gcp linux 1 275 16 8:22
💚​ gcp windows 1 277 14 6:20
Test Name aws linux aws windows azure linux azure windows gcp linux gcp windows
💚​ TestAccept 💚​R 💚​R 💚​R 💚​R 💚​R 💚​R
Top 4 slowest tests (at least 2 minutes):
duration env testname
5:10 aws windows TestAccept
3:54 gcp windows TestAccept
3:28 azure windows TestAccept
2:37 azure linux TestFilerWorkspaceFilesExtensionsReadDir

@caroline-db
caroline-db force-pushed the air/b300-support branch 2 times, most recently from 53d3fd1 to 689bdff Compare September 11, 2026 18:35

@ben-hansen-db ben-hansen-db left a comment

Copy link
Copy Markdown
Contributor

Choose a reason for hiding this comment

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

lgtm, note Pardis' work on using our internal list api, we should do that here as well but this is fine for now

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

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

3 participants