Skip to content

Commit 065cb5b

Browse files
authored
Add missing NeuronPlugin configure_single_process function (pytorch#8694)
1 parent cff9f4e commit 065cb5b

File tree

1 file changed

+3
-0
lines changed

1 file changed

+3
-0
lines changed

torch_xla/_internal/neuron.py

+3
Original file line numberDiff line numberDiff line change
@@ -116,5 +116,8 @@ def library_path(self):
116116
def configure_multiprocess(self, local_rank, local_world_size):
117117
initialize_env(local_rank, local_world_size)
118118

119+
def configure_single_process(self):
120+
initialize_env(0, 1)
121+
119122
def physical_chip_count(self):
120123
return num_local_processes()

0 commit comments

Comments
 (0)