set workqueue as the numba threading library

rapidsai · May 30, 2022 · 4b356dd · 4b356dd
1 parent 52635b9
commit 4b356dd
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/ci/gpu/build.sh b/ci/gpu/build.sh
@@ -35,6 +35,9 @@ unset GIT_DESCRIBE_TAG
 # ucx-py version
 export UCX_PY_VERSION='0.26.*'
 
+# configure numba threading library
+export NUMBA_THREADING_LAYER=workqueue
+
 ################################################################################
 # SETUP - Check environment
 ################################################################################