jax
[ROCm] Cherry-pick: Fix inter-block write race in pallas test_non_range_while_loop
#764
Merged
Merged
[ROCm] Cherry-pick: Fix inter-block write race in pallas test_non_range_while_loop #764
magaonka-amd
merged
1 commit into
ROCm:rocm-jaxlib-v0.9.2
from
magaonka-amd:cherry-pick/pallas-non-range-while-loop-rocm-race-v092
[ROCm] Use grid=(1,) in pallas test_non_range_while_loop
98fdf93e
psanal35
approved these changes
on 2026-04-29
Login to write a write a comment.
Login via GitHub