[ROCm] Cherry-pick: Add ROCm xdist device pinning in conftest.py #744
magaonka-amd
merged
1 commit into
ROCm:rocm-jaxlib-v0.9.2
from
magaonka-amd:cherry-pick-rocm-xdist-conftest
Disable GPU memory preallocation in ROCm pytest
f8ac3eaf
gulsumgudukbay
approved these changes
on 2026-04-02
Login to write a write a comment.
Login via GitHub