Skip to content

add THREADING_LAYER_PRIORITY & NUMBA_THREADING_LAYER_PRIORITY #7234

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

Merged
sklam merged 21 commits into from
Oct 5, 2021
Merged

add THREADING_LAYER_PRIORITY & NUMBA_THREADING_LAYER_PRIORITY #7234

Changes from 1 commit
Commits
Show all changes
21 commits
Select commit Hold shift + click to select a range
86a896e
add THREADING_LAYER_PRIORITY & NUMBA_THREADING_LAYER_PRIORITY
ickc Jul 22, 2021
b4d3909
add TestThreadingLayerPriority
ickc Jul 22, 2021
2bb235a
fix test_invalid_env_var
ickc Jul 22, 2021
c030b83
fix flake8
ickc Jul 22, 2021
b25abe8
add skip_parfors_unsupported
ickc Jul 22, 2021
328f4db
Update docs/source/reference/envvars.rst
ickc Aug 26, 2021
8906558
Update docs/source/user/threading-layer.rst
ickc Aug 26, 2021
0377d0d
check len(namedbackends)
ickc Aug 26, 2021
d4b0a71
fix ref
ickc Aug 26, 2021
76236e0
shorten line
ickc Aug 26, 2021
ccb4888
add test_omp, test_tbb, test_workqueue
ickc Aug 26, 2021
814398d
move import to the top
ickc Aug 26, 2021
92abe9d
use assertRaises as suggested
ickc Aug 26, 2021
7536815
Update numba/np/ufunc/parallel.py
ickc Aug 26, 2021
b11cf88
Update docs/source/user/threading-layer.rst
ickc Aug 26, 2021
9199d93
Add wordings similar to that in "Setting the threading layer"
ickc Aug 26, 2021
c19b36a
fix flake8
ickc Aug 26, 2021
7bf4c3e
fix bug that stderr might include more than we print explicitly
ickc Aug 27, 2021
a92b40e
fix flake8
ickc Aug 27, 2021
7afb27f
Update numba/np/ufunc/parallel.py
ickc Sep 10, 2021
cbad639
Update docs/source/reference/envvars.rst
ickc Sep 10, 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
Prev Previous commit
Next Next commit
Add wordings similar to that in "Setting the threading layer" 
to mention setting numba.config.THREADING_LAYER_PRIORITY
  • Loading branch information
@ickc
ickc committed Aug 26, 2021
commit 9199d93a8d8ac01f9cc0ba4feeac8758fe6b5fd6
8 changes: 8 additions & 0 deletions docs/source/user/threading-layer.rst
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -67,9 +67,17 @@ By default the threading layers are searched in the order of ``'tbb'``,
maintaining the selection of a threading layer based on availability, the
environment variable :envvar:`NUMBA_THREADING_LAYER_PRIORITY` can be used.

Note that it can also be set via
:py:data:`numba.config.THREADING_LAYER_PRIORITY`.
Similar to :py:data:`numba.config.THREADING_LAYER`,
it must occur logically before any Numba based
compilation for a parallel target has occurred.

For example, to instruct Numba to choose ``omp`` first if available,
then ``tbb`` and so on, set the environment variable as
``NUMBA_THREADING_LAYER_PRIORITY="omp tbb workqueue"``.
Or programmatically,
``numba.config.THREADING_LAYER_PRIORITY = ["omp", "tbb", "workqueue"]``.

Selecting a threading layer for safe parallel execution
~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
Expand Down