source: XIOS/dev/branch_openmp/extern/ep_dev/ep_free.cpp @ 1515

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

save dev

File size: 4.3 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_Comm_free(MPI_Comm *comm)
10  {
11    if(! (*comm)->is_ep)
12    {
13      if((*comm)->mpi_comm != static_cast< ::MPI_Comm*>(MPI_COMM_NULL->mpi_comm))
14      {
15        ::MPI_Comm mpi_comm = to_mpi_comm((*comm)->mpi_comm);
16
17        ::MPI_Comm_free(&mpi_comm);
18        Debug("comm is MPI, freed\n");
19      }
20      return 0;
21    }
22
23    else if((*comm)->is_intercomm)
24    {
25      return MPI_Comm_free_intercomm(comm);
26    }
27
28    else
29    {
30      int ep_rank_loc, num_ep;
31
32      ep_rank_loc = (*comm)->ep_comm_ptr->size_rank_info[1].first;
33      num_ep = (*comm)->ep_comm_ptr->size_rank_info[1].second;
34
35      MPI_Barrier(*comm);
36
37      if(ep_rank_loc == 0)
38      {
39        Debug("comm is EP, mpi_comm_ptr != NULL\n");
40
41        delete (*comm)->my_buffer;
42
43
44        if((*comm)->ep_barrier != NULL)
45        {
46          (*comm)->ep_barrier->~ep_barrier();
47          Debug("ep_barrier freed\n");
48        }
49
50
51        if( ! (*comm)->ep_rank_map->empty() )
52        {
53          (*comm)->ep_rank_map->clear();
54          delete (*comm)->ep_rank_map;
55          Debug("ep_rank_map emptied and freed\n");
56        }
57
58        for(int i=0; i<num_ep; i++)
59        {
60          (*comm)->ep_comm_ptr->comm_list[i]->ep_comm_ptr->message_queue->clear();
61          delete (*comm)->ep_comm_ptr->comm_list[i]->ep_comm_ptr->message_queue;
62          Debug("message queue freed\n");
63         
64
65          if((*comm)->ep_comm_ptr->comm_list[i]->ep_comm_ptr != NULL)
66          {
67            delete (*comm)->ep_comm_ptr->comm_list[i]->ep_comm_ptr;
68            Debug("ep_comm_ptr freed\n");
69          }
70        }
71
72        if( to_mpi_comm((*comm)->mpi_comm) != *static_cast< ::MPI_Comm*>(MPI_COMM_NULL->mpi_comm) 
73          && to_mpi_comm((*comm)->mpi_comm) != *static_cast< ::MPI_Comm*>(MPI_COMM_WORLD->mpi_comm))
74        {
75          ::MPI_Comm mpi_comm = to_mpi_comm((*comm)->mpi_comm);
76          ::MPI_Comm_free(&mpi_comm);
77          Debug("mpi_comm freed\n");
78        }
79
80       //if(comm != NULL) {delete[] comm->ep_comm_ptr->comm_list; Debug("comm freed\n");}
81
82      }
83
84      return 0;
85    }
86
87
88
89  }
90
91  int MPI_Comm_free_intercomm(MPI_Comm *comm)
92  {
93    int ep_rank_loc, num_ep;
94    //MPI_Barrier(*comm);
95
96    return 0;
97
98    ep_rank_loc = (*comm)->ep_comm_ptr->size_rank_info[1].first;
99    num_ep = (*comm)->ep_comm_ptr->size_rank_info[1].second;
100
101
102
103    //MPI_Barrier(*comm);
104
105    if(ep_rank_loc == 0)
106    {
107      Debug("comm is EP, mpi_comm_ptr != NULL\n");
108
109
110      if((*comm)->ep_barrier != NULL)
111      {
112        (*comm)->ep_barrier->~ep_barrier();
113        Debug("ep_barrier freed\n");
114      }
115
116
117      if( ! (*comm)->ep_rank_map->empty() )
118      {
119        (*comm)->ep_rank_map->clear();
120        delete (*comm)->ep_rank_map;
121        Debug("ep_rank_map emptied and freed\n");
122      }
123
124      for(int i=0; i<num_ep; i++)
125      {
126        (*comm)->ep_comm_ptr->comm_list[i]->ep_comm_ptr->message_queue->clear();
127        Debug("message queue freed\n");
128
129        #pragma omp critical (memory_free)
130        if((*comm)->ep_comm_ptr->comm_list[i]->ep_comm_ptr->intercomm != NULL)
131        {
132          (*comm)->ep_comm_ptr->comm_list[i]->ep_comm_ptr->intercomm->local_rank_map->clear();
133          (*comm)->ep_comm_ptr->comm_list[i]->ep_comm_ptr->intercomm->remote_rank_map->clear();
134          (*comm)->ep_comm_ptr->comm_list[i]->ep_comm_ptr->intercomm->intercomm_rank_map->clear();
135          Debug("intercomm local/remote/intercomm_rank_map emptied\n");
136        } 
137
138        if((*comm)->ep_comm_ptr->comm_list[i]->ep_comm_ptr != NULL)
139        {
140          delete (*comm)->ep_comm_ptr->comm_list[i]->ep_comm_ptr;
141          Debug("ep_comm_ptr freed\n");
142        }
143      }
144
145      if((*comm)->mpi_comm != static_cast< ::MPI_Comm*>(MPI_COMM_NULL->mpi_comm))
146      {
147        ::MPI_Comm mpi_comm = to_mpi_comm((*comm)->mpi_comm);
148        ::MPI_Comm_free(&mpi_comm);
149        Debug("mpi_comm freed\n");
150      }
151
152      if((*comm)->ep_comm_ptr->intercomm->mpi_inter_comm != static_cast< ::MPI_Comm*>(MPI_COMM_NULL->mpi_comm))
153      {
154        ::MPI_Comm mpi_comm = to_mpi_comm((*comm)->ep_comm_ptr->comm_list[0]->ep_comm_ptr->intercomm->mpi_inter_comm);
155        ::MPI_Comm_free(&mpi_comm);
156        Debug("mpi_intercomm freed\n");
157      }
158
159     if(comm != NULL) {delete[] (*comm)->ep_comm_ptr->comm_list; Debug("comm freed\n");}
160
161    }
162
163   
164   
165    return 0;
166  }
167
168
169
170}
171
172
Note: See TracBrowser for help on using the repository browser.