From 817744762bd6b96aa5209b38b4f6f7ea822253f8 Mon Sep 17 00:00:00 2001
From: Michael Green <59619482+mikegre-google@users.noreply.github.com>
Date: Wed, 30 Oct 2024 16:46:29 +0000
Subject: [PATCH] Fixes issue #7774 (#8322)

---
 docs/source/perf/ddp.md | 3 +--
 1 file changed, 1 insertion(+), 2 deletions(-)

diff --git a/docs/source/perf/ddp.md b/docs/source/perf/ddp.md
index 826895d5ac8..a84946cc728 100644
--- a/docs/source/perf/ddp.md
+++ b/docs/source/perf/ddp.md
@@ -77,8 +77,7 @@ import torch_xla.runtime as xr
 import torch_xla.distributed.xla_backend
 
 def setup(rank, world_size):
-    os.environ['MASTER_ADDR'] = 'localhost'
-    os.environ['MASTER_PORT'] = '12355'
+    os.environ['PJRT_DEVICE'] = 'TPU'
 
     # initialize the xla process group
     dist.init_process_group("xla", rank=rank, world_size=world_size)