浏览代码

datawizard: Only call progression hook once per datawizard call

Samuel Thibault 4 年之前
父节点
当前提交
d730232078
共有 1 个文件被更改,包括 3 次插入1 次删除
  1. 3 1
      src/datawizard/datawizard.c

+ 3 - 1
src/datawizard/datawizard.c

@@ -60,7 +60,6 @@ static int ___starpu_datawizard_progress(unsigned memory_node, unsigned peer_nod
 		if (pushed)
 			ret = 1;
 	}
-	_starpu_execute_registered_progression_hooks();
 
 	return ret;
 }
@@ -82,6 +81,7 @@ int __starpu_datawizard_progress(unsigned may_alloc, unsigned push_requests)
 				ret |=  ___starpu_datawizard_progress(numa, numa2, _STARPU_DATA_REQUEST_IN, may_alloc, push_requests);
 				ret |=  ___starpu_datawizard_progress(numa, numa2, _STARPU_DATA_REQUEST_OUT, may_alloc, push_requests);
 			}
+		_starpu_execute_registered_progression_hooks();
 
 		return ret;
 	}
@@ -104,6 +104,8 @@ int __starpu_datawizard_progress(unsigned may_alloc, unsigned push_requests)
 			}
         }
 
+	_starpu_execute_registered_progression_hooks();
+
         return ret;
 }