Skip to content

Fall back to first visible GPU when local rank exceeds device count - #3843

Open
rishabhsai wants to merge 1 commit into
pytorch:masterfrom
rishabhsai:fix-slurm-ntasks-per-gpu-local-rank
Open

rishabhsai wants to merge 1 commit into
pytorch:masterfrom
rishabhsai:fix-slurm-ntasks-per-gpu-local-rank