在上一篇中我們對非阻塞通信做了一個非常概要的介紹介汹,下面我們將依次介紹非重復(fù)的非阻塞通信的四種通信模式却嗡,并給出相應(yīng)的例程。讓我們首先從非重復(fù)的標準通信開始嘹承。
非重復(fù)非阻塞的標準通信模式是與阻塞的標準通信模式相對應(yīng)的窗价,其通信方法(MPI.Comm 類的方法)接口有一個前綴修飾 I/i,如下:
isend(self, obj, int dest, int tag=0)
irecv(self, buf=None, int source=ANY_SOURCE, int tag=ANY_TAG)
Isend(self, buf, int dest, int tag=0)
Irecv(self, buf, int source=ANY_SOURCE, int tag=ANY_TAG)
這些方法調(diào)用中的參數(shù)是與阻塞標準通信模式的方法調(diào)用參數(shù)一樣的赶撰。
需要注意的是舌镶,上面雖然給出的是非阻塞的發(fā)送和非阻塞的接收方法柱彻,但非阻塞發(fā)送可與阻塞接收相匹配,反之餐胀,阻塞發(fā)送也可與非阻塞接收相匹配哟楷。
非阻塞的發(fā)送和接收都會返回一個 MPI.Request 對象,我們需要用到它的一些方法來等待否灾、測試或是取消本次通信卖擅。具體來說,我們經(jīng)常會用到的 MPI.Request 對象方法有:
test(self, Status status=None)
testall(type cls, requests, statuses=None)
testany(type cls, requests, Status status=None)
wait(self, Status status=None)
waitall(type cls, requests, statuses=None)
waitany(type cls, requests, Status status=None)
Test(self, Status status=None)
Testall(type cls, requests, statuses=None)
Testany(type cls, requests, Status status=None)
Testsome(type cls, requests, statuses=None)
Wait(self, Status status=None)
Waitall(type cls, requests, statuses=None)
Waitany(type cls, requests, Status status=None)
Waitsome(type cls, requests, statuses=None)
Cancel(self)
其中以小寫字母開頭的 test墨技,wait 方法通常與以 i 開頭的通信方法配合使用惩阶,而以大寫字母開頭的 Test, Wait 方法通常與以 I 開頭的通信方法配合使用扣汪。這是因為 test 會返回一個二元 tuple断楷,其第一個元素是一個 bool 值, True 表示操作完成崭别,F(xiàn)alse 表示尚未完成冬筒,第二個元素為 None 或者為接收操作完成后所接收到的消息,wait 會返回 None 或者所接收到的消息茅主。而 Test 和 Wait 方法則只會返回 True 或 False 以表明是否完成了相關(guān)操作舞痰,而不會返回所接收到消息。
下面給出非重復(fù)非阻塞標準通信的使用例程:
# isend_irecv.py
from mpi4py import MPI
comm = MPI.COMM_WORLD
rank = comm.Get_rank()
send_obj = {'a': [1, 2.4, 'abc', -2.3+3.4J],
'b': {2, 3, 4}}
if rank == 0:
send_req = comm.isend(send_obj, dest=1, tag=11)
send_req.wait()
print 'process %d sends %s' % (rank, send_obj)
elif rank == 1:
recv_req = comm.irecv(source=0, tag=11)
recv_obj = recv_req.wait()
print 'process %d receives %s' % (rank, recv_obj)
運行結(jié)果如下:
$ mpiexec -n 2 python isend_irecv.py
process 0 sends {'a': [1, 2.4, 'abc', (-2.3+3.4j)], 'b': set([2, 3, 4])}
process 1 receives {'a': [1, 2.4, 'abc', (-2.3+3.4j)], 'b': set([2, 3, 4])}
# Isend_Irecv.py
import numpy as np
from mpi4py import MPI
comm = MPI.COMM_WORLD
rank = comm.Get_rank()
count = 10
send_buf = np.arange(count, dtype='i')
recv_buf = np.empty(count, dtype='i')
if rank == 0:
send_req = comm.Isend(send_buf, dest=1, tag=11)
send_req.Wait()
print 'process %d sends %s' % (rank, send_buf)
elif rank == 1:
recv_req = comm.Irecv(recv_buf, source=0, tag=11)
recv_req.Wait()
print 'process %d receives %s' % (rank, recv_buf)
運行結(jié)果如下:
$ mpiexec -n 2 python Isend_Irecv.py
process 0 sends [0 1 2 3 4 5 6 7 8 9]
process 1 receives [0 1 2 3 4 5 6 7 8 9]
上面我們介紹了 mpi4py 中非重復(fù)非阻塞標準通信模式诀姚,在下一篇中我們將介紹非重復(fù)非阻塞緩沖通信响牛。