+/* 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"
/* IMPLEMENTED BY PITCH PATARASUK
smpi_coll_tuned_allreduce_lr(void *sbuf, void *rbuf, int rcount,
MPI_Datatype dtype, MPI_Op op, MPI_Comm comm)
{
- int tag = 5000;
+ int tag = COLL_TAG_ALLREDUCE;
MPI_Status status;
int rank, i, size, count;
int send_offset, recv_offset;
int remainder, remainder_flag, remainder_offset;
- rank = smpi_comm_rank(MPI_COMM_WORLD);
- size = smpi_comm_size(MPI_COMM_WORLD);
+ rank = smpi_comm_rank(comm);
+ size = smpi_comm_size(comm);
/* make it compatible with all data type */
MPI_Aint extent;
/* when communication size is not divisible by number of process:
call the native implementation for the remain chunk at the end of the operation */
- else if (rcount % size != 0) {
+ if (rcount % size != 0) {
remainder = rcount % size;
remainder_flag = 1;
remainder_offset = (rcount / size) * size * extent;
} else {
- remainder_flag = remainder_offset = 0;
+ remainder = remainder_flag = remainder_offset = 0;
}
/* size of each point-to-point communication is equal to the size of the whole message
((rank + size - 1) % size), tag + i, comm, &status);
// compute result to rbuf+recv_offset
- star_reduction(op, (char *) sbuf + recv_offset, (char *) rbuf + recv_offset,
+ smpi_op_apply(op, (char *) sbuf + recv_offset, (char *) rbuf + recv_offset,
&count, &dtype);
}