From bb1cb4db953e1d4f8ef850f07ba3d8ff7d0eb918 Mon Sep 17 00:00:00 2001 From: Defne Dilbaz Date: Thu, 20 Nov 2025 16:08:58 -0500 Subject: [PATCH] Fix typo in warning message for SPMD mode --- torch_xla/runtime.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/torch_xla/runtime.py b/torch_xla/runtime.py index 4e6352bc5527..baa76a4511f0 100644 --- a/torch_xla/runtime.py +++ b/torch_xla/runtime.py @@ -207,7 +207,7 @@ def use_spmd(auto: Optional[bool] = False): warnings.warn( "Replicating tensors already initialized on non-virtual XLA device for SPMD " "to force SPMD mode. This is one-time overhead to setup, and to minimize such, " - "please set SPMD mode before initializting tensors " + "please set SPMD mode before initializing tensors " "(i.e., call use_spmd() in the beginning of the program).") torch_xla._XLAC._xla_force_spmd_device() xm.wait_device_ops()