source: XIOS/dev/branch_openmp/extern/ep_dev/ep_probe.cpp @ 1500

Last change on this file since 1500 was 1500, checked in by yushan, 3 years ago

save dev

File size: 4.7 KB
Line 
1#include "ep_lib.hpp"
2#include <mpi.h>
3#include "ep_declaration.hpp"
4#include "ep_mpi.hpp"
5
6namespace ep_lib
7{
8
9  int MPI_Iprobe(int src, int tag, MPI_Comm comm, int *flag, MPI_Status *status)
10  {
11    *flag = false;
12
13    if(!comm->is_ep)
14    {
15      ::MPI_Status *mpi_status = static_cast< ::MPI_Status* >(status->mpi_status);
16      ::MPI_Iprobe(src<0? MPI_ANY_SOURCE : src, tag<0? MPI_ANY_TAG: tag, to_mpi_comm(comm->mpi_comm), flag, mpi_status);
17
18      status->mpi_status = mpi_status;
19      status->ep_src = src;
20      status->ep_tag = tag;
21      return 0;
22    }
23
24    Debug("calling MPI_Iprobe EP\n");
25
26    Message_Check(comm);
27
28    #pragma omp flush
29
30    #pragma omp critical (_query)
31    if(!comm->ep_comm_ptr->message_queue->empty())
32    {
33      for(Message_list::iterator it = comm->ep_comm_ptr->message_queue->begin(); it!= comm->ep_comm_ptr->message_queue->end(); ++it)
34      {
35        bool src_matched = src<0? true: (*it)->ep_src == src;
36        bool tag_matched = tag<0? true: (*it)->ep_tag == tag;
37       
38        if(src_matched && tag_matched)       
39        {
40          Debug("find message\n");
41          *flag = true;
42
43          ::MPI_Status mpi_status = *(static_cast< ::MPI_Status *>((*it)->mpi_status));
44
45          status->mpi_status = new ::MPI_Status(mpi_status);
46          status->ep_src = (*it)->ep_src;
47          status->ep_tag = (*it)->ep_tag;
48
49          break;
50        }
51
52      }
53    }
54
55    return 0;
56  }
57
58
59
60  int MPI_Improbe(int src, int tag, MPI_Comm comm, int *flag, MPI_Message *message, MPI_Status *status)
61  {
62    *flag = false;
63    if(!comm->is_ep)
64    {
65      Debug("calling MPI_Improbe MPI\n");
66
67      ::MPI_Status mpi_status;
68      ::MPI_Message mpi_message;
69
70      #ifdef _openmpi
71      #pragma omp critical (_mpi_call)
72      {
73        ::MPI_Iprobe(src<0? MPI_ANY_SOURCE : src, tag<0? MPI_ANY_TAG: tag, to_mpi_comm(comm->mpi_comm), flag, &mpi_status);
74        if(*flag)
75        {
76          ::MPI_Mprobe(src<0? MPI_ANY_SOURCE : src, tag<0? MPI_ANY_TAG: tag, to_mpi_comm(comm->mpi_comm), &mpi_message, &mpi_status);
77        }
78      }
79      #elif _intelmpi
80        ::MPI_Improbe(src<0? MPI_ANY_SOURCE : src, tag<0? MPI_ANY_TAG: tag, to_mpi_comm(comm->mpi_comm), flag, &mpi_message, &mpi_status);
81      #endif
82       
83      status->mpi_status = &mpi_status;
84      status->ep_src = src;
85      status->ep_tag = tag;
86
87      (*message)->mpi_message = &message;
88      (*message)->ep_src = src;
89      (*message)->ep_tag = tag;
90     
91     
92      return 0;
93    }
94
95   
96
97    #pragma omp flush
98
99    #pragma omp critical (_query)
100    if(! comm->ep_comm_ptr->message_queue->empty())
101    {
102      for(Message_list::iterator it = comm->ep_comm_ptr->message_queue->begin(); it!= comm->ep_comm_ptr->message_queue->end(); ++it)
103      {
104
105        //printf("in ep_lib::Improbe, it->mpi_message = %p, it->mpi_status = %p, it->ep_src = %d, it->ep_tag = %d\n",
106        //                            (*(*it))->mpi_message,(*(*it))->mpi_status,(*(*it))->ep_src,(*(*it))->ep_tag);
107                                         
108        bool src_matched = src<0? true: (*it)->ep_src == src;
109        bool tag_matched = tag<0? true: (*it)->ep_tag == tag;
110       
111        if(src_matched && tag_matched)
112        {
113          *flag = true;
114
115          status->mpi_status = (*it)->mpi_status;
116          status->ep_src = (*it)->ep_src;
117          status->ep_tag = (*it)->ep_tag;
118
119          (*message)->mpi_message = (*it)->mpi_message;
120          (*message)->ep_src = (*it)->ep_src;
121          (*message)->ep_tag = (*it)->ep_tag;
122         
123          int test_count;
124          ::MPI_Get_count(to_mpi_status_ptr(status), 1275070475, &test_count);
125          printf("status = %p, test_count = %d\n", to_mpi_status_ptr(status), test_count);
126         
127         
128          ::MPI_Get_count(static_cast< ::MPI_Status* >(comm->ep_comm_ptr->message_queue->back()->mpi_status), 1275070475, &test_count);
129          printf("in ep_lib::Improbe, status = %p, test_count = %d\n", static_cast< ::MPI_Status* >(comm->ep_comm_ptr->message_queue->back()->mpi_status), test_count);
130                                     
131
132          #pragma omp critical (_query2)
133          {             
134            printf("delete %p : in ep_lib::Message_Check, delete (*it)->mpi_message\n", (*it)->mpi_message);
135            printf("delete %p : in ep_lib::Message_Check, delete (*it)->mpi_status\n", (*it)->mpi_status);
136            printf("delete %p : in ep_lib::Message_Check, delete (*it)\n", (*it));
137           
138            delete (*it)->mpi_message;     
139            delete (*it)->mpi_status; 
140            delete *it;
141           
142            comm->ep_comm_ptr->message_queue->erase(it);
143            printf("message queue siez = %lu\n", comm->ep_comm_ptr->message_queue->size());
144            #pragma omp flush
145          }
146         
147          break;
148        }
149
150      }
151    }
152  }
153
154}
155
Note: See TracBrowser for help on using the repository browser.