c++ - 避免在 C++ MPI 中为 cout 重复 if 语句

Question

我正在用 MPI(C 接口(interface)，无提升等)编写一个 C++ 程序。在我的程序中，我有很多输出，无论是 cout 还是文件，都只在等级 0 上完成。是否有避免的好方法 写作 if (rank == 0) cout << string ？

有两种方法可以在我的计算机上使用我的 MPI 实现，但对我来说这两种方法都不稳定:

if (rank != 0)
  cout.setstate(ios_base::badbit);

这有效地禁用了除 0 以外的所有等级的输出，但它被允许吗？这样做会不会有问题？

另一个想法是创建一个未打开的 ofstream，并将输出重定向到那里。

ostream* os;
ofstream nullstream;
if (rank == 0)
  os = &cout;
else
  os = &nullstream;
*os << "rank " << rank << endl;

这使 nullstream 处于错误状态，但实际上也禁用了所有非 0 的等级上的输出...

这个问题对我来说似乎很常见，所以如果它已经在其他地方得到回答，我很抱歉。我没有通过搜索找到答案，很高兴能重定向到现有问题。

Answer 1

如果您愿意牺牲 Windows 系统的可移植性，在 POSIX 系统上静音输出的正确方法是将标准输出(以及可选的标准错误)重定向到 /dev/null:

int main() {
   ...
   std::ofstream sink("/dev/null");

   if (rank != 0) {
     // Mute standard output
     std::cout.rdbuf(sink.rdbuf());
     // Optionally mute standard error
     std::cerr.rdbuf(sink.rdbuf());      
   }

   no_output_from_other_ranks_from_now_on();
   ...
}

改编自this answer .