Просмотр исходного кода

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

Samuel Thibault лет назад: 12
Родитель
Сommit
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