finufft: use OMP-determined num threads by default

flatironinstitute · Mar 27, 2024 · bee05cf · bee05cf
1 parent f6fddad
commit bee05cf
Showing 1 changed file with 1 addition and 5 deletions.
diff --git a/src/nifty_ls/finufft.py b/src/nifty_ls/finufft.py
@@ -1,13 +1,12 @@
 from __future__ import annotations
 
-import os
 
 import finufft
 import numpy as np
 
 from . import cpu_helpers
 
-DEFAULT_NTHREAD = len(os.sched_getaffinity(0))
+DEFAULT_NTHREAD = 0  # let finufft detect according to OMP
 
 FFTW_MEASURE = 0
 FFTW_ESTIMATE = 64
@@ -41,9 +40,6 @@ def lombscargle(
         # TODO: better thread heuristics?
         nthreads = 1 if Nf <= 10**5 else DEFAULT_NTHREAD
 
-    if nthreads < 1:
-        nthreads = DEFAULT_NTHREAD
-
     dtype = t.dtype
 
     if finufft_kwargs['eps'] == 'default':