Skip to content

Comments

Disable auto_sharding test on ROCm.#617

Open
zoranjovanovic-ns wants to merge 1 commit intorocm-jaxlib-v0.8.2from
rocm-jaxlib-v0.8.2-disable_auto_sharding_test
Open

Disable auto_sharding test on ROCm.#617
zoranjovanovic-ns wants to merge 1 commit intorocm-jaxlib-v0.8.2from
rocm-jaxlib-v0.8.2-disable_auto_sharding_test

Conversation

@zoranjovanovic-ns
Copy link

Motivation

Test auto_sharding_gpu_compiler_test is expected to fail on ROCm.

Technical Details

Test should be disabled.

Test Plan

Run the tests.

Test Result

Test is not executed.

Submission Checklist

@zoranjovanovic-ns zoranjovanovic-ns added cherry-pick-candidate Mark a PR to be cherry-picked into the next ROCm JAX. Remove IIF the latest upstream contain the PR. Upstream rocm-jaxlib-v0.8.2 labels Feb 4, 2026
Copy link
Collaborator

@i-chaochen i-chaochen left a comment

Choose a reason for hiding this comment

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

does CI on single-GPU pipeline still check this UT? IIUC we wouldn't run no_oss after we had this 4b08798

@i-chaochen i-chaochen removed cherry-pick-candidate Mark a PR to be cherry-picked into the next ROCm JAX. Remove IIF the latest upstream contain the PR. Upstream labels Feb 4, 2026
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants