Fix code caching for OpenMP regions (CPU and GPU)

Python-for-HPC / numbaWithOpenmp

BSD 2-Clause "Simplified" License

3 stars 4 forks source link

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

Open DrTodd13 opened 3 months ago

DrTodd13 commented 3 months ago

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 commented 1 month ago

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 commented 3 weeks ago

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