Skip to content

Pull requests: ROCm/xla

New pull request New
87 Open 544 Closed
87 Open 544 Closed
Author
Filter by author
Loading
Label
Filter by label
Loading
Use alt + click/return to exclude labels
or ⇧ + click/return for logical OR
Projects
Filter by project
Loading
Milestones
Filter by milestone
Loading
Reviews
Filter by reviews
No reviews Review required Approved review Changes requested
Assignee
Filter by who’s assigned
Assigned to nobody Loading
Sort
Sort by
Newest Oldest Most commented Least commented Recently updated Least recently updated Best match
Most reactions
👍 👎 😄 🎉 😕 ❤️ 🚀 👀

Pull requests list

[ROCm] Add TSAN annotations to RocmStream::DoHostCallbackWithStatus rocm-jaxlib-v0.9.0
#631 opened Feb 19, 2026 by draganmladjenovic Loading…
2
Leo/mock matrix
#622 opened Feb 5, 2026 by leo-amd Loading…
Ignore tsan warning due to the rocm barrier
#618 opened Feb 4, 2026 by alekstheod Loading…
1
Disable auto_sharding test on ROCm. rocm-jaxlib-v0.8.2
#617 opened Feb 4, 2026 by zoranjovanovic-ns Loading…
1 task
1
Test tsan
#614 opened Feb 3, 2026 by alekstheod Loading…
1 task
[ROCm] Add thread safety test for hipblaslt GetAlgorithms help wanted Extra attention is needed rocm-jaxlib-v0.8.0
#609 opened Feb 1, 2026 by i-chaochen • Draft
test max num trace events 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.9.0 Upstream
#606 opened Jan 29, 2026 by cj401-amd Loading…
1
Introduce jax-ut-pipeline
#602 opened Jan 29, 2026 by alekstheod Loading…
2
Introduce jax u-test pipeline
#601 opened Jan 29, 2026 by alekstheod Loading…
[Test only do no merge] Test cdx mode
#600 opened Jan 29, 2026 by alekstheod Loading…
Sync mgpu tests with xla_mgpu config
#597 opened Jan 28, 2026 by alekstheod Loading…
1
Fix NCCL/RCCL rendezvous deadlock when mixing clique sizes bug Something isn't working 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.0 Upstream
#589 opened Jan 27, 2026 by phambinhfin Loading…
6
suppress cuda error messages (harmless) when doing training on ROCm cherry-pick-candidate Mark a PR to be cherry-picked into the next ROCm JAX. Remove IIF the latest upstream contain the PR. open-upstream Tag when you want a copy of this PR to be opened on upstream rocm-jaxlib-v0.8.0 Upstream
#580 opened Jan 26, 2026 by cj401-amd Loading…
7
suppress cuda error messages (harmless) when doing training on ROCm 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
#579 opened Jan 26, 2026 by cj401-amd Loading…
CI: Cherry-pick to 0.9.0 Internal CI on GHA 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.9.0
#576 opened Jan 26, 2026 by leo-amd Loading…
CI: Cherry-pick to 0.8.2 Internal CI on GHA rocm-jaxlib-v0.8.2
#575 opened Jan 26, 2026 by leo-amd Loading…
CI: Cherry-pick to 0.8.0 Internal CI on GHA rocm-jaxlib-v0.8.0
#574 opened Jan 26, 2026 by leo-amd Loading…
3
CI: Moved Intenal XLA CI to GHA. Added switching between runners based on the target repo cherry-pick-candidate Mark a PR to be cherry-picked into the next ROCm JAX. Remove IIF the latest upstream contain the PR.
#573 opened Jan 26, 2026 by leo-amd Loading…
Remove NFC changes from v0.8.2 rocm-jaxlib-v0.8.2
#566 opened Jan 23, 2026 by nurmukhametov Loading…
[ROCm] Fix reduction emitter tests to work on ROCm rocm-jaxlib-v0.8.2
#542 opened Jan 20, 2026 by nurmukhametov Loading…
1
Cleanup bazel disk cache
#525 opened Jan 16, 2026 by hsharsha Loading…
1 task
[DO NOT MERGE] CI testing
#524 opened Jan 16, 2026 by mmakevic-amd Loading…
[WIP] [ROCm][XLA:GPU] Prefer loop emitter for sibling concat fusions
#518 opened Jan 14, 2026 by nurmukhametov Loading…
Fix crash manifesting after pr226
#515 opened Jan 12, 2026 by mminutoli • Draft
1 task done
3
HipBlasLT-Ext GroupedGemm support for RaggedDot op
#514 opened Jan 12, 2026 by mfrancepillois • Draft
109
ProTip! Updated in the last three days: updated:>2026-02-21.