#include "private.h"
#include <vector>
+#include "src/smpi/smpi_group.hpp"
XBT_LOG_NEW_DEFAULT_SUBCATEGORY(smpi_rma, smpi, "Logging specific to SMPI (RMA operations)");
-msg_bar_t creation_bar = nullptr;
-
typedef struct s_smpi_mpi_win{
void* base;
MPI_Aint size;
MPI_Info info;
int assert;
std::vector<MPI_Request> *requests;
+ xbt_mutex_t mut;
msg_bar_t bar;
MPI_Win* connected_wins;
char* name;
MPI_Win smpi_mpi_win_create( void *base, MPI_Aint size, int disp_unit, MPI_Info info, MPI_Comm comm){
- MPI_Win win;
-
- int comm_size = smpi_comm_size(comm);
- int rank=smpi_comm_rank(comm);
+ int comm_size = comm->size();
+ int rank = comm->rank();
XBT_DEBUG("Creating window");
- win = xbt_new(s_smpi_mpi_win_t, 1);
+ MPI_Win win = xbt_new(s_smpi_mpi_win_t, 1);
win->base = base;
win->size = size;
win->disp_unit = disp_unit;
win->opened = 0;
win->group = MPI_GROUP_NULL;
win->requests = new std::vector<MPI_Request>();
+ win->mut=xbt_mutex_init();
win->connected_wins = xbt_new0(MPI_Win, comm_size);
win->connected_wins[rank] = win;
win->count = 0;
int smpi_mpi_win_free( MPI_Win* win){
//As per the standard, perform a barrier to ensure every async comm is finished
MSG_barrier_wait((*win)->bar);
+ xbt_mutex_acquire((*win)->mut);
delete (*win)->requests;
+ xbt_mutex_release((*win)->mut);
xbt_free((*win)->connected_wins);
if ((*win)->name != nullptr){
xbt_free((*win)->name);
}
mpi_coll_barrier_fun((*win)->comm);
- int rank=smpi_comm_rank((*win)->comm);
+ int rank=(*win)->comm->rank();
if(rank == 0)
MSG_barrier_destroy((*win)->bar);
+ xbt_mutex_destroy((*win)->mut);
xbt_free(*win);
*win = MPI_WIN_NULL;
return MPI_SUCCESS;
void smpi_mpi_win_get_group(MPI_Win win, MPI_Group* group){
if(win->comm != MPI_COMM_NULL){
- *group = smpi_comm_group(win->comm);
+ *group = win->comm->group();
} else {
*group = MPI_GROUP_NULL;
}
win->name = xbt_strdup(name);
}
-int smpi_mpi_win_fence( int assert, MPI_Win win){
+int smpi_mpi_win_fence(int assert, MPI_Win win)
+{
XBT_DEBUG("Entering fence");
- if(win->opened==0)
+ if (win->opened == 0)
win->opened=1;
- if(assert != MPI_MODE_NOPRECEDE){
+ if (assert != MPI_MODE_NOPRECEDE) {
+ // This is not the first fence => finalize what came before
MSG_barrier_wait(win->bar);
-
+ xbt_mutex_acquire(win->mut);
+ // This (simulated) mutex ensures that no process pushes to the vector of requests during the waitall.
+ // Without this, the vector could get redimensionned when another process pushes.
+ // This would result in the array used by smpi_mpi_waitall() to be invalidated.
+ // Another solution would be to copy the data and cleanup the vector *before* smpi_mpi_waitall
std::vector<MPI_Request> *reqs = win->requests;
int size = static_cast<int>(reqs->size());
// start all requests that have been prepared by another process
- if(size>0){
- for(auto req: *reqs){
- if (req && (req->flags & PREPARED))
- smpi_mpi_start(req);
- }
+ if (size > 0) {
+ for (const auto& req : *reqs) {
+ if (req && (req->flags & PREPARED))
+ smpi_mpi_start(req);
+ }
- MPI_Request* treqs = &(*reqs)[0];
+ MPI_Request* treqs = &(*reqs)[0];
- smpi_mpi_waitall(size,treqs,MPI_STATUSES_IGNORE);
+ smpi_mpi_waitall(size, treqs, MPI_STATUSES_IGNORE);
}
win->count=0;
+ xbt_mutex_release(win->mut);
}
win->assert = assert;
MSG_barrier_wait(win->bar);
- XBT_DEBUG("Leaving fence ");
+ XBT_DEBUG("Leaving fence");
return MPI_SUCCESS;
}
void* recv_addr = static_cast<void*> ( static_cast<char*>(recv_win->base) + target_disp * recv_win->disp_unit);
XBT_DEBUG("Entering MPI_Put to %d", target_rank);
- if(target_rank != smpi_comm_rank(win->comm)){
+ if(target_rank != win->comm->rank()){
//prepare send_request
MPI_Request sreq = smpi_rma_send_init(origin_addr, origin_count, origin_datatype, smpi_process_index(),
- smpi_group_index(smpi_comm_group(win->comm),target_rank), SMPI_RMA_TAG+1, win->comm, MPI_OP_NULL);
+ win->comm->group()->index(target_rank), SMPI_RMA_TAG+1, win->comm, MPI_OP_NULL);
//prepare receiver request
MPI_Request rreq = smpi_rma_recv_init(recv_addr, target_count, target_datatype, smpi_process_index(),
- smpi_group_index(smpi_comm_group(win->comm),target_rank), SMPI_RMA_TAG+1, recv_win->comm, MPI_OP_NULL);
+ win->comm->group()->index(target_rank), SMPI_RMA_TAG+1, recv_win->comm, MPI_OP_NULL);
//push request to receiver's win
+ xbt_mutex_acquire(recv_win->mut);
recv_win->requests->push_back(rreq);
-
+ xbt_mutex_release(recv_win->mut);
//start send
smpi_mpi_start(sreq);
//push request to sender's win
+ xbt_mutex_acquire(win->mut);
win->requests->push_back(sreq);
+ xbt_mutex_release(win->mut);
}else{
smpi_datatype_copy(origin_addr, origin_count, origin_datatype, recv_addr, target_count, target_datatype);
}
void* send_addr = static_cast<void*>(static_cast<char*>(send_win->base) + target_disp * send_win->disp_unit);
XBT_DEBUG("Entering MPI_Get from %d", target_rank);
- if(target_rank != smpi_comm_rank(win->comm)){
+ if(target_rank != win->comm->rank()){
//prepare send_request
MPI_Request sreq = smpi_rma_send_init(send_addr, target_count, target_datatype,
- smpi_group_index(smpi_comm_group(win->comm),target_rank), smpi_process_index(), SMPI_RMA_TAG+2, send_win->comm,
+ win->comm->group()->index(target_rank), smpi_process_index(), SMPI_RMA_TAG+2, send_win->comm,
MPI_OP_NULL);
//prepare receiver request
MPI_Request rreq = smpi_rma_recv_init(origin_addr, origin_count, origin_datatype,
- smpi_group_index(smpi_comm_group(win->comm),target_rank), smpi_process_index(), SMPI_RMA_TAG+2, win->comm,
+ win->comm->group()->index(target_rank), smpi_process_index(), SMPI_RMA_TAG+2, win->comm,
MPI_OP_NULL);
//start the send, with another process than us as sender.
smpi_mpi_start(sreq);
-
//push request to receiver's win
+ xbt_mutex_acquire(send_win->mut);
send_win->requests->push_back(sreq);
+ xbt_mutex_release(send_win->mut);
//start recv
smpi_mpi_start(rreq);
-
//push request to sender's win
+ xbt_mutex_acquire(win->mut);
win->requests->push_back(rreq);
+ xbt_mutex_release(win->mut);
}else{
smpi_datatype_copy(send_addr, target_count, target_datatype, origin_addr, origin_count, origin_datatype);
}
//As the tag will be used for ordering of the operations, add count to it
//prepare send_request
MPI_Request sreq = smpi_rma_send_init(origin_addr, origin_count, origin_datatype,
- smpi_process_index(), smpi_group_index(smpi_comm_group(win->comm),target_rank), SMPI_RMA_TAG+3+win->count, win->comm, op);
+ smpi_process_index(), win->comm->group()->index(target_rank), SMPI_RMA_TAG+3+win->count, win->comm, op);
//prepare receiver request
MPI_Request rreq = smpi_rma_recv_init(recv_addr, target_count, target_datatype,
- smpi_process_index(), smpi_group_index(smpi_comm_group(win->comm),target_rank), SMPI_RMA_TAG+3+win->count, recv_win->comm, op);
+ smpi_process_index(), win->comm->group()->index(target_rank), SMPI_RMA_TAG+3+win->count, recv_win->comm, op);
win->count++;
//push request to receiver's win
+ xbt_mutex_acquire(recv_win->mut);
recv_win->requests->push_back(rreq);
+ xbt_mutex_release(recv_win->mut);
//start send
smpi_mpi_start(sreq);
//push request to sender's win
+ xbt_mutex_acquire(win->mut);
win->requests->push_back(sreq);
+ xbt_mutex_release(win->mut);
return MPI_SUCCESS;
}
must complete, without further dependencies. */
//naive, blocking implementation.
- int i=0,j=0;
- int size = smpi_group_size(group);
- MPI_Request* reqs = xbt_new0(MPI_Request, size);
-
- while(j!=size){
- int src=smpi_group_index(group,j);
- if(src!=smpi_process_index()){
- reqs[i]=smpi_irecv_init(nullptr, 0, MPI_CHAR, src,SMPI_RMA_TAG+4, MPI_COMM_WORLD);
- i++;
- }
- j++;
+ int i = 0;
+ int j = 0;
+ int size = group->size();
+ MPI_Request* reqs = xbt_new0(MPI_Request, size);
+
+ while (j != size) {
+ int src = group->index(j);
+ if (src != smpi_process_index() && src != MPI_UNDEFINED) {
+ reqs[i] = smpi_irecv_init(nullptr, 0, MPI_CHAR, src, SMPI_RMA_TAG + 4, MPI_COMM_WORLD);
+ i++;
+ }
+ j++;
}
size=i;
smpi_mpi_startall(size, reqs);
xbt_free(reqs);
win->opened++; //we're open for business !
win->group=group;
- smpi_group_use(group);
+ group->use();
return MPI_SUCCESS;
}
int smpi_mpi_win_post(MPI_Group group, int assert, MPI_Win win){
//let's make a synchronous send here
- int i=0,j=0;
- int size = smpi_group_size(group);
+ int i = 0;
+ int j = 0;
+ int size = group->size();
MPI_Request* reqs = xbt_new0(MPI_Request, size);
while(j!=size){
- int dst=smpi_group_index(group,j);
- if(dst!=smpi_process_index()){
+ int dst=group->index(j);
+ if(dst!=smpi_process_index() && dst!=MPI_UNDEFINED){
reqs[i]=smpi_mpi_send_init(nullptr, 0, MPI_CHAR, dst, SMPI_RMA_TAG+4, MPI_COMM_WORLD);
i++;
}
xbt_free(reqs);
win->opened++; //we're open for business !
win->group=group;
- smpi_group_use(group);
+ group->use();
return MPI_SUCCESS;
}
xbt_die("Complete called on already opened MPI_Win");
XBT_DEBUG("Entering MPI_Win_Complete");
- int i=0,j=0;
- int size = smpi_group_size(win->group);
+ int i = 0;
+ int j = 0;
+ int size = win->group->size();
MPI_Request* reqs = xbt_new0(MPI_Request, size);
while(j!=size){
- int dst=smpi_group_index(win->group,j);
- if(dst!=smpi_process_index()){
+ int dst=win->group->index(j);
+ if(dst!=smpi_process_index() && dst!=MPI_UNDEFINED){
reqs[i]=smpi_mpi_send_init(nullptr, 0, MPI_CHAR, dst, SMPI_RMA_TAG+5, MPI_COMM_WORLD);
i++;
}
xbt_free(reqs);
//now we can finish RMA calls
-
+ xbt_mutex_acquire(win->mut);
std::vector<MPI_Request> *reqqs = win->requests;
size = static_cast<int>(reqqs->size());
XBT_DEBUG("Win_complete - Finishing %d RMA calls", size);
- // start all requests that have been prepared by another process
- for (auto req: *reqqs){
- if (req && (req->flags & PREPARED))
- smpi_mpi_start(req);
+ if (size > 0) {
+ // start all requests that have been prepared by another process
+ for (const auto& req : *reqqs) {
+ if (req && (req->flags & PREPARED))
+ smpi_mpi_start(req);
+ }
+
+ MPI_Request* treqs = &(*reqqs)[0];
+ smpi_mpi_waitall(size, treqs, MPI_STATUSES_IGNORE);
+ reqqs->clear();
}
+ xbt_mutex_release(win->mut);
- MPI_Request* treqs = &(*reqqs)[0];
- smpi_mpi_waitall(size,treqs,MPI_STATUSES_IGNORE);
- reqqs->clear();
- smpi_group_unuse(win->group);
+ win->group->unuse();
win->opened--; //we're closed for business !
return MPI_SUCCESS;
}
//naive, blocking implementation.
XBT_DEBUG("Entering MPI_Win_Wait");
int i=0,j=0;
- int size = smpi_group_size(win->group);
+ int size = win->group->size();
MPI_Request* reqs = xbt_new0(MPI_Request, size);
while(j!=size){
- int src=smpi_group_index(win->group,j);
- if(src!=smpi_process_index()){
+ int src=win->group->index(j);
+ if(src!=smpi_process_index() && src!=MPI_UNDEFINED){
reqs[i]=smpi_irecv_init(nullptr, 0, MPI_CHAR, src,SMPI_RMA_TAG+5, MPI_COMM_WORLD);
i++;
}
smpi_mpi_request_free(&reqs[i]);
}
xbt_free(reqs);
-
+ xbt_mutex_acquire(win->mut);
std::vector<MPI_Request> *reqqs = win->requests;
size = static_cast<int>(reqqs->size());
- XBT_DEBUG("Win_complete - Finishing %d RMA calls", size);
+ XBT_DEBUG("Win_wait - Finishing %d RMA calls", size);
+ if (size > 0) {
+ // start all requests that have been prepared by another process
+ for (const auto& req : *reqqs) {
+ if (req && (req->flags & PREPARED))
+ smpi_mpi_start(req);
+ }
- // start all requests that have been prepared by another process
- for(auto req: *reqqs){
- if (req && (req->flags & PREPARED))
- smpi_mpi_start(req);
+ MPI_Request* treqs = &(*reqqs)[0];
+ smpi_mpi_waitall(size, treqs, MPI_STATUSES_IGNORE);
+ reqqs->clear();
}
+ xbt_mutex_release(win->mut);
- MPI_Request* treqs = &(*reqqs)[0];
- smpi_mpi_waitall(size,treqs,MPI_STATUSES_IGNORE);
- reqqs->clear();
- smpi_group_unuse(win->group);
+ win->group->unuse();
win->opened--; //we're opened for business !
return MPI_SUCCESS;
}
