如何使用boost mpi传递用户定义的结构

时间:2010-04-21 17:35:48

标签: boost mpi

我正在尝试使用boost :: mpi :: send()调用发送一个名为ABC的用户定义结构。

给定的struct包含一个向量“data”,其大小在运行时确定。 struct ABC的对象由master发送给slave。 但是从属设备需要知道向量“数据”的大小,以便从站上有足够的缓冲区来接收这些数据。 我可以通过首先发送大小并在接收结构ABC的对象之前在从属设备上初始化足够的缓冲区来解决它。但这违背了使用STL容器的全部目的。

有谁知道更好的方法来处理这个问题?非常感谢任何建议。

以下是描述我的程序意图的示例代码。由于上述原因,此代码在运行时失败。

struct ABC
{
 double cur_stock_price;
 double strike_price;
 double risk_free_rate;
 double option_price;
 std::vector <char> data;
};

namespace boost
{
 namespace serialization
 {
   template<class Archive>
   void serialize (Archive &ar,
                   struct ABC &abc,
                   unsigned int version)
   {
     ar & abc.cur_stock_price;
     ar & abc.strike_price;
     ar & abc.risk_free_rate;
     ar & abc.option_price;
     ar & bopr.data;
   }
 }
}

BOOST_IS_MPI_DATATYPE (ABC);

int main(int argc, char* argv[])
{
 mpi::environment env (argc, argv);
 mpi::communicator world;

 if (world.rank () == 0)
 {
   ABC abc_obj;
  abc.cur_stock_price = 1.0;
  abc.strike_price = 5.0;
  abc.risk_free_rate = 2.5;
  abc.option_price = 3.0;
  abc_obj.data.push_back ('a');
  abc_obj.data.push_back ('b');

   world.send ( 1, ANY_TAG, abc_obj;);
   std::cout << "Rank 0 OK!" << std::endl;
 }

else if (world.rank () == 1)
{
   ABC abc_obj;

  // Fails here because abc_obj is not big enough
   world.recv (0,ANY_TAG, abc_obj;);
   std::cout << "Rank 1 OK!" << std::endl;

   for (int i = 0; i < abc_obj;.data.size(); i++)
     std::cout << i << "=" << abc_obj.data[i] << std::endl;
 }

 MPI_Finalize();
 return 0;
}

2 个答案:

答案 0 :(得分:3)

你不应该在消息中发送vector对象本身,因为接收者只需要它的内容,并且vector的内部状态可能在接收之后都被搞砸了(内存)在发送端有效的地址可能在接收端无效。

相反,这就是你需要做的事情:

  1. 使用四个双精度数组和一个简单的char数组定义一个单独的结构。
  2. 每当您需要发送时,请创建此新结构的临时变量,并使用您要发送的vector的内容填充char数组。
  3. 定义与此特定struct对象的大小匹配的临时MPI数据类型。
  4. 将临时结构作为此临时数据类型的实例发送。
  5. 在适当大的接收缓冲区中接收(或提前发送大小)
  6. 根据char数组的内容重建ABC结构。
  7. 至少,这就是你需要用vanilla C ++和MPI做的事情。我不熟悉boost,所以我不知道它是否能让这些步骤变得更容易。

答案 1 :(得分:0)

BOOST_IS_MPI_DATATYPE仅适用于固定长度数据。你的数组不是固定的长度,这就是它失败的原因。