浏览代码

Add missing prefetch optimization for MPI M/S

Samuel Thibault 4 年之前
父节点
当前提交
08ed2e6e6d
共有 1 个文件被更改,包括 2 次插入0 次删除
  1. 2 0
      src/core/sched_policy.c

+ 2 - 0
src/core/sched_policy.c

@@ -637,6 +637,8 @@ int _starpu_push_task_to_workers(struct starpu_task *task)
 					starpu_prefetch_task_input_on_node(task, config->opencl_nodeid);
 				else if (task->where == STARPU_MIC && config->mic_nodeid >= 0)
 					starpu_prefetch_task_input_on_node(task, config->mic_nodeid);
+				else if (task->where == STARPU_MPI_MS && config->mpi_nodeid >= 0)
+					starpu_prefetch_task_input_on_node(task, config->mpi_nodeid);
 			}
 
 			STARPU_ASSERT(sched_ctx->sched_policy->push_task);