update debug_single_process to false (#7695)

pytorch · Jul 17, 2024 · 44f88a9 · 44f88a9
1 parent a6ab3fe
commit 44f88a9
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 1 deletion.
diff --git a/test/test_train_mp_imagenet.py b/test/test_train_mp_imagenet.py
@@ -375,4 +375,5 @@ def _mp_fn(index, flags):
 if __name__ == '__main__':
   # if running with torchrun, nprocs argument will be omitted.
   debug_single_process = True if FLAGS.num_cores == 1 else False
-  torch_xla.launch(_mp_fn, args=(FLAGS,), debug_single_process=True)
+  torch_xla.launch(
+      _mp_fn, args=(FLAGS,), debug_single_process=debug_single_process)
diff --git a/torch_xla/torch_xla.py b/torch_xla/torch_xla.py
@@ -87,6 +87,7 @@ def manual_seed(seed, device=None):
   xm.set_rng_state(seed, device)
 
 
+# TODO(wcromar): Update args to type ParamSpec.
 def launch(
     fn: Callable,
     args: Tuple = (),