我在使用boost.mpi库进行自定义reduce操作编译时遇到了麻烦。我正在尝试按元素添加两个std::vector
。当我使用两个向量一起添加时,操作正常,但是当我尝试使用mpi::reduce
时,我得到错误“没有已知的从'int'转换为'const std :: vector'的第一个参数”。
#include <boost/mpi.hpp>
namespace mpi = boost::mpi;
template <typename T>
struct elementwise_add {
std::vector<T> operator()(const std::vector<T>& a, const std::vector<T>& b) const {
std::vector<T> result(a.size());
std::transform(a.begin(), a.end(), b.begin(), result.begin(), std::plus<T>());
return result;
}
};
int main() {
mpi::environment env;
mpi::communicator world;
int N = 10;
std::vector<int> a(N,1), b(N,1);
auto op = elementwise_add<int>();
auto c = op(a, b); // works fine
if (world.rank() == 0) {
std::vector<int> sum(N,0);
mpi::reduce(world, a, sum, op, 0); // errors
} else {
mpi::reduce(world, a, op, 0);
}
}
编译:
mpic++ -O3 -std=c++11 -I/usr/local/include/ -I. -lboost_mpi-mt -lboost_serialization-mt test.cc -o test
答案 0 :(得分:2)
使用boost 1.54编译你的代码很好。但是,在版本1.55中引入了两个reduce()
重载:
template<typename T, typename Op>
void
reduce(const communicator & comm, std::vector<T> const & in_values, Op op,
int root)
{
reduce(comm, &in_values.front(), in_values.size(), op, root);
}
template<typename T, typename Op>
void
reduce(const communicator & comm, std::vector<T> const & in_values,
std::vector<T> & out_values, Op op, int root)
{
out_values.resize(in_values.size());
reduce(comm, &in_values.front(), in_values.size(), &out_values.front(), op,
root);
}
这意味着当您调用reduce()
并提供std::vector
作为输入时(就像在代码中一样),将传递指向数据的指针。
请注意,如果reduce()
被赋予指针,它将默认按每个元素运行。因此,您不应传递用于组合向量的运算符,而应传递用于组合ints
的运算符(例如std::plus<int>()
)。
以下代码适合您(升级版&gt; = 1.55):
#include <boost/mpi.hpp>
namespace mpi = boost::mpi;
int main() {
mpi::environment env;
mpi::communicator world;
int N = 10;
std::vector<int> a(N,1);
if (world.rank() == 0) {
std::vector<int> sum(N,0);
mpi::reduce(world, a, sum, std::plus<int>(), 0);
} else {
mpi::reduce(world, a, std::plus<int>(), 0);
}
}
如果您希望您的代码也能使用早期版本的boost(&lt; 1.55),那么您必须自己将指针传递给数据。像这样:
if (world.rank() == 0) {
std::vector<int> sum(N,0);
mpi::reduce(world, &a.front(), N, &sum.front(), std::plus<int>(), 0);
} else {
mpi::reduce(world, &a.front(), N, std::plus<int>(), 0);
}