73ef4ff3
 Hu Chunming
提交三方库
|
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
|
// Copyright AlainMiniussi 20014 - 20015.
// Distributed under the Boost Software License, Version 1.0.
// (See accompanying file LICENSE_1_0.txt or copy at
// http://www.boost.org/LICENSE_1_0.txt)
#include <cstdlib>
#include "debugger.hpp"
std::vector<int> extract_paused_ranks(int argc, char** argv) {
std::vector<int> paused;
for (int i=1; i < argc; ++i) {
paused.push_back(std::atoi(argv[i]));
}
return paused;
}
void wait_for_debugger(std::vector<int> const& processes, boost::mpi::communicator const& comm)
{
int i = 1;
bool waiting = std::find(processes.begin(), processes.end(), comm.rank()) != processes.end();
for (int r = 0; r < comm.size(); ++r) {
if (comm.rank() == r) {
std::cout << "Rank " << comm.rank() << " has PID " << getpid();
if (waiting) {
std::cout << " and is waiting.";
}
std::cout << std::endl;
}
comm.barrier();
}
if (std::find(processes.begin(), processes.end(), comm.rank()) != processes.end()) {
while (i != 0) {
sleep(5);
}
}
std::cout << "Rank " << comm.rank() << " will proceed.\n";
}
void wait_for_debugger(boost::mpi::communicator const& comm)
{
std::vector<int> all;
for (int r = 0; r < comm.size(); ++r) {
all.push_back(r);
}
wait_for_debugger(all, comm);
}
|
