Add one more get_context() (google#89)

Northbadge · web-flow · commit 4566f1b815eb · 2022-08-08T12:37:37.000-07:00
diff --git a/compiler_opt/distributed/local/local_worker_manager.py b/compiler_opt/distributed/local/local_worker_manager.py
@@ -226,7 +226,7 @@ class LocalWorkerPool(AbstractContextManager):
   def __init__(self, worker_class: 'type[Worker]', count: Optional[int], *args,
                **kwargs):
     if not count:
-      count = multiprocessing.cpu_count()
+      count = multiprocessing.get_context().cpu_count()
     self._stubs = [
         _make_stub(worker_class, *args, **kwargs) for _ in range(count)
     ]

Original file line number	Diff line number	Diff line change
`@@ -226,7 +226,7 @@ class LocalWorkerPool(AbstractContextManager):`
`226`	`226`	`def __init__(self, worker_class: 'type[Worker]', count: Optional[int], *args,`
`227`	`227`	`**kwargs):`
`228`	`228`	`if not count:`
`229`		`- count = multiprocessing.cpu_count()`
	`229`	`+ count = multiprocessing.get_context().cpu_count()`
`230`	`230`	`self._stubs = [`
`231`	`231`	`_make_stub(worker_class, args, *kwargs) for _ in range(count)`
`232`	`232`	`]`