+/* Copyright (c) 2013-2014. The SimGrid Team.
+ * All rights reserved. */
+
+/* This program is free software; you can redistribute it and/or modify it
+ * under the terms of the license (GNU LGPL) which comes with this package. */
+
#include "colls_private.h"
/*
int recv_idx, last_idx = 0, newdst;
int dst, send_cnt, recv_cnt, newroot, newdst_tree_root;
int newroot_tree_root, new_count;
- int tag = COLL_TAG_REDUCE;
+ int tag = COLL_TAG_REDUCE,temporary_buffer=0;
void *send_ptr, *recv_ptr, *tmp_buf;
cnts = NULL;
return 0;
rank = smpi_comm_rank(comm);
comm_size = smpi_comm_size(comm);
+
- extent = smpi_datatype_get_extent(datatype);
+ extent = smpi_datatype_get_extent(datatype);
+ /* If I'm not the root, then my recvbuf may not be valid, therefore
+ I have to allocate a temporary one */
+ if (rank != root && !recvbuf) {
+ temporary_buffer=1;
+ recvbuf = (void *)smpi_get_tmp_recvbuffer(count * extent);
+ }
/* find nearest power-of-two less than or equal to comm_size */
pof2 = 1;
while (pof2 <= comm_size)
if (count < comm_size) {
new_count = comm_size;
- 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 * count);
+ send_ptr = (void *) smpi_get_tmp_sendbuffer(new_count * extent);
+ recv_ptr = (void *) smpi_get_tmp_recvbuffer(new_count * extent);
+ tmp_buf = (void *) smpi_get_tmp_sendbuffer(new_count * extent);
+ memcpy(send_ptr, sendbuf != MPI_IN_PLACE ? sendbuf : recvbuf, extent * count);
//if ((rank != root))
smpi_mpi_sendrecv(send_ptr, new_count, datatype, rank, tag,
}
}
memcpy(recvbuf, recv_ptr, extent * count);
- free(send_ptr);
- free(recv_ptr);
+ smpi_free_tmp_buffer(send_ptr);
+ smpi_free_tmp_buffer(recv_ptr);
}
- else if (count >= comm_size) {
- tmp_buf = (void *) xbt_malloc(count * extent);
+ else /* (count >= comm_size) */ {
+ tmp_buf = (void *) smpi_get_tmp_sendbuffer(count * extent);
//if ((rank != root))
- smpi_mpi_sendrecv(sendbuf, count, datatype, rank, tag,
+ smpi_mpi_sendrecv(sendbuf != MPI_IN_PLACE ? sendbuf : recvbuf, count, datatype, rank, tag,
recvbuf, count, datatype, rank, tag, comm, &status);
rem = comm_size - pof2;
}
}
if (tmp_buf)
- free(tmp_buf);
+ smpi_free_tmp_buffer(tmp_buf);
+ if(temporary_buffer==1) smpi_free_tmp_buffer(recvbuf);
if (cnts)
free(cnts);
if (disps)