jax
20f77158 - [ROCm] Set HSA_NO_SCRATCH_RECLAIM=1 in ROCm test workflows

Commit
52 days ago
[ROCm] Set HSA_NO_SCRATCH_RECLAIM=1 in ROCm test workflows Required on gfx950 with recent ROCm to avoid HIP "invalid device function" failures during multi-accelerator collectives; the env var must be set before HIP initialization (JAX import time). (cherry picked from commit 756fa6a25251eed0a64ba6e1e4a0f212b8abf4b1)
Committer
Parents
Loading