Add missing NeuronPlugin configure_single_process function (pytorch#8694

)
rpsilva-aws · Feb 11, 2025 · 065cb5b · 065cb5b
1 parent cff9f4e
commit 065cb5b
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/torch_xla/_internal/neuron.py b/torch_xla/_internal/neuron.py
@@ -116,5 +116,8 @@ def library_path(self):
   def configure_multiprocess(self, local_rank, local_world_size):
     initialize_env(local_rank, local_world_size)
 
+  def configure_single_process(self):
+    initialize_env(0, 1)
+
   def physical_chip_count(self):
     return num_local_processes()