mpi4py 中的 Status 對象

上一篇中我們介紹了 mpi4py 中的 info 和 assertion轻纪,下面我們將介紹 mpi4py 中的 Status 對象轰驳。

在 MPI-1 中应役,Status 對象僅用于點到點通信,而 MPI-2 則在通用化請求以及并行 I/O 中使用 Status 對象显沈。因此新的 MPI.Status 對象在不同上下文環(huán)境下刁卜,可能存在有效域或無效域志电,特別是在通用化請求操作中,應(yīng)用程序需要在回調(diào)函數(shù)中設(shè)置某些狀態(tài)信息蛔趴。為管理泛化后的 MPI.Status 對象挑辆,MPI-2 新增加了下列函數(shù):MPI.Status.Set_elements 和 MPI.Status.Set_cancelled。

需要注意的是夺脾,Status 對象最好對應(yīng)每個操作單獨使用,而不推薦復(fù)用的方式茉继。此外咧叭,如果在通用化請求中需要傳遞額外信息,則不應(yīng)修改 Status 對象中各域的語義烁竭,而推薦借助 extra_status 參數(shù)傳遞信息菲茬。通用化請求將在下一篇中介紹。

下面給出 MPI.Status 對象的方法接口派撕。

MPI.Status.Get_source(self)

獲得并返回接收消息的源婉弹。也可以通過屬性 source 獲取。

MPI.Status.Set_source(self, int source)

設(shè)置接收消息的源终吼。也可以通過屬性 source 設(shè)置镀赌。

MPI.Status.Get_tag(self)

獲得并返回接收消息的 tag。也可以通過屬性 tag 獲取际跪。

MPI.Status.Set_tag(self, int tag)

設(shè)置接收消息的tag商佛。也可以通過屬性 tag 設(shè)置喉钢。

MPI.Status.Get_error(self)

獲得接收消息的錯誤。也可以通過屬性 error 獲取良姆。

MPI.Status.Set_error(self, int error)

設(shè)置接收消息的錯誤肠虽。也可以通過屬性 error 設(shè)置。

MPI.Status.Get_count(self, Datatype datatype=BYTE)

返回接收到的元素個數(shù)(以 datatype 為單位計算)玛追。如果 datatype 的 size 為 0税课,該方法也會返回 0;如果接收到的數(shù)據(jù)不是 datatype 的整數(shù)倍痊剖,則會返回 MPI.UNDEFINED韩玩。也可以通過屬性 count 獲取以 MPI.BYTE 為單位的計數(shù)。

MPI.Status.Get_elements(self, Datatype datatype)

返回 datatype 中的基本數(shù)據(jù)類型的個數(shù)邢笙。如果支持 MPI-3啸如,mpi4py 實際調(diào)用的是 MPI_Get_elements_x 以支持大的計數(shù)。

MPI.Status.Set_elements(self, Datatype datatype, Count count)

設(shè)置 datatype 中的基本數(shù)據(jù)類型的個數(shù)氮惯。如果支持 MPI-3叮雳,mpi4py 實際調(diào)用的是 MPI_Status_set_elements_x 以支持大的計數(shù)。

MPI.Status.Is_cancelled(self)

檢測某個通信請求是否被取消妇汗。也可用通過屬性 cancelled 來查詢帘不。

MPI.Status.Set_cancelled(self, bool flag)

將取消狀態(tài)設(shè)置為 flag注意:該方法只能用于通用化請求的回調(diào)函數(shù)中杨箭。

例程

下面給出使用例程寞焙。

# status.py

"""
Demonstrates the usage of MPI.Status.

Run this with 2 processes like:
$ mpiexec -n 2 python status.py
"""

import numpy as np
from mpi4py import MPI


comm = MPI.COMM_WORLD
rank = comm.rank

status = MPI.Status()
print 'status.source == MPI.ANY_SOURCE:', status.source == MPI.ANY_SOURCE
print 'status.tag == MPI.ANY_TAG:', status.tag == MPI.ANY_TAG
print 'status.error == MPI.SUCCESS:', status.error == MPI.SUCCESS
print 'MPI.UNDEFINED = %d' % MPI.UNDEFINED

comm.Barrier()

