mpi学习日志(14):mpi4py与probe

今天我们来学习一个颇重要的函数,probe.

probe的中文意思是"探查",那么我们可能已经猜出这个函数的作用了,probe函数是用来探查当前进程是否收到消息的.

probe函数也有大小写版本和阻塞非阻塞版本,合起来就有4个探查函数了.

不过,貌似大小写版本在使用是一样的,只是实现上不一样.(或者是我还未发现效果上的差别)

而阻塞和非阻塞版本则有差别,阻塞版本的probe必须在收到消息后才返回,返回一个true.而非阻塞版本的iprobe则是马上返回一个true或者false,true表示有消息,false则是没有消息.

无论是哪个版本的probe函数,它的参数表都是一样的.

第一个参数是source,用来限制发送方,默认是all.

第二个参数是tag,用来限制tag.我们好像未讨论过tag,其实在之前的许多函数里都有tag参数,它是int类型,用来给消息打标签,来区分不同的消息.默认也是all.

第三个参数是status,它是一个output参数,就是说你给它传递一个空的status对象,函数返回后这个status对象会存放一些返回信息,例如source和tag.默认是none.

(status对象还有很多的属性,以后可能会介绍)

老实说,一开始我没有想到这个probe函数有什么意义,它大概是实现一些很高级的同步问题.

后来我发现它能实现一些有趣的功能,比如上一篇说道,mpi4py的irecv并不能很好地实现,而现在我们可以用iprobe和recv来实现irecv.

实现irecv有什么好处,好处就是你又可以随意地发送数据了,比如发送一个字典dict,发送一个混杂着字符串和数字的列表list.

代码:

运行结果:

代码解释:

这个代码的样子和上一篇的非阻塞Irecv是几乎一样的,并且我们在这里不需要被buffer限制,可以随心所欲.

不过有一点要注意的是,probe函数只是针对点对点通信的探查,它并不能用来探查广播之类的多点通信.

要想非阻塞地广播,还是老老实实使用Ibcast.