Browse Source

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

Samuel Thibault 12 years ago
parent
commit
4cae2beddb
1 changed files with 4 additions and 0 deletions
  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