if rank == 0:
    send_buf = np.arange(10, dtype='i')
    # send 10 MPI.INTs, 40 MPI.BYTEs
    comm.Send(send_buf, dest=1, tag=0)

    # create a datatype and send it (multiple of sizeof(int))
    send_type = MPI.Datatype.Create_struct([1, 16], [0, 4], [MPI.INT, MPI.CHAR])
    send_type.Commit()
    # send 1 MPI.INT, 16 MPI.CHARs (= 4 MPI.INTs), 1 + 16 basic elements
    comm.Send([send_buf, 1, send_type], dest=1, tag=1)
    send_type.Free()

    # create a datatype and send it (not a multiple of sizeof(int))
    send_type = MPI.Datatype.Create_struct([1, 17], [0, 4], [MPI.INT, MPI.CHAR])
    send_type.Commit()
    # send 1 MPI.INT, 17 MPI.CHARs (not a multiple of MPI.INT), 1 + 17 basic elements
    comm.Send([send_buf, 1, send_type], dest=1, tag=2)
    send_type.Free()
elif rank == 1:
    recv_buf = np.full(10, -1, dtype='i')
    # receive 10 ints
    status = MPI.Status()
    comm.Recv(recv_buf, source=0, tag=0, status=status)
    print
    print 'Get count with MPI.INT:', status.Get_count(MPI.INT)
    print 'Get elements with MPI.INT:', status.Get_elements(MPI.INT)
    print 'Get count with MPI.BYTE:', status.Get_count(MPI.BYTE)
    print 'Get elements with MPI.BYTE:', status.Get_elements(MPI.BYTE)
    print

    # create a datatype and receive it
    recv_type = MPI.Datatype.Create_struct([1, 36], [0, 4], [MPI.INT, MPI.CHAR])
    recv_type.Commit()
    status = MPI.Status()
    comm.Recv([recv_buf, 1, recv_type], source=0, tag=1, status=status)
    print 'Get count with MPI.INT:', status.Get_count(MPI.INT)
    print 'Get elements with MPI.INT:', status.Get_elements(MPI.INT)
    print 'Get count with recv_type:', status.Get_count(recv_type)
    print 'Get elements with recv_type:', status.Get_elements(recv_type)
    print

    status = MPI.Status()
    comm.Recv([recv_buf, 1, recv_type], source=0, tag=2, status=status)
    print 'Get count with MPI.INT:', status.Get_count(MPI.INT)
    print 'Get elements with MPI.INT:', status.Get_elements(MPI.INT)
    print 'Get count with recv_type:', status.Get_count(recv_type)
    print 'Get elements with recv_type:', status.Get_elements(recv_type)

    recv_type.Free()

運行結(jié)果如下:

$ mpiexec -n 2 python status.py
status.source == MPI.ANY_SOURCE: True
status.tag == MPI.ANY_TAG: True
status.error == MPI.SUCCESS: True
MPI.UNDEFINED = -32766
status.source == MPI.ANY_SOURCE: True
status.tag == MPI.ANY_TAG: True
status.error == MPI.SUCCESS: True
MPI.UNDEFINED = -32766

Get count with MPI.INT: 10
Get elements with MPI.INT: 10
Get count with MPI.BYTE: 40
Get elements with MPI.BYTE: 40

Get count with MPI.INT: 5
Get elements with MPI.INT: 5
Get count with recv_type: -32766
Get elements with recv_type: 17

Get count with MPI.INT: -32766
Get elements with MPI.INT: -32766
Get count with recv_type: -32766
Get elements with recv_type: 18

以上介紹了 mpi4py 中的 Status 對象,在下一篇中我們將介紹 mpi4py 中的通用化請求互婿。

