Pass LOCAL_RANK
to torch_distributed_zero_first()
#5114
Merged
LOCAL_RANK
to torch_distributed_zero_first()
#5114