loadBalance on loop tasks entry to address #76 (comment)

mratsim · Dec 29, 2019 · f1d734a · f1d734a
1 parent 902e996
commit f1d734a
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 0 deletions.
diff --git a/weave/parallel_for.nim b/weave/parallel_for.nim
@@ -32,6 +32,7 @@ template parallelForWrapper(
   ## Loop prologue, epilogue,
   ## remoteAccum, resultTy and returnStmt
   ## are unused
+  loadBalance(Weave)
 
   block:
     let this = myTask()

diff --git a/weave/parallel_for_staged.nim b/weave/parallel_for_staged.nim
@@ -29,6 +29,7 @@ template parallelStagedWrapper(
   ## Also poll steal requests in-between iterations
   ##
   ## remoteAccum and resultFlowvarType are unused
+  loadBalance(Weave)
 
   prologue
 

diff --git a/weave/parallel_reduce.nim b/weave/parallel_reduce.nim
@@ -28,6 +28,7 @@ template parallelReduceWrapper(
   ## To be called within a loop task
   ## Gets the loop bounds and iterate the over them
   ## Also poll steal requests in-between iterations
+  loadBalance(Weave)
 
   prologue