python mpi简单应用

最新推荐文章于 2025-04-15 20:44:12 发布

DS_HY

最新推荐文章于 2025-04-15 20:44:12 发布

阅读量716

点赞数

分类专栏：并行计算 python

本文链接：https://blog.csdn.net/sinat_36215255/article/details/103480732

版权

这篇博客介绍了如何在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