]> SALOME platform Git repositories - tools/libbatch.git/blob - src/Core/Batch_MpiImpl.cxx
Salome HOME
dcd68b0c99e22688a9be7448e66e85db2a229929
[tools/libbatch.git] / src / Core / Batch_MpiImpl.cxx
1 //  Copyright (C) 2007-2012  CEA/DEN, EDF R&D, OPEN CASCADE
2 //
3 //  Copyright (C) 2003-2007  OPEN CASCADE, EADS/CCR, LIP6, CEA/DEN,
4 //  CEDRAT, EDF R&D, LEG, PRINCIPIA R&D, BUREAU VERITAS
5 //
6 //  This library is free software; you can redistribute it and/or
7 //  modify it under the terms of the GNU Lesser General Public
8 //  License as published by the Free Software Foundation; either
9 //  version 2.1 of the License.
10 //
11 //  This library is distributed in the hope that it will be useful,
12 //  but WITHOUT ANY WARRANTY; without even the implied warranty of
13 //  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
14 //  Lesser General Public License for more details.
15 //
16 //  You should have received a copy of the GNU Lesser General Public
17 //  License along with this library; if not, write to the Free Software
18 //  Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307 USA
19 //
20 //  See http://www.salome-platform.org/ or email : webmaster.salome@opencascade.com
21 //
22
23 #include <iostream>
24 #include <sstream>
25 #include <string>
26 #include "Batch_MpiImpl.hxx"
27
28 using namespace std;
29
30 // Constructor
31 MpiImpl::MpiImpl()
32 {
33 }
34
35 // Destructor
36 MpiImpl::~MpiImpl()
37 {
38 }
39
40 // lam implementation
41 // Constructor
42 MpiImpl_LAM::MpiImpl_LAM() : MpiImpl()
43 {
44 }
45
46 // Destructor
47 MpiImpl_LAM::~MpiImpl_LAM()
48 {
49 }
50
51 string MpiImpl_LAM::size()
52 {
53   return "${LAMWORLD}";
54 }
55
56 string MpiImpl_LAM::rank()
57 {
58   return "${LAMRANK}";
59 }
60
61 string MpiImpl_LAM::boot(const string machinefile, const unsigned int nbnodes)
62 {
63   ostringstream oss;
64   oss << "lamboot " << machinefile << endl;
65   return oss.str();
66 }
67
68 string MpiImpl_LAM::run(const string machinefile, const unsigned int nbproc, const string fileNameToExecute)
69 {
70   ostringstream oss;
71   oss << "mpirun -np " << nbproc << " " << fileNameToExecute << endl;
72   return oss.str();
73 }
74
75 string MpiImpl_LAM::halt()
76 {
77   ostringstream oss;
78   oss << "lamhalt" << endl;
79   return oss.str();
80 }
81
82 string MpiImpl_LAM::name()
83 {
84   return "lam";
85 }
86
87 // mpich1 implementation
88 // Constructor
89 MpiImpl_MPICH1::MpiImpl_MPICH1() : MpiImpl()
90 {
91 }
92
93 // Destructor
94 MpiImpl_MPICH1::~MpiImpl_MPICH1()
95 {
96 }
97
98 string MpiImpl_MPICH1::size()
99 {
100   return "${MPIRUN_NPROCS}";
101 }
102
103 string MpiImpl_MPICH1::rank()
104 {
105   return "${MPIRUN_RANK}";
106 }
107
108 string MpiImpl_MPICH1::boot(const string machinefile, const unsigned int nbnodes)
109 {
110   return "";
111 }
112
113 string MpiImpl_MPICH1::run(const string machinefile, const unsigned int nbproc, const string fileNameToExecute)
114 {
115   ostringstream oss;
116   oss << "mpirun -machinefile " << machinefile << " -np " << nbproc << " " << fileNameToExecute << endl;
117   return oss.str();
118 }
119
120 string MpiImpl_MPICH1::halt()
121 {
122   return "";
123 }
124
125 string MpiImpl_MPICH1::name()
126 {
127   return "mpich";
128 }
129
130 // mpich2 implementation
131 // Constructor
132 MpiImpl_MPICH2::MpiImpl_MPICH2() : MpiImpl()
133 {
134 }
135
136 // Destructor
137 MpiImpl_MPICH2::~MpiImpl_MPICH2()
138 {
139 }
140
141 string MpiImpl_MPICH2::size()
142 {
143   return "${PMI_SIZE}";
144 }
145
146 string MpiImpl_MPICH2::rank()
147 {
148   return "${PMI_RANK}";
149 }
150
151 string MpiImpl_MPICH2::boot(const string machinefile, const unsigned int nbnodes)
152 {
153   ostringstream oss;
154   oss << "mpdboot" << " -n " << nbnodes;
155   if (machinefile!="")
156     oss  << " -f " << machinefile;
157   oss << endl;
158   return oss.str();
159 }
160
161 string MpiImpl_MPICH2::run(const string machinefile, const unsigned int nbproc, const string fileNameToExecute)
162 {
163   ostringstream oss;
164   oss << "mpirun -np " << nbproc << " " << fileNameToExecute << endl;
165   return oss.str();
166 }
167
168 string MpiImpl_MPICH2::halt()
169 {
170   ostringstream oss;
171   oss << "mpdallexit" << endl;
172   return oss.str();
173 }
174
175 string MpiImpl_MPICH2::name()
176 {
177   return "mpich";
178 }
179
180 // openmpi implementation
181 // Constructor
182 MpiImpl_OPENMPI::MpiImpl_OPENMPI() : MpiImpl()
183 {
184 }
185
186 // Destructor
187 MpiImpl_OPENMPI::~MpiImpl_OPENMPI()
188 {
189 }
190
191 string MpiImpl_OPENMPI::size()
192 {
193   return "${OMPI_MCA_ns_nds_num_procs}";
194 }
195
196 string MpiImpl_OPENMPI::rank()
197 {
198   return "${OMPI_MCA_ns_nds_vpid}";
199 }
200
201 string MpiImpl_OPENMPI::boot(const string machinefile, const unsigned int nbnodes)
202 {
203   return "";
204 }
205
206 string MpiImpl_OPENMPI::run(const string machinefile, const unsigned int nbproc, const string fileNameToExecute)
207 {
208   ostringstream oss;
209   oss << "mpirun -hostfile " << machinefile << " -np " << nbproc << " " << fileNameToExecute << endl;
210   return oss.str();
211 }
212
213 string MpiImpl_OPENMPI::halt()
214 {
215   return "";
216 }
217
218 string MpiImpl_OPENMPI::name()
219 {
220   return "openmpi";
221 }
222
223 // ompi implementation
224 // Constructor
225 MpiImpl_OMPI::MpiImpl_OMPI() : MpiImpl()
226 {
227 }
228
229 // Destructor
230 MpiImpl_OMPI::~MpiImpl_OMPI()
231 {
232 }
233
234 string MpiImpl_OMPI::size()
235 {
236   return "${OMPI_MCA_ns_nds_num_procs}";
237 }
238
239 string MpiImpl_OMPI::rank()
240 {
241   return "${OMPI_MCA_ns_nds_vpid}";
242 }
243
244 string MpiImpl_OMPI::boot(const string machinefile, const unsigned int nbnodes)
245 {
246   return "";
247 }
248
249 string MpiImpl_OMPI::run(const string machinefile, const unsigned int nbproc, const string fileNameToExecute)
250 {
251   ostringstream oss;
252   oss << "mpirun -hostfile " << machinefile << " -np " << nbproc << " " << fileNameToExecute << endl;
253   return oss.str();
254 }
255
256 string MpiImpl_OMPI::halt()
257 {
258   return "";
259 }
260
261 string MpiImpl_OMPI::name()
262 {
263   return "ompi";
264 }
265
266 // slurm implementation
267 // Constructor
268 MpiImpl_SLURM::MpiImpl_SLURM() : MpiImpl()
269 {
270 }
271
272 // Destructor
273 MpiImpl_SLURM::~MpiImpl_SLURM()
274 {
275 }
276
277 string MpiImpl_SLURM::size()
278 {
279   return "${SLURM_NPROCS}";
280 }
281
282 string MpiImpl_SLURM::rank()
283 {
284   return "${SLURM_PROCID}";
285 }
286
287 string MpiImpl_SLURM::boot(const string machinefile, const unsigned int nbnodes)
288 {
289   return "";
290 }
291
292 string MpiImpl_SLURM::run(const string machinefile, const unsigned int nbproc, const string fileNameToExecute)
293 {
294   ostringstream oss;
295   oss << "srun " << fileNameToExecute << endl;
296   return oss.str();
297 }
298
299 string MpiImpl_SLURM::halt()
300 {
301   return "";
302 }
303
304 string MpiImpl_SLURM::name()
305 {
306   return "slurm";
307 }
308
309 // prun implementation
310 // Constructor
311 MpiImpl_PRUN::MpiImpl_PRUN() : MpiImpl()
312 {
313 }
314
315 // Destructor
316 MpiImpl_PRUN::~MpiImpl_PRUN()
317 {
318 }
319
320 string MpiImpl_PRUN::size()
321 {
322   return "${RMS_NPROCS}";
323 }
324
325 string MpiImpl_PRUN::rank()
326 {
327   return "${RMS_RANK}";
328 }
329
330 string MpiImpl_PRUN::boot(const string machinefile, const unsigned int nbnodes)
331 {
332   return "";
333 }
334
335 string MpiImpl_PRUN::run(const string machinefile, const unsigned int nbproc, const string fileNameToExecute)
336 {
337   ostringstream oss;
338   oss << "prun -n " << nbproc << " " << "-p mpi " << fileNameToExecute << endl;
339   return oss.str();
340 }
341
342 string MpiImpl_PRUN::halt()
343 {
344   return "";
345 }
346
347 string MpiImpl_PRUN::name()
348 {
349   return "prun";
350 }