X-Git-Url: http://info.iut-bm.univ-fcomte.fr/pub/gitweb/simgrid.git/blobdiff_plain/40f10b96e597c4a0ff061d2b8f61dd6e7fbe2086..74430cd04d3de487fb936817266b3af0df7638c0:/src/smpi/smpi_rma.c diff --git a/src/smpi/smpi_rma.c b/src/smpi/smpi_rma.c index 89224f8027..356ab585e0 100644 --- a/src/smpi/smpi_rma.c +++ b/src/smpi/smpi_rma.c @@ -18,11 +18,12 @@ typedef struct s_smpi_mpi_win{ MPI_Aint size; int disp_unit; MPI_Comm comm; - //MPI_Info info + MPI_Info info; int assert; xbt_dynar_t requests; xbt_bar_t bar; MPI_Win* connected_wins; + char* name; } s_smpi_mpi_win_t; @@ -39,8 +40,11 @@ MPI_Win smpi_mpi_win_create( void *base, MPI_Aint size, int disp_unit, MPI_Info win->size = size; win->disp_unit = disp_unit; win->assert = 0; - //win->info = info; + win->info = info; + if(info!=MPI_INFO_NULL) + info->refcount++; win->comm = comm; + win->name = NULL; win->requests = xbt_dynar_new(sizeof(MPI_Request), NULL); win->connected_wins = xbt_malloc0(comm_size*sizeof(MPI_Win)); win->connected_wins[rank] = win; @@ -74,11 +78,36 @@ int smpi_mpi_win_free( MPI_Win* win){ xbt_barrier_wait((*win)->bar); xbt_dynar_free(&(*win)->requests); xbt_free((*win)->connected_wins); + if ((*win)->name != NULL){ + xbt_free((*win)->name); + } + if((*win)->info!=MPI_INFO_NULL){ + MPI_Info_free(&(*win)->info); + } xbt_free(*win); win = MPI_WIN_NULL; return MPI_SUCCESS; } +void smpi_mpi_win_get_name(MPI_Win win, char* name, int* length){ + if(win->name==NULL){ + *length=0; + name=NULL; + return; + } + *length = strlen(win->name); + strcpy(name, win->name); +} + +void smpi_mpi_win_get_group(MPI_Win win, MPI_Group* group){ + if(win->comm != MPI_COMM_NULL) + *group = smpi_comm_group(win->comm); +} + +void smpi_mpi_win_set_name(MPI_Win win, char* name){ + win->name = strdup(name);; +} + int smpi_mpi_win_fence( int assert, MPI_Win win){ @@ -116,7 +145,7 @@ int smpi_mpi_put( void *origin_addr, int origin_count, MPI_Datatype origin_datat //get receiver pointer MPI_Win recv_win = win->connected_wins[target_rank]; - void* recv_addr = (void*) ( ((char*)recv_win->base) + target_disp * smpi_datatype_size(target_datatype)); + void* recv_addr = (void*) ( ((char*)recv_win->base) + target_disp * recv_win->disp_unit); smpi_datatype_use(origin_datatype); smpi_datatype_use(target_datatype); XBT_DEBUG("Entering MPI_Put to %d", target_rank); @@ -124,11 +153,11 @@ int smpi_mpi_put( void *origin_addr, int origin_count, MPI_Datatype origin_datat if(target_rank != smpi_comm_rank(win->comm)){ //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), RMA_TAG+1, win->comm); + smpi_process_index(), smpi_group_index(smpi_comm_group(win->comm),target_rank), 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), RMA_TAG+1, recv_win->comm); + smpi_process_index(), smpi_group_index(smpi_comm_group(win->comm),target_rank), RMA_TAG+1, recv_win->comm, MPI_OP_NULL); //push request to receiver's win xbt_dynar_push_as(recv_win->requests, MPI_Request, rreq); @@ -138,6 +167,9 @@ int smpi_mpi_put( void *origin_addr, int origin_count, MPI_Datatype origin_datat //push request to sender's win xbt_dynar_push_as(win->requests, MPI_Request, sreq); + }else{ + smpi_datatype_copy(origin_addr, origin_count, origin_datatype, + recv_addr, target_count, target_datatype); } return MPI_SUCCESS; @@ -149,7 +181,7 @@ int smpi_mpi_get( void *origin_addr, int origin_count, MPI_Datatype origin_datat //get sender pointer MPI_Win send_win = win->connected_wins[target_rank]; - void* send_addr = (void*)( ((char*)send_win->base) + target_disp * smpi_datatype_size(target_datatype)); + void* send_addr = (void*)( ((char*)send_win->base) + target_disp * send_win->disp_unit); smpi_datatype_use(origin_datatype); smpi_datatype_use(target_datatype); XBT_DEBUG("Entering MPI_Get from %d", target_rank); @@ -157,12 +189,15 @@ int smpi_mpi_get( void *origin_addr, int origin_count, MPI_Datatype origin_datat if(target_rank != smpi_comm_rank(win->comm)){ //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(), RMA_TAG+2, send_win->comm); + smpi_group_index(smpi_comm_group(win->comm),target_rank), smpi_process_index(), 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(), RMA_TAG+2, win->comm); - + smpi_group_index(smpi_comm_group(win->comm),target_rank), smpi_process_index(), 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_dynar_push_as(send_win->requests, MPI_Request, sreq); @@ -171,6 +206,9 @@ int smpi_mpi_get( void *origin_addr, int origin_count, MPI_Datatype origin_datat //push request to sender's win xbt_dynar_push_as(win->requests, MPI_Request, rreq); + }else{ + smpi_datatype_copy(send_addr, target_count, target_datatype, + origin_addr, origin_count, origin_datatype); } return MPI_SUCCESS; @@ -180,33 +218,33 @@ int smpi_mpi_get( void *origin_addr, int origin_count, MPI_Datatype origin_datat int smpi_mpi_accumulate( void *origin_addr, int origin_count, MPI_Datatype origin_datatype, int target_rank, MPI_Aint target_disp, int target_count, MPI_Datatype target_datatype, MPI_Op op, MPI_Win win) { + //FIXME: local version //get receiver pointer MPI_Win recv_win = win->connected_wins[target_rank]; - void* recv_addr = (void*)( ((char*)recv_win->base) + target_disp * smpi_datatype_size(target_datatype) ); + void* recv_addr = (void*)( ((char*)recv_win->base) + target_disp * recv_win->disp_unit); XBT_DEBUG("Entering MPI_Accumulate to %d", target_rank); smpi_datatype_use(origin_datatype); smpi_datatype_use(target_datatype); - if(target_rank != smpi_comm_rank(win->comm)){ + //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), RMA_TAG+3, win->comm); + smpi_process_index(), smpi_group_index(smpi_comm_group(win->comm),target_rank), RMA_TAG+3, win->comm, op); //prepare receiver request - MPI_Request rreq = smpi_rma_recv_init(NULL, 0, target_datatype, - smpi_process_index(), smpi_group_index(smpi_comm_group(win->comm),target_rank), RMA_TAG+3, recv_win->comm); - rreq->flags |= ACCUMULATE; + 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), RMA_TAG+3, recv_win->comm, op); //push request to receiver's win xbt_dynar_push_as(recv_win->requests, MPI_Request, rreq); //start send smpi_mpi_start(sreq); + //push request to sender's win xbt_dynar_push_as(win->requests, MPI_Request, sreq); - } - //perform actual accumulation - smpi_op_apply(op, origin_addr, recv_addr, &origin_count, &origin_datatype); + + return MPI_SUCCESS; }