Skip to content

Commit b6caad8

Browse files
authored
Changing set method defaults for arm vs intel Mac
Changing default multiprocessing start method to "fork" for ARM Macs and leaving default "forkserver" for intel macs.
1 parent 67bb45d commit b6caad8

File tree

1 file changed

+5
-1
lines changed

1 file changed

+5
-1
lines changed

pymc/parallel_sampling.py

Lines changed: 5 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -408,7 +408,11 @@ def __init__(
408408
# Closes issue https://github.com/pymc-devs/pymc/issues/3849
409409
# Related issue https://github.com/pymc-devs/pymc/issues/5339
410410
if platform.system() == "Darwin":
411-
mp_ctx = "fork"
411+
if platform.processor() == 'arm':
412+
mp_ctx = 'fork'
413+
else:
414+
mp_ctx = "forkserver"
415+
412416
mp_ctx = multiprocessing.get_context(mp_ctx)
413417

414418
step_method_pickled = None

0 commit comments

Comments
 (0)