int src_traced = smpi_group_index(smpi_comm_group(comm), src);
TRACE_smpi_ptp_in(rank, src_traced, dst_traced, __FUNCTION__);
TRACE_smpi_send(rank, rank, dst_traced);
- TRACE_smpi_send(rank, src_traced, rank);
#endif
#ifdef HAVE_TRACING
TRACE_smpi_ptp_out(rank, src_traced, dst_traced, __FUNCTION__);
- TRACE_smpi_recv(rank, rank, dst_traced);
TRACE_smpi_recv(rank, src_traced, rank);
TRACE_smpi_computing_in(rank);
#endif
: -1;
TRACE_smpi_computing_out(rank);
- MPI_Group group = smpi_comm_group((*request)->comm);
- int src_traced = smpi_group_index(group, (*request)->src);
- int dst_traced = smpi_group_index(group, (*request)->dst);
+ int src_traced = (*request)->src;
+ int dst_traced = (*request)->dst;
int is_wait_for_receive = (*request)->recv;
TRACE_smpi_ptp_in(rank, src_traced, dst_traced, __FUNCTION__);
#endif
|| recvtype == MPI_DATATYPE_NULL) {
retval = MPI_ERR_TYPE;
} else {
- smpi_mpi_gather(sendbuf, sendcount, sendtype, recvbuf, recvcount,
+ mpi_coll_gather_fun(sendbuf, sendcount, sendtype, recvbuf, recvcount,
recvtype, root, comm);
retval = MPI_SUCCESS;
}
} else if (recvcounts == NULL || displs == NULL) {
retval = MPI_ERR_ARG;
} else {
- smpi_mpi_allgatherv(sendbuf, sendcount, sendtype, recvbuf, recvcounts,
+ mpi_coll_allgatherv_fun(sendbuf, sendcount, sendtype, recvbuf, recvcounts,
displs, recvtype, comm);
retval = MPI_SUCCESS;
}
int PMPI_Reduce_scatter(void *sendbuf, void *recvbuf, int *recvcounts,
MPI_Datatype datatype, MPI_Op op, MPI_Comm comm)
{
- int retval, i, size, count;
- int *displs;
+ int retval;
int rank = comm != MPI_COMM_NULL ? smpi_process_index() : -1;
-
smpi_bench_end();
#ifdef HAVE_TRACING
TRACE_smpi_computing_out(rank);
} else if (recvcounts == NULL) {
retval = MPI_ERR_ARG;
} else {
- /* arbitrarily choose root as rank 0 */
- /* TODO: faster direct implementation ? */
- size = smpi_comm_size(comm);
- count = 0;
- displs = xbt_new(int, size);
- for (i = 0; i < size; i++) {
- count += recvcounts[i];
- displs[i] = 0;
- }
- mpi_coll_reduce_fun(sendbuf, recvbuf, count, datatype, op, 0, comm);
- smpi_mpi_scatterv(recvbuf, recvcounts, displs, datatype, recvbuf,
- recvcounts[rank], datatype, 0, comm);
- xbt_free(displs);
+
+ mpi_coll_reduce_scatter_fun(sendbuf, recvbuf, recvcounts,
+ datatype, op, comm);
retval = MPI_SUCCESS;
}
#ifdef HAVE_TRACING
} else if (count<0){
retval = MPI_ERR_COUNT;
} else {
- retval = smpi_datatype_contiguous(count, old_type, new_type);
+ retval = smpi_datatype_contiguous(count, old_type, new_type, 0);
}
smpi_bench_begin();
return retval;