Skip to content

Comments

Sync mgpu tests with xla_mgpu config#597

Open
alekstheod wants to merge 1 commit intorocm-jaxlib-v0.7.1from
sync_mgpu_testing_0.7.1
Open

Sync mgpu tests with xla_mgpu config#597
alekstheod wants to merge 1 commit intorocm-jaxlib-v0.7.1from
sync_mgpu_testing_0.7.1

Conversation

@alekstheod
Copy link
Collaborator

Migrate changes from: #592

Copilot AI review requested due to automatic review settings January 28, 2026 14:20
Copy link

Copilot AI left a comment

Choose a reason for hiding this comment

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

Pull request overview

This PR synchronizes multi-GPU test configurations by migrating changes from an upstream XLA pull request. The changes introduce new CI configuration options to better handle multi-GPU and single-GPU test scenarios.

Changes:

  • Added logic to handle --config=ci_multi_gpu and --config=ci_single_gpu flags in the test runner
  • Created a new ci_multi_gpu configuration that maps to the existing xla_mgpu config

Reviewed changes

Copilot reviewed 2 out of 2 changed files in this pull request and generated no comments.

File Description
build_tools/rocm/run_xla_ci_build.sh Adds conditional logic to set appropriate tag filters based on CI configuration type (multi-GPU vs single-GPU)
build_tools/rocm/rocm_xla.bazelrc Maps the new ci_multi_gpu configuration to the existing xla_mgpu test configuration

💡 Add Copilot custom instructions for smarter, more guided reviews. Learn how to get started.

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.

1 participant