offsets.cpp
2.59 KB
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
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
// Copyright Alain Miniussi 2014.
// 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)
// Authors: Alain Miniussi
#include <boost/mpi/detail/offsets.hpp>
#include <boost/mpi/detail/antiques.hpp>
namespace boost { namespace mpi {
namespace detail {
// Convert a sequence of sizes [S0..Sn] to a sequence displacement
// [O0..On] where O[0] = 0 and O[k+1] = O[k]+S[k].
void
sizes2offsets(int const* sizes, int* offsets, int n)
{
offsets[0] = 0;
for(int i = 1; i < n; ++i) {
offsets[i] = offsets[i-1] + sizes[i-1];
}
}
// Convert a sequence of sizes [S0..Sn] to a sequence displacement
// [O0..On] where O[0] = 0 and O[k+1] = O[k]+S[k].
void
sizes2offsets(std::vector<int> const& sizes, std::vector<int>& offsets)
{
int sz = sizes.size();
offsets.resize(sz);
sizes2offsets(c_data(sizes), c_data(offsets), sz);
}
// Given a sequence of sizes (typically the number of records dispatched
// to each process in a scater) and a sequence of displacements (typically the
// slot index at with those record starts), convert the later to a number
// of skipped slots.
void
offsets2skipped(int const* sizes, int const* offsets, int* skipped, int n)
{
skipped[0] = 0;
for(int i = 1; i < n; ++i) {
skipped[i] -= offsets[i-1] + sizes[i-1];
}
}
// Reconstruct offsets from sizes assuming no padding.
// Only takes place if on the root process and if
// displs are not already provided.
// If memory was allocated, returns a pointer to it
// otherwise null.
int*
make_offsets(communicator const& comm, int const* sizes, int const* displs, int root)
{
if (root == -1 || root == comm.rank()) {
assert(sizes);
if (!displs) {
int nproc = comm.size();
int* offsets = new int[nproc];
displs = offsets;
sizes2offsets(sizes, offsets, nproc);
return offsets;
} else {
return 0;
}
} else {
return 0;
}
}
// Reconstruct skip slots from sizes and offsets.
// Only takes place if on the root process and if
// displs are provided.
// If memory was allocated, returns a pointer to it
// otherwise null.
int*
make_skipped_slots(communicator const& comm, int const* sizes, int const* displs, int root)
{
if (root == -1 || root == comm.rank()) {
assert(sizes);
if (displs) {
int nproc = comm.size();
int* skipped = new int[nproc];
std::copy(displs, displs+nproc, skipped);
offsets2skipped(sizes, displs, skipped, nproc);
return skipped;
} else {
return 0;
}
} else {
return 0;
}
}
}
}}