Fix a typo

pytorch · Apr 21, 2023 · 674e53c · 674e53c
1 parent 68824ea
commit 674e53c
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/torch_xla/core/xla_model.py b/torch_xla/core/xla_model.py
@@ -97,7 +97,7 @@ def xrt_world_size(defval=1):
   if pjrt.using_pjrt():
     g_xrt_world_size = pjrt.world_size()
   else:
-    g_xrt_world_size = xu.getenv_as(xenv.XRT_WORLD_SIZE, int, defval=defval)
+    g_xrt_world_size = xu.getenv_as(xenv.WORLD_SIZE, int, defval=defval)
   return g_xrt_world_size
 
 g_ordinal = None