rearranging to fit the original function order; convenience version of execParallel()

python/test/test_wgridder_experiments.py

@@ -56,13 +56,13 @@ def explicit_gridder(uvw, freq, ms, wgt, nxdirty, nydirty, xpixsize, ypixsize,
 
 @pmp('nxdirty', [16, 64])
 @pmp('nydirty', [64])
-@pmp("nrow", (1, 100, 10000))
-@pmp("nchan", (1,))
+@pmp("nrow", (1, 100))
+@pmp("nchan", (1, 7))
 @pmp("epsilon", list(10.**np.linspace(-2.,-12.,100)))
 @pmp("singleprec", (False,))
 @pmp("wstacking", (True,))
 @pmp("use_wgt", (True,))
-@pmp("nthreads", (1,))
+@pmp("nthreads", (1, 10))
 @pmp("fov", (10.,))
 def test_ms2dirty_against_wdft2(nxdirty, nydirty, nrow, nchan, epsilon, singleprec, wstacking, use_wgt, fov, nthreads):
     if singleprec and epsilon < 5e-5:

src/ducc0/infra/threading.h

@@ -23,6 +23,7 @@
 #define DUCC0_THREADING_H
 
 #include <functional>
+#include "ducc0/infra/misc_utils.h"
 
 namespace ducc0 {
 
@@ -61,6 +62,28 @@ void execGuided(size_t nwork, size_t nthreads, size_t chunksize_min,
   double fact_max, std::function<void(Scheduler &)> func);
 void execParallel(size_t nthreads, std::function<void(Scheduler &)> func);
 
+inline void execParallel(size_t nwork, size_t nthreads,
+  std::function<void(size_t, size_t)> func)
+  {
+  execParallel(nthreads, [&](Scheduler &sched)
+    {
+    auto tid = sched.thread_num();
+    auto [lo, hi] = calcShare(nthreads, tid, nwork);
+    func(lo, hi);
+    });
+  }
+
+inline void execParallel(size_t nwork, size_t nthreads,
+  std::function<void(size_t, size_t, size_t)> func)
+  {
+  execParallel(nthreads, [&](Scheduler &sched)
+    {
+    auto tid = sched.thread_num();
+    auto [lo, hi] = calcShare(nthreads, tid, nwork);
+    func(tid, lo, hi);
+    });
+  }
+
 } // end of namespace detail_threading
 
 using detail_threading::max_threads;