python mpi4py_python-使用mpi4py并行化函数调用

在您的代码中,根进程是唯一一个调用comm.gather()的进程,因为根进程是唯一一个调用并行成本函数的进程.因此,程序面临僵局.您已经很清楚这个问题.

为了克服这种僵局,其他进程必须根据最小化需求多次调用cost函数.由于此调用数量是未知的,因此while循环似乎适合这些过程.

将定义while循环的停止条件.该标志将从根进程广播到所有进程,因为根进程是唯一意识到minimal()函数结束的事实的人.广播必须在成本函数中执行,因为所有过程都必须在每次迭代中测试最小化函数的结尾.由于最小化使用了函数的返回值,因此标志为passed by reference via a mutable type

最后,这是您的问题的潜在解决方案.它由mpirun -np 4 python main.py运行.我使用fmin()而不是minimal(),因为我的scipy版本已过时.

#from scipy.optimize import minimize

from scipy.optimize import fmin

from mpi4py import MPI

import numpy as np

comm = MPI.COMM_WORLD

size = comm.Get_size()

rank = comm.Get_rank()

N = 100 # for testing

step = N//size # say that N is divisible by size

def parallel_function_caller(x,stopp):

stopp[0]=comm.bcast(stopp[0], root=0)

summ=0

if stopp[0]==0:

#your function here in parallel</