Simpler wavelet implementation with vertical parallelism (#93)

* replace old wavelet code with simplified 2D versions * incorporate new wavelets and psi operator --------- Co-authored-by: landmanbester <lbester@ska.ac.za>
ratt-ru · Mar 15, 2024 · a545cfd · a545cfd
1 parent 891d30c
commit a545cfd
Show file tree

Hide file tree

Showing 15 changed files with 812 additions and 3,277 deletions.
diff --git a/pfb/__init__.py b/pfb/__init__.py
@@ -31,11 +31,11 @@ def set_client(opts, stack, log, scheduler='distributed'):
     os.environ["OPENBLAS_NUM_THREADS"] = str(opts.nvthreads)
     os.environ["MKL_NUM_THREADS"] = str(opts.nvthreads)
     os.environ["VECLIB_MAXIMUM_THREADS"] = str(opts.nvthreads)
+    os.environ["NUMBA_NUM_THREADS"] = str(opts.nvthreads)
     import numexpr as ne
     max_cores = ne.detect_number_of_cores()
     # ne_threads = min(max_cores, opts.nvthreads)
     os.environ["NUMEXPR_NUM_THREADS"] = str(max_cores)
-    os.environ["NUMBA_NUM_THREADS"] = str(max_cores)
 
     import dask
     if scheduler=='distributed':