Skip to content

Conversation

digicosmos86
Copy link
Contributor

@digicosmos86 digicosmos86 commented Dec 2, 2022

What is this PR about?
Per #6362, when the user provides a str to mp_ctx on ARM-based Macs, this string will be overridden without warnings to "fork", which might be incompatible with some JAX operations in parallel settings. This change was introduced in #6218.

Closes #6362

Bugfixes / New features

  • Changed one if-statement so user provided strs are not overridden.
  • Added a debugging message about the incompatibility of "fork" and JAX.

@codecov
Copy link

codecov bot commented Dec 2, 2022

Codecov Report

Merging #6363 (2dde4c6) into main (5688555) will decrease coverage by 0.00%.
The diff coverage is 50.00%.

Additional details and impacted files

Impacted file tree graph

@@            Coverage Diff             @@
##             main    #6363      +/-   ##
==========================================
- Coverage   94.72%   94.72%   -0.01%     
==========================================
  Files         132      132              
  Lines       26740    26741       +1     
==========================================
  Hits        25330    25330              
- Misses       1410     1411       +1     
Impacted Files Coverage Δ
pymc/sampling/parallel.py 87.45% <50.00%> (-0.31%) ⬇️

@michaelosthege michaelosthege merged commit 636b957 into pymc-devs:main Dec 4, 2022
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
bug jax macOS macOS related
Projects
None yet
Development

Successfully merging this pull request may close these issues.

New mp_ctx defaults in 4.3.0+ (#6218) for M1 Macs could break parallel sampling for M1 Mac users using JAX
3 participants