Skip to content
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

Wrappers for MKL threading using Domains #119

Merged
merged 1 commit into from
Jul 31, 2024
Merged

Conversation

ViralBShah
Copy link
Collaborator

@ViralBShah ViralBShah commented Aug 7, 2023

Attempt to fix #74 - but will not work because the symbol_lookup matches the function name to the dlopen'ed library.

Opening this PR here to document the challenge with implementing this.

@ViralBShah ViralBShah marked this pull request as draft August 7, 2023 00:55
@staticfloat staticfloat marked this pull request as ready for review July 31, 2024 17:39
@staticfloat
Copy link
Member

I reworked the implementation here to special-case the MKL thread getters/setters in a different way. Let's see if CI is happy with this.

This uses the `MKL_Domain_*` functions to get/set the thread count for
the BLAS domain only, so FFT, PARDISO, etc... domains in MKL are not
affected bt `lbt_set_num_threads()`.  It also adds a test to show that
this behavior is reasonable when MKL is loaded.
@staticfloat staticfloat merged commit ead47b5 into main Jul 31, 2024
5 checks passed
@staticfloat staticfloat deleted the vs/mkl_domains branch July 31, 2024 18:10
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.

[MKL] Get/Set number of threads - should get/set for BLAS, not for the whole MKL
2 participants