Skip to content

Pull requests: ROCm/xla

New pull request New
84 Open 503 Closed
84 Open 503 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

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…
1
Modified triton_fusion_numerics_verifier_test to work on ROCm.
#587 opened Jan 27, 2026 by zoranjovanovic-ns Loading…
1 task
4
Avoid upcast of lib func operands to F32 for F16 type.
#586 opened Jan 27, 2026 by zoranjovanovic-ns Loading…
1 task
1
Fixed dot_algorithms_test. Updated support_legacy and test itself. open-upstream Tag when you want a copy of this PR to be opened on upstream rocm-jaxlib-v0.8.2
#582 opened Jan 27, 2026 by zoranjovanovic-ns Loading…
1 task
2
Fix expected output in fusion_emitter_int4_device_test for ROCm. open-upstream Tag when you want a copy of this PR to be opened on upstream rocm-jaxlib-v0.8.2
#581 opened Jan 26, 2026 by zoranjovanovic-ns Loading…
1 task
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…
Fixed missing rtne in Triton to pass support_test. 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
#578 opened Jan 26, 2026 by zoranjovanovic-ns Loading…
1 task
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…
Port v0.8.2 to v0.9.0 part 1 rocm-jaxlib-v0.9.0
#570 opened Jan 23, 2026 by nurmukhametov Loading…
1 task done
5
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
78
Enable FP8 Triton GEMM support for ROCm GPUs Upstream
#511 opened Jan 7, 2026 by nurmukhametov Loading…
4
Rocm jaxlib v0.8.0 conv fixes rocm-jaxlib-v0.8.0 rocm-jaxlib-v0.8.2
#510 opened Dec 28, 2025 by draganmladjenovic Loading…
3
Migrate parametrized hermetic rocm rocm-jaxlib-v0.8.0 rocm-jaxlib-v0.8.2
#505 opened Dec 18, 2025 by alekstheod Loading…
1
Migrate multigpu, singlegpu bazelrc settings
#504 opened Dec 18, 2025 by alekstheod Loading…
1
Enable matmul perf table gen tooling for 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
#500 opened Dec 18, 2025 by Eetusjo Loading…
1
ProTip! Exclude everything labeled bug with -label:bug.