最后編輯于
?著作權(quán)歸作者所有,轉(zhuǎn)載或內(nèi)容合作請聯(lián)系作者
  • 序言:七十年代末捣郊,一起剝皮案震驚了整個濱河市,隨后出現(xiàn)的幾起案子慈参,更是在濱河造成了極大的恐慌呛牲,老刑警劉巖,帶你破解...
    沈念sama閱讀 218,607評論 6 507
  • 序言:濱河連續(xù)發(fā)生了三起死亡事件驮配,死亡現(xiàn)場離奇詭異娘扩,居然都是意外死亡,警方通過查閱死者的電腦和手機壮锻,發(fā)現(xiàn)死者居然都...
    沈念sama閱讀 93,239評論 3 395
  • 文/潘曉璐 我一進店門琐旁,熙熙樓的掌柜王于貴愁眉苦臉地迎上來,“玉大人猜绣,你說我怎么就攤上這事灰殴。” “怎么了掰邢?”我有些...
    開封第一講書人閱讀 164,960評論 0 355
  • 文/不壞的土叔 我叫張陵验懊,是天一觀的道長擅羞。 經(jīng)常有香客問我,道長义图,這世上最難降的妖魔是什么减俏? 我笑而不...
    開封第一講書人閱讀 58,750評論 1 294
  • 正文 為了忘掉前任,我火速辦了婚禮碱工,結(jié)果婚禮上娃承,老公的妹妹穿的比我還像新娘。我一直安慰自己怕篷,他們只是感情好历筝,可當(dāng)我...
    茶點故事閱讀 67,764評論 6 392
  • 文/花漫 我一把揭開白布。 她就那樣靜靜地躺著廊谓,像睡著了一般梳猪。 火紅的嫁衣襯著肌膚如雪。 梳的紋絲不亂的頭發(fā)上蒸痹,一...
    開封第一講書人閱讀 51,604評論 1 305
  • 那天春弥,我揣著相機與錄音,去河邊找鬼叠荠。 笑死匿沛,一個胖子當(dāng)著我的面吹牛,可吹牛的內(nèi)容都是我干的榛鼎。 我是一名探鬼主播逃呼,決...
    沈念sama閱讀 40,347評論 3 418
  • 文/蒼蘭香墨 我猛地睜開眼,長吁一口氣:“原來是場噩夢啊……” “哼者娱!你這毒婦竟也來了抡笼?” 一聲冷哼從身側(cè)響起,我...
    開封第一講書人閱讀 39,253評論 0 276
  • 序言:老撾萬榮一對情侶失蹤黄鳍,失蹤者是張志新(化名)和其女友劉穎推姻,沒想到半個月后,有當(dāng)?shù)厝嗽跇淞掷锇l(fā)現(xiàn)了一具尸體际起,經(jīng)...
    沈念sama閱讀 45,702評論 1 315
  • 正文 獨居荒郊野嶺守林人離奇死亡拾碌,尸身上長有42處帶血的膿包…… 初始之章·張勛 以下內(nèi)容為張勛視角 年9月15日...
    茶點故事閱讀 37,893評論 3 336
  • 正文 我和宋清朗相戀三年吐葱,在試婚紗的時候發(fā)現(xiàn)自己被綠了街望。 大學(xué)時的朋友給我發(fā)了我未婚夫和他白月光在一起吃飯的照片。...
    茶點故事閱讀 40,015評論 1 348
  • 序言:一個原本活蹦亂跳的男人離奇死亡弟跑,死狀恐怖灾前,靈堂內(nèi)的尸體忽然破棺而出,到底是詐尸還是另有隱情孟辑,我是刑警寧澤哎甲,帶...
    沈念sama閱讀 35,734評論 5 346
  • 正文 年R本政府宣布蔫敲,位于F島的核電站,受9級特大地震影響炭玫,放射性物質(zhì)發(fā)生泄漏奈嘿。R本人自食惡果不足惜,卻給世界環(huán)境...
    茶點故事閱讀 41,352評論 3 330
  • 文/蒙蒙 一吞加、第九天 我趴在偏房一處隱蔽的房頂上張望裙犹。 院中可真熱鬧,春花似錦衔憨、人聲如沸叶圃。這莊子的主人今日做“春日...
    開封第一講書人閱讀 31,934評論 0 22
  • 文/蒼蘭香墨 我抬頭看了看天上的太陽掺冠。三九已至,卻和暖如春码党,著一層夾襖步出監(jiān)牢的瞬間德崭,已是汗流浹背。 一陣腳步聲響...
    開封第一講書人閱讀 33,052評論 1 270
  • 我被黑心中介騙來泰國打工闽瓢, 沒想到剛下飛機就差點兒被人妖公主榨干…… 1. 我叫王不留接癌,地道東北人。 一個月前我還...
    沈念sama閱讀 48,216評論 3 371
  • 正文 我出身青樓扣讼,卻偏偏與公主長得像缺猛,于是被迫代替她去往敵國和親。 傳聞我的和親對象是個殘疾皇子椭符,可洞房花燭夜當(dāng)晚...
    茶點故事閱讀 44,969評論 2 355

推薦閱讀更多精彩內(nèi)容

  • 在上一篇中我們介紹了 mpi4py 中的錯誤處理荔燎,下面我們將介紹 mpi4py 中的 info 和 asserti...
    自可樂閱讀 726評論 0 0
  • Android 自定義View的各種姿勢1 Activity的顯示之ViewRootImpl詳解 Activity...
    passiontim閱讀 172,154評論 25 707
  • 問答 1.有如下代碼,解釋Person销钝、 prototype有咨、proto、p蒸健、constructor之間的關(guān)聯(lián)座享。 ...
    GarenWang閱讀 369評論 0 0
  • 2017年6月23日盯捌,周五晚7點淳衙,妹妹微信里說大表姐的老公走了,腦溢血。好突然箫攀,之前沒有半點消息說表姐夫的身體不好...
    長豐105閱讀 289評論 0 0
  • SPI的全名為Service Provider Interface.普通開發(fā)人員可能不熟悉肠牲,因為這個是針對廠商或者...
    小陳阿飛閱讀 243評論 0 0