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.

Sign up for GitHub

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

[MRG] FEA Make it possible to pass pre-inspected modules to threadpool_limit #95

Merged
merged 20 commits into from
Sep 24, 2021
Merged

[MRG] FEA Make it possible to pass pre-inspected modules to threadpool_limit #95

Show file tree
Hide file tree
Changes from 5 commits
Commits
Show all changes
20 commits
Select commit Hold shift + click to select a range
4b9bfd2
wip
jeremiedbb Sep 16, 2021
7c213f1
cln
jeremiedbb Sep 16, 2021
4ba0ccd
fix select
jeremiedbb Sep 16, 2021
7a7a9fd
test for select
jeremiedbb Sep 16, 2021
1f30714
add threadpool_limits + controller test
jeremiedbb Sep 16, 2021
83bbb18
wip
jeremiedbb Sep 17, 2021
20c4f65
address some comments + doc select
jeremiedbb Sep 17, 2021
e45c5bb
keep get set public
jeremiedbb Sep 17, 2021
8428a80
black
jeremiedbb Sep 17, 2021
c0c86e6
add a method to return the context manager
jeremiedbb Sep 17, 2021
a221220
cln
jeremiedbb Sep 17, 2021
ebb0a03
Merge branch 'master' into make-modules-reusable
jeremiedbb Sep 23, 2021
826eda8
remove list-like api
jeremiedbb Sep 23, 2021
6621a88
readme
jeremiedbb Sep 23, 2021
f78dcaa
change log
jeremiedbb Sep 23, 2021
e6ad48d
address comment
jeremiedbb Sep 23, 2021
3745899
black
jeremiedbb Sep 24, 2021
238d198
Merge branch 'master' into make-modules-reusable
jeremiedbb Sep 24, 2021
9002c72
cln
jeremiedbb Sep 24, 2021
de1078d
show equivalence in snippet
jeremiedbb Sep 24, 2021
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
8 changes: 4 additions & 4 deletions tests/_openmp_test_helper/nested_prange_blas.pyx
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -20,7 +20,7 @@ IF USE_BLIS:
ELSE:
from scipy.linalg.cython_blas cimport dgemm

from threadpoolctl import _ThreadpoolInfo, _ALL_USER_APIS
from threadpoolctl import ThreadpoolController


def check_nested_prange_blas(double[:, ::1] A, double[:, ::1] B, int nthreads):
Expand All @@ -43,12 +43,12 @@ def check_nested_prange_blas(double[:, ::1] A, double[:, ::1] B, int nthreads):
int prange_num_threads
int *prange_num_threads_ptr = &prange_num_threads

inner_info = [None]
inner_ctl = [None]

with nogil, parallel(num_threads=nthreads):
if openmp.omp_get_thread_num() == 0:
with gil:
inner_info[0] = _ThreadpoolInfo(user_api=_ALL_USER_APIS)
inner_ctl[0] = ThreadpoolController()

prange_num_threads_ptr[0] = openmp.omp_get_num_threads()

Expand All @@ -62,4 +62,4 @@ def check_nested_prange_blas(double[:, ::1] A, double[:, ::1] B, int nthreads):
&alpha, &B[0, 0], &k, &A[i * chunk_size, 0], &k,
&beta, &C[i * chunk_size, 0], &n)

return np.asarray(C), prange_num_threads, inner_info[0]
return np.asarray(C), prange_num_threads, inner_ctl[0]
Loading