Skip to content

Pull requests: ROCm/xla

New pull request New
104 Open 604 Closed
104 Open 604 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

Code review comments.
#698 opened Mar 19, 2026 by zoranjovanovic-ns Loading…
1 task
[ROCm] Use BFCAllocator for scratch allocations needed for MIOpen autotuning open-upstream Tag when you want a copy of this PR to be opened on upstream rocm-jaxlib-v0.8.2 rocm-jaxlib-v0.9.0
#695 opened Mar 19, 2026 by draganmladjenovic Loading…
1 task
13
[rocm-jaxlib-v0.9.1] Enable postsubmit benchmarks in CI
#691 opened Mar 19, 2026 by mmakevic-amd Loading…
5
Enable ROCm Triton backend for AllReduce
#684 opened Mar 18, 2026 by mfrancepillois • Draft
74
Phambinh/implement emptynode
#680 opened Mar 18, 2026 by phambinhfin Loading…
5
Add ci trigger
#678 opened Mar 18, 2026 by alekstheod Loading…
73
Test
#674 opened Mar 17, 2026 by alekstheod Loading…
2
Backport from upstream to 0.9.1 branch
#672 opened Mar 17, 2026 by hsharsha Loading…
1 task
36
Add error checking and status propagation to ROCm profiler
#671 opened Mar 17, 2026 by magaonka-amd • Draft
1 task
25
Add scope_range_id support to ROCm profiler
#670 opened Mar 17, 2026 by magaonka-amd • Draft
1 task
32
Compute theoretical occupancy for ROCm profiler kernels
#669 opened Mar 16, 2026 by magaonka-amd • Draft
1 task
7
Add hipCUB support for scan (prefix sum) op
#668 opened Mar 16, 2026 by magaonka-amd Loading…
1 task
7
Enable autotuner rocblas/hipblaslt fission backends
#663 opened Mar 10, 2026 by Eetusjo Loading…
14
Fix issue with missing upcast/downcast for bf16 libdevice calls. 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
#661 opened Mar 9, 2026 by zoranjovanovic-ns Loading…
1 task
3
Fix issue with missing upcast/downcast for bf16 libdevice calls. 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
#660 opened Mar 9, 2026 by zoranjovanovic-ns Loading…
14
Recover from crashes in Triton pass pipeline
#659 opened Mar 9, 2026 by nurmukhametov Loading…
1 task done
4
Fix AnnotationMap memory leak in ROCm profiler across profiling sessions open-upstream Tag when you want a copy of this PR to be opened on upstream rocm-jaxlib-v0.9.0
#656 opened Mar 4, 2026 by magaonka-amd Loading…
1 task
10
Dev rocm scaled dot
#655 opened Mar 4, 2026 by ScXfjiang Loading…
54
Pemeliya/v0.9.0 split grid dim x
#653 opened Mar 4, 2026 by pemeliya Loading…
62
Rocm jaxlib v0.9.0 dev scaled dot
#644 opened Feb 27, 2026 by ScXfjiang Loading…
64
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
ProTip! Find all pull requests that aren't related to any open issues with -linked:issue.