Skip to content

Fix code caching for OpenMP regions (CPU and GPU) #29

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
merged 1 commit into from
Nov 6, 2024

Conversation

DrTodd13
Copy link
Contributor

  1. Fix dispatcher pickling to only return output_types if it is a object mode lifted with.
  2. Move openmp library opening to the first openmp compilation or the first time a cached openmp function is loaded.
  3. Make openmp_tag pickleable by not storing the arg if it contains a llvmlite AllocaInstr.

@ggeorgakoudis
Copy link
Contributor

ggeorgakoudis commented Sep 8, 2024

Pause on this branch @DrTodd13. I'm restructuring numba to remove distributing llvm-tools and libraries and use instead llvm-config, which affects init.

@ggeorgakoudis
Copy link
Contributor

@DrTodd13 can we revive this PR on the latest develop version?

@DrTodd13 DrTodd13 force-pushed the lazy_lib_init branch 2 times, most recently from 6f02329 to 610f45a Compare October 15, 2024 18:44
@ggeorgakoudis ggeorgakoudis changed the title Lazy initialize libraries. Fix code caching for OpenMP regions (CPU and GPU) Oct 29, 2024
…ect mode lifted with.

2) Make openmp_tag pickleable by not storing the arg if it contains a llvmlite AllocaInstr.
3) Don't pickle anything for LiftedCode's function IR except the arg count and the func_id.
4) Return all of openmp_region_start for pickling.
5) Comment on why we need new class.
@ggeorgakoudis
Copy link
Contributor

@DrTodd13 Rebased on a more recent develop that fixes an issue with gitlab tests on the ruby machine

@ggeorgakoudis ggeorgakoudis self-requested a review November 6, 2024 01:51
@ggeorgakoudis ggeorgakoudis merged commit 296fb85 into develop Nov 6, 2024
5 checks passed
@ggeorgakoudis ggeorgakoudis deleted the lazy_lib_init branch November 6, 2024 01:52
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