parallel loop split

Author: uecker

src/num/multind.c

@@ -216,6 +216,17 @@ static void md_loop_r(int D, const long dim[D], unsigned long flags, long pos[D]
  *
  */
 void md_parallel_loop(int D, const long dim[static D], unsigned long flags, md_loop_fun_t fun)
+{
+	NESTED(void, fun2, (unsigned long flags2, long *pos))
+	{
+		md_loop_r(D, dim, flags2, pos, fun);
+	};
+
+	md_parallel_loop_split(D, dim, flags, fun2);
+}
+
+
+void md_parallel_loop_split(int D, const long dim[static D], unsigned long flags, md_loop_fun2_t fun)
 {
 	flags &= md_nontriv_dims(D, dim);
 
@@ -247,10 +258,11 @@ void md_parallel_loop(int D, const long dim[static D], unsigned long flags, md_l
 		// Recover place in parallel iteration space
 		long pos[D];
 		md_unravel_index(D, pos, ~0UL, pdims, i);
-		md_loop_r(D, dim, flags, pos, fun);
+		fun(flags, pos);
 	}
 }
 
+
 /**
  * Generic function which loops over all dimensions and calls a given
  * function passing the current indices as argument.

src/num/multind.h

@@ -27,6 +27,7 @@
 typedef void CLOSURE_TYPE(md_nary_fun_t)(void* ptr[]);
 typedef void CLOSURE_TYPE(md_trafo_fun_t)(long N, long str, void* ptr);
 typedef void CLOSURE_TYPE(md_loop_fun_t)(const long* pos);
+typedef void CLOSURE_TYPE(md_loop_fun2_t)(unsigned long flags, long* pos);
 
 extern void md_unravel_index(int D, long pos[__VLA(D)], unsigned long flags, const long dims[__VLA(D)], long index);
 extern long md_ravel_index(int D, const long pos[__VLA(D)], unsigned long flags, const long dims[__VLA(D)]);
@@ -35,6 +36,7 @@ extern void md_nary(int C, int D, const long dim[__VLA(D)], const long* str[__VL
 
 extern void md_parallel_nary(int C, int D, const long dim[__VLA(D)], unsigned long flags, const long* str[__VLA(C)], void* ptr[__VLA(C)], md_nary_fun_t fun);
 extern void md_parallel_loop(int D, const long dim[__VLA(D)], unsigned long flags, md_loop_fun_t fun);
+extern void md_parallel_loop_split(int D, const long dim[__VLA(D)], unsigned long flags, md_loop_fun2_t fun);
 
 extern void md_loop(int D, const long dim[__VLA(D)], md_loop_fun_t fun);