瀏覽代碼

port r11283 from 1.1: Prefetch data for tasks with execute_on_a_specific_worker too

Samuel Thibault 13 年之前
父節點
當前提交
4cae2beddb
共有 1 個文件被更改,包括 4 次插入0 次删除
  1. 4 0
      src/core/sched_policy.c

+ 4 - 0
src/core/sched_policy.c

@@ -413,6 +413,10 @@ int _starpu_push_task_to_workers(struct starpu_task *task)
 	int ret;
 	if (STARPU_UNLIKELY(task->execute_on_a_specific_worker))
 	{
+		unsigned node = starpu_worker_get_memory_node(task->workerid);
+		if (starpu_get_prefetch_flag())
+			starpu_prefetch_task_input_on_node(task, node);
+
 		ret = _starpu_push_task_on_specific_worker(task, task->workerid);
 	}
 	else