Browse Source

Add missing prefetch optimization for MPI M/S

Samuel Thibault 4 years ago
parent
commit
08ed2e6e6d
1 changed files with 2 additions and 0 deletions
  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);