We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 5688555 commit 636b957Copy full SHA for 636b957
pymc/sampling/parallel.py
@@ -386,9 +386,13 @@ def __init__(
386
if mp_ctx is None or isinstance(mp_ctx, str):
387
# Closes issue https://github.com/pymc-devs/pymc/issues/3849
388
# Related issue https://github.com/pymc-devs/pymc/issues/5339
389
- if platform.system() == "Darwin":
+ if mp_ctx is None and platform.system() == "Darwin":
390
if platform.processor() == "arm":
391
mp_ctx = "fork"
392
+ logger.debug(
393
+ "mp_ctx is set to 'fork' for MacOS with ARM architecture. "
394
+ + "This might cause unexpected behavior with JAX, which is inherently multithreaded."
395
+ )
396
else:
397
mp_ctx = "forkserver"
398
0 commit comments