Ignore:
Timestamp:
05/18/17 17:40:03 (7 years ago)
Author:
yushan
Message:

test_remap back to work. No thread for now

File:
1 edited

Legend:

Unmodified
Added
Removed

  • XIOS/dev/branch_yushan_merged/extern/src_ep_dev/ep_wait.cpp

    r1134 r1138  
    11/*! 
    2    \file ep_wait.cpp 
    3    \since 2 may 2016 
     2  \file ep_wait.cpp 
     3  \since 2 may 2016 
    44 
    5    \brief Definitions of MPI wait function: MPI_Wait, MPI_Waitall, MPI_Waitsome, MPI_Waitany 
    6  */ 
     5  \brief Definitions of MPI wait function: MPI_Wait, MPI_Waitall, MPI_Waitsome, MPI_Waitany 
     6  */ 
    77 
    88#include "ep_lib.hpp" 
    … …  
    1616namespace ep_lib {       
    1717 
    18         int MPI_Wait(MPI_Request *request, MPI_Status *status) 
    19         { 
     18    int MPI_Wait(MPI_Request *request, MPI_Status *status) 
     19    { 
    2020 
    21     if(request->type == 1) 
    22     { 
    23       ::MPI_Request mpi_request = static_cast< ::MPI_Request >(request->mpi_request); 
    24       ::MPI_Status mpi_status; 
    25       ::MPI_Wait(&mpi_request, &mpi_status); 
    26  
    27  
    28        
    29       status->mpi_status = &mpi_status; 
    30       status->ep_src = request->ep_src; 
    31       status->ep_tag = request->ep_tag; 
    32       status->ep_datatype = request->ep_datatype; 
    33  
    34       return 0; 
    35     } 
    36  
    37     if(request->type == 2) 
    38     { 
    39       int flag = false; 
    40       MPI_Message message; 
    41  
    42       while(!flag) 
    43       { 
    44         Message_Check(request->comm); 
    45         #pragma omp flush 
    46         MPI_Improbe(request->ep_src, request->ep_tag, request->comm, &flag, &message, status); 
    47       } 
    48  
    49       int count; 
    50       MPI_Get_count(status, request->ep_datatype, &count); 
    51       MPI_Mrecv(request->buf, count, request->ep_datatype, &message, status); 
    52       status->ep_datatype = request->ep_datatype; 
    53  
    54       //check_sum_recv(request->buf, count, request->ep_datatype, request->ep_src, request->ep_tag, request->comm, 2); 
    55  
    56       return 0; 
    57     } 
    58  
    59     if(request->type == 3) 
    60     { 
    61       ::MPI_Request mpi_request = static_cast< ::MPI_Request >(request->mpi_request); 
    62       ::MPI_Status mpi_status; 
    63       ::MPI_Wait(&mpi_request, &mpi_status); 
    64        
    65       status->mpi_status = new ::MPI_Status(mpi_status); 
    66       status->ep_src = request->ep_src; 
    67       status->ep_tag = request->ep_tag; 
    68       status->ep_datatype = request->ep_datatype; 
    69  
    70       //int count; 
    71       //MPI_Get_count(status, request->ep_datatype, &count); 
    72       //check_sum_recv(request->buf, count, request->ep_datatype, request->ep_src, request->ep_tag, request->comm, 2); 
    73     } 
    74           return MPI_SUCCESS; 
    75         } 
     21        if(request->type == 1) 
     22        { 
     23            ::MPI_Request mpi_request = static_cast< ::MPI_Request >(request->mpi_request); 
     24            ::MPI_Status mpi_status; 
     25            ::MPI_Wait(&mpi_request, &mpi_status); 
    7626 
    7727 
    7828 
    79          
     29            status->mpi_status = &mpi_status; 
     30            status->ep_src = request->ep_src; 
     31            status->ep_tag = request->ep_tag; 
     32            status->ep_datatype = request->ep_datatype; 
     33 
     34            return 0; 
     35        } 
     36 
     37        if(request->type == 2) 
     38        { 
     39            int flag = false; 
     40            MPI_Message message; 
     41 
     42            while(!flag) 
     43            { 
     44                Message_Check(request->comm); 
     45#pragma omp flush 
     46                MPI_Improbe(request->ep_src, request->ep_tag, request->comm, &flag, &message, status); 
     47            } 
     48 
     49            int count; 
     50            MPI_Get_count(status, request->ep_datatype, &count); 
     51            MPI_Mrecv(request->buf, count, request->ep_datatype, &message, status); 
     52            status->ep_datatype = request->ep_datatype; 
     53 
     54            //check_sum_recv(request->buf, count, request->ep_datatype, request->ep_src, request->ep_tag, request->comm, 2); 
     55 
     56            return 0; 
     57        } 
     58 
     59        if(request->type == 3) 
     60        { 
     61            ::MPI_Request mpi_request = static_cast< ::MPI_Request >(request->mpi_request); 
     62            ::MPI_Status mpi_status; 
     63            ::MPI_Wait(&mpi_request, &mpi_status); 
     64 
     65            status->mpi_status = new ::MPI_Status(mpi_status); 
     66            status->ep_src = request->ep_src; 
     67            status->ep_tag = request->ep_tag; 
     68            status->ep_datatype = request->ep_datatype; 
     69 
     70            //int count; 
     71            //MPI_Get_count(status, request->ep_datatype, &count); 
     72            //check_sum_recv(request->buf, count, request->ep_datatype, request->ep_src, request->ep_tag, request->comm, 2); 
     73        } 
     74        return MPI_SUCCESS; 
     75    } 
    8076 
    8177 
    82         int MPI_Waitall(int count, MPI_Request *array_of_requests, MPI_Status *array_of_statuses) 
    83         { 
    84     int dest_rank; 
    85     MPI_Comm_rank(MPI_COMM_WORLD, &dest_rank); 
    86      
    87     int finished = 0; 
    88     bool finished_index[count]; 
    8978 
    90           for(int i=0; i<count; i++) 
     79 
     80 
     81 
     82    int MPI_Waitall(int count, MPI_Request *array_of_requests, MPI_Status *array_of_statuses) 
    9183    { 
    92       finished_index[i] = false; 
     84        int dest_rank; 
     85        MPI_Comm_rank(MPI_COMM_WORLD, &dest_rank); 
     86 
     87        int finished = 0; 
     88        bool finished_index[count]; 
     89 
     90        for(int i=0; i<count; i++) 
     91        { 
     92            finished_index[i] = false; 
     93        } 
     94 
     95        while(finished < count) 
     96        { 
     97            for(int i=0; i<count; i++) 
     98            { 
     99                if(finished_index[i] == false) // this request has not been tested. 
     100                { 
     101                    if(array_of_requests[i].type != 2) // isend or imrecv 
     102                    {       
     103                        MPI_Wait(&array_of_requests[i], &array_of_statuses[i]); 
     104                        if(array_of_requests[i].type == 3) 
     105                        { 
     106                            //int check_count; 
     107                            //MPI_Get_count(&array_of_statuses[i], array_of_requests[i].ep_datatype, &check_count); 
     108                            //check_sum_recv(array_of_requests[i].buf, count, array_of_requests[i].ep_datatype, array_of_requests[i].ep_src, array_of_requests[i].ep_tag, array_of_requests[i].comm, 2); 
     109                        } 
     110                        finished++; 
     111                        finished_index[i] = true; 
     112                    } 
     113                    else // irecv 
     114                    { 
     115                        int flag = false; 
     116                        MPI_Message message; 
     117 
     118                        MPI_Improbe(array_of_requests[i].ep_src, array_of_requests[i].ep_tag, array_of_requests[i].comm, &flag, &message, &array_of_statuses[i]); 
     119 
     120                        if(flag) 
     121                        { 
     122                            //printf("dest_rank = %d, Waiting one message with src = %d, tag = %d, buf = %p\n", dest_rank, array_of_requests[i].ep_src, array_of_requests[i].ep_tag, array_of_requests[i].buf); 
     123                            int recv_count; 
     124                            MPI_Get_count(&array_of_statuses[i], array_of_requests[i].ep_datatype, &recv_count); 
     125                            MPI_Mrecv(array_of_requests[i].buf, recv_count, array_of_requests[i].ep_datatype, &message, &array_of_statuses[i]); 
     126                            //check_sum_recv(array_of_requests[i].buf, recv_count, array_of_requests[i].ep_datatype, array_of_requests[i].ep_src, array_of_requests[i].ep_tag, array_of_requests[i].comm, 2); 
     127 
     128                            finished++; 
     129                            finished_index[i] = true; 
     130                        } 
     131                    } 
     132                } 
     133            }     
     134        } 
     135        return MPI_SUCCESS; 
    93136    } 
    94  
    95     while(finished < count) 
    96     { 
    97       for(int i=0; i<count; i++) 
    98       { 
    99         if(finished_index[i] == false) // this request has not been tested. 
    100         { 
    101           if(array_of_requests[i].type != 2) // isend or imrecv 
    102           {       
    103             MPI_Wait(&array_of_requests[i], &array_of_statuses[i]); 
    104             if(array_of_requests[i].type == 3) 
    105             { 
    106               //int check_count; 
    107               //MPI_Get_count(&array_of_statuses[i], array_of_requests[i].ep_datatype, &check_count); 
    108               //check_sum_recv(array_of_requests[i].buf, count, array_of_requests[i].ep_datatype, array_of_requests[i].ep_src, array_of_requests[i].ep_tag, array_of_requests[i].comm, 2); 
    109             } 
    110             finished++; 
    111             finished_index[i] = true; 
    112           } 
    113           else // irecv 
    114           { 
    115             int flag = false; 
    116             MPI_Message message; 
    117              
    118             MPI_Improbe(array_of_requests[i].ep_src, array_of_requests[i].ep_tag, array_of_requests[i].comm, &flag, &message, &array_of_statuses[i]); 
    119              
    120             if(flag) 
    121             { 
    122               //printf("dest_rank = %d, Waiting one message with src = %d, tag = %d, buf = %p\n", dest_rank, array_of_requests[i].ep_src, array_of_requests[i].ep_tag, array_of_requests[i].buf); 
    123               int recv_count; 
    124               MPI_Get_count(&array_of_statuses[i], array_of_requests[i].ep_datatype, &recv_count); 
    125               MPI_Mrecv(array_of_requests[i].buf, recv_count, array_of_requests[i].ep_datatype, &message, &array_of_statuses[i]); 
    126               //check_sum_recv(array_of_requests[i].buf, recv_count, array_of_requests[i].ep_datatype, array_of_requests[i].ep_src, array_of_requests[i].ep_tag, array_of_requests[i].comm, 2); 
    127  
    128               finished++; 
    129               finished_index[i] = true; 
    130             } 
    131           } 
    132         } 
    133       }     
    134     } 
    135           return MPI_SUCCESS; 
    136         } 
    137137 
    138138 
Note: See TracChangeset for help on using the changeset viewer.