Some fine tuning on the number of threads

Blosc · Jan 14, 2025 · d656d13 · d656d13
1 parent e27acd4
commit d656d13
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/src/blosc2/__init__.py b/src/blosc2/__init__.py
@@ -12,6 +12,8 @@
 
 from enum import Enum
 
+import numexpr
+
 from .version import __version__
 
 __version__ = __version__
@@ -195,7 +197,9 @@ class Tuner(Enum):
 # Protection against too many threads
 nthreads = min(nthreads, 32)
 # Experiments say that, when using a large number of threads, it is better to not use them all
-nthreads -= nthreads // 8
+if nthreads > 16:
+    nthreads -= nthreads // 8
+numexpr.set_num_threads(nthreads)
 
 # This import must be before ndarray and schunk
 from .storage import (  # noqa: I001