mpi4py - 分散和聚集循环的死锁

Question

嗨，我正在使用 mpi4py 来玩 MPI。 我的用例是我有一个 Python 队列 object 包含要处理的任务，如下所示：

from queue import Queue
my_queue = Queue()
my_queue.put({'task': [1, 2, 3]})
# while True:
if comm.rank == 0:
  task = my_queue.get()
else:
  task = None
work = comm.scatter(task, root=0)
calc = do_calculation(work) # whatever calculation
result = comm.gather(calc, root=0)
if comm.rank == 0:
  print(result)

这工作得很好，如果我继续追加任务（所以一开始我将 2 个任务放入队列并复制上面的代码）它也可以工作。 我现在的目标是让这段代码在无限循环中运行，并且无论何时将某些内容放入队列（即使用单独的线程等），它都应该被处理（my_queue.get() 应该是阻塞的，所以这不是问题）。

但是当我尝试将这段代码包装在一个无限循环while True: （见评论）该程序不会产生任何 output 并且只是锁定（似乎是死锁）。

Answer 1

我找到了我的问题的答案。 没有僵局。 代码运行完美，唯一的问题是标准输出的缓冲区没有被刷新。 这导致结果没有打印到控制台，我假设代码产生了死锁。

添加行sys.stdout.flush()以手动刷新缓冲区会将结果打印到控制台并解决我的问题。