Skip to content

Conversation

ricardoV94
Copy link
Member

  • Recompile Elemwise when openmp_elemwise_minsize changes (otherwise it won't have an effect)
  • Remove warnings and errors from ScalarLoop (OpenMP fails if any are raised)

@ricardoV94 ricardoV94 force-pushed the elemwise_minsize branch 2 times, most recently from 5980165 to c5e4e96 Compare May 17, 2023 21:01
@ricardoV94 ricardoV94 force-pushed the elemwise_minsize branch 2 times, most recently from bfcdd1a to 455dc49 Compare May 18, 2023 11:29
This was problematic when OpenMP was used in the Elemwise outer loop

We add one extra output flag stating whether iteration converged or not. This however breaks Hyp2F1 grad in python mode because it goes beyond the Elemwise limit on number of operands. To fix it we split the grad when on python mode
@ricardoV94 ricardoV94 requested a review from twiecki May 18, 2023 12:45
@ricardoV94 ricardoV94 marked this pull request as ready for review May 18, 2023 13:11
Copy link
Member

@twiecki twiecki left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Tbh I only understand 10% of the changes, but I trust it's good stuff ;).

@ricardoV94
Copy link
Member Author

Tbh I only understand 10% of the changes, but I trust it's good stuff ;).

Only time will tell :D

@ricardoV94 ricardoV94 merged commit a8cc03c into pymc-devs:main May 19, 2023
@ricardoV94 ricardoV94 deleted the elemwise_minsize branch May 23, 2023 08:34
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants