Skip to content

Pull requests: ROCm/xla

New pull request New
101 Open 644 Closed
101 Open 644 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] Remove deprecated hipCtx_t and HIP context APIs
#746 opened Mar 24, 2026 by phambinhfin Loading…
4
[TEST] llama31_8b_bf16_1x8_no_te for benchmark
#739 opened Mar 24, 2026 by i-chaochen Loading…
1 task
2
Switching to amd_comgr_static for ROCm 7.12 and newer.
#736 opened Mar 23, 2026 by mminutoli • Draft
1 task
2
[ROCm] Porting CUB sort FFI handler consolidation to ROCm.
#726 opened Mar 20, 2026 by tsrw2048 Loading…
17
Code review comments.
#698 opened Mar 19, 2026 by zoranjovanovic-ns Loading…
1 task
Enable ROCm Triton backend for AllReduce
#684 opened Mar 18, 2026 by mfrancepillois Loading…
84
Phambinh/implement emptynode
#680 opened Mar 18, 2026 by phambinhfin Loading…
5
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…
Disable auto_sharding test on ROCm. rocm-jaxlib-v0.8.2
#617 opened Feb 4, 2026 by zoranjovanovic-ns Loading…
1 task
1
[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…
ProTip! Adding no:label will show everything without a label.