jax
[ROCm] Cherry-pick: Override HIP_VISIBLE_DEVICES per ROCm xdist worker
#761
Merged

[ROCm] Cherry-pick: Override HIP_VISIBLE_DEVICES per ROCm xdist worker #761

magaonka-amd
magaonka-amd Override HIP_VISIBLE_DEVICES per ROCm xdist worker
9fe1cd43
psanal35
psanal35 approved these changes on 2026-04-29
magaonka-amd magaonka-amd merged c0ddc7f3 into rocm-jaxlib-v0.9.2 134 days ago

Login to write a write a comment.

Login via GitHub

Reviewers
Assignees
No one assigned
Labels
Milestone