python mpi简单应用

最新推荐文章于 2024-07-04 10:16:12 发布

原创

最新推荐文章于 2024-07-04 10:16:12 发布 · 734 阅读

3 ·

CC 4.0 BY-SA版权

这篇博客介绍了如何在Python中使用MPI进行并行计算，通过一个简单的求π值程序展示基本语法。注意在Python中，C语言MPI的wtime函数不可用，需要关注值传递和主进程的正确结束。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

借鉴博客学习一下py下mpi的基本语法

https://blog.csdn.net/u012526003/article/details/80299113

发送

comm.send( data, pid)

接受

data_recv = comm.recv(data)

广播：

comm.bcast(data, 0) (None, 0 )

散播：均分

comm_scatter(data, root)

收集数据：

comm_gather(data, pid)

规约：

comm_reduce(data , pid , op=MPI.SUM)

下面让我们来写一个最简单的入门程序——求π值

没找到mpi的函数库，原来c的mpi里面的wtime函数在Python中没有，以此注意n的传值，并且保证主进程最后一个结束输出pi

from mpi4py import MPI

def func(x):
   return (4.0 / (1.0 + x * x))

comm = MPI.COMM_WORLD
comm_rank = comm.Get_rank()
comm_size = comm.Get_size()

PI = 3.141592653589
num = 1
if comm_rank == 0:
   num = 10000
#   starttime = MPI.wtime()
n = comm.bcast(num if comm_rank == 0 else None, root=0)
print("rank %d, n = %d" % (comm_rank, n))

h = 1.0/n
sum = 0.0
for i in range(comm_rank, n, comm_size):
   x = h * (0.5 + i