Explorar el Código

Add missing prefetch optimization for MPI M/S

Samuel Thibault hace 4 años
padre
commit
08ed2e6e6d
Se han modificado 1 ficheros con 2 adiciones y 0 borrados
  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);