-#include "colls.h"
+#include "colls_private.h"
/*
reduce
if (count == 0)
return 0;
- MPI_Comm_rank(comm, &rank);
- MPI_Comm_size(comm, &comm_size);
+ rank = smpi_comm_rank(comm);
+ comm_size = smpi_comm_size(comm);
- MPI_Type_extent(datatype, &extent);
- MPI_Type_size(datatype, &type_size);
+ extent = smpi_datatype_get_extent(datatype);
+ type_size = smpi_datatype_size(datatype);
/* find nearest power-of-two less than or equal to comm_size */
pof2 = 1;
if (count < comm_size) {
new_count = comm_size;
- send_ptr = (void *) malloc(new_count * extent);
- recv_ptr = (void *) malloc(new_count * extent);
- tmp_buf = (void *) malloc(new_count * extent);
+ send_ptr = (void *) xbt_malloc(new_count * extent);
+ recv_ptr = (void *) xbt_malloc(new_count * extent);
+ tmp_buf = (void *) xbt_malloc(new_count * extent);
memcpy(send_ptr, sendbuf, extent * new_count);
//if ((rank != root))
- MPI_Sendrecv(send_ptr, new_count, datatype, rank, tag,
+ smpi_mpi_sendrecv(send_ptr, new_count, datatype, rank, tag,
recv_ptr, new_count, datatype, rank, tag, comm, &status);
rem = comm_size - pof2;
if (rank < 2 * rem) {
if (rank % 2 != 0) {
/* odd */
- MPI_Send(recv_ptr, new_count, datatype, rank - 1, tag, comm);
+ smpi_mpi_send(recv_ptr, new_count, datatype, rank - 1, tag, comm);
newrank = -1;
} else {
- MPI_Recv(tmp_buf, count, datatype, rank + 1, tag, comm, &status);
+ smpi_mpi_recv(tmp_buf, count, datatype, rank + 1, tag, comm, &status);
star_reduction(op, tmp_buf, recv_ptr, &new_count, &datatype);
newrank = rank / 2;
}
} else /* rank >= 2*rem */
newrank = rank - rem;
- cnts = (int *) malloc(pof2 * sizeof(int));
- disps = (int *) malloc(pof2 * sizeof(int));
+ cnts = (int *) xbt_malloc(pof2 * sizeof(int));
+ disps = (int *) xbt_malloc(pof2 * sizeof(int));
if (newrank != -1) {
for (i = 0; i < (pof2 - 1); i++)
}
/* Send data from recvbuf. Recv into tmp_buf */
- MPI_Sendrecv((char *) recv_ptr +
+ smpi_mpi_sendrecv((char *) recv_ptr +
disps[send_idx] * extent,
send_cnt, datatype,
dst, tag,
for (i = 1; i < pof2; i++)
disps[i] = disps[i - 1] + cnts[i - 1];
- MPI_Recv(recv_ptr, cnts[0], datatype, 0, tag, comm, &status);
+ smpi_mpi_recv(recv_ptr, cnts[0], datatype, 0, tag, comm, &status);
newrank = 0;
send_idx = 0;
last_idx = 2;
} else if (newrank == 0) {
- MPI_Send(recv_ptr, cnts[0], datatype, root, tag, comm);
+ smpi_mpi_send(recv_ptr, cnts[0], datatype, root, tag, comm);
newrank = -1;
}
newroot = 0;
}
if (newdst_tree_root == newroot_tree_root) {
- MPI_Send((char *) recv_ptr +
+ smpi_mpi_send((char *) recv_ptr +
disps[send_idx] * extent,
send_cnt, datatype, dst, tag, comm);
break;
} else {
- MPI_Recv((char *) recv_ptr +
+ smpi_mpi_recv((char *) recv_ptr +
disps[recv_idx] * extent,
recv_cnt, datatype, dst, tag, comm, &status);
}
else if (count >= comm_size) {
- tmp_buf = (void *) malloc(count * extent);
+ tmp_buf = (void *) xbt_malloc(count * extent);
//if ((rank != root))
- MPI_Sendrecv(sendbuf, count, datatype, rank, tag,
+ smpi_mpi_sendrecv(sendbuf, count, datatype, rank, tag,
recvbuf, count, datatype, rank, tag, comm, &status);
rem = comm_size - pof2;
if (rank < 2 * rem) {
if (rank % 2 != 0) { /* odd */
- MPI_Send(recvbuf, count, datatype, rank - 1, tag, comm);
+ smpi_mpi_send(recvbuf, count, datatype, rank - 1, tag, comm);
newrank = -1;
}
else {
- MPI_Recv(tmp_buf, count, datatype, rank + 1, tag, comm, &status);
+ smpi_mpi_recv(tmp_buf, count, datatype, rank + 1, tag, comm, &status);
star_reduction(op, tmp_buf, recvbuf, &count, &datatype);
newrank = rank / 2;
}
} else /* rank >= 2*rem */
newrank = rank - rem;
- cnts = (int *) malloc(pof2 * sizeof(int));
- disps = (int *) malloc(pof2 * sizeof(int));
+ cnts = (int *) xbt_malloc(pof2 * sizeof(int));
+ disps = (int *) xbt_malloc(pof2 * sizeof(int));
if (newrank != -1) {
for (i = 0; i < (pof2 - 1); i++)
}
/* Send data from recvbuf. Recv into tmp_buf */
- MPI_Sendrecv((char *) recvbuf +
+ smpi_mpi_sendrecv((char *) recvbuf +
disps[send_idx] * extent,
send_cnt, datatype,
dst, tag,
for (i = 1; i < pof2; i++)
disps[i] = disps[i - 1] + cnts[i - 1];
- MPI_Recv(recvbuf, cnts[0], datatype, 0, tag, comm, &status);
+ smpi_mpi_recv(recvbuf, cnts[0], datatype, 0, tag, comm, &status);
newrank = 0;
send_idx = 0;
last_idx = 2;
} else if (newrank == 0) {
- MPI_Send(recvbuf, cnts[0], datatype, root, tag, comm);
+ smpi_mpi_send(recvbuf, cnts[0], datatype, root, tag, comm);
newrank = -1;
}
newroot = 0;
}
if (newdst_tree_root == newroot_tree_root) {
- MPI_Send((char *) recvbuf +
+ smpi_mpi_send((char *) recvbuf +
disps[send_idx] * extent,
send_cnt, datatype, dst, tag, comm);
break;
} else {
- MPI_Recv((char *) recvbuf +
+ smpi_mpi_recv((char *) recvbuf +
disps[recv_idx] * extent,
recv_cnt, datatype, dst, tag, comm, &status);
}