Skip to content

Modify fusion_emitter_large_test to work on ROCm.#593

Merged
zoranjovanovic-ns merged 1 commit into
rocm-jaxlib-v0.8.2from
rocm-jaxlib-v0.8.2-fusion_emitter_large_test
Jan 29, 2026
Merged

Modify fusion_emitter_large_test to work on ROCm.#593
zoranjovanovic-ns merged 1 commit into
rocm-jaxlib-v0.8.2from
rocm-jaxlib-v0.8.2-fusion_emitter_large_test

Conversation

@zoranjovanovic-ns
Copy link
Copy Markdown

Motivation

Failing fusion_emitter_large_test

Technical Details

Skipping only the tests that require more than 4gb of memory.
This PR is equivalent of: #568

Test Plan

Run fusion_emitter_large_test

Test Result

Test pass

Submission Checklist

@i-chaochen i-chaochen 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 Jan 28, 2026
Copy link
Copy Markdown
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.

I have added 3 labels (0.8.2, release candidate and upstream) on this PR. Please add related labels for other PRs as well.

@zoranjovanovic-ns zoranjovanovic-ns merged commit a35713e into rocm-jaxlib-v0.8.2 Jan 29, 2026
6 of 8 checks passed
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

cherry-pick-candidate Mark a PR to be cherry-picked into the next ROCm JAX. Remove IIF the latest upstream contain the PR. rocm-jaxlib-v0.8.2 Upstream

Projects

None yet

Development

Successfully merging this pull request may close these issues.

3 participants