猿代碼MPI并行計算實戰(zhàn)課程
2023-07-16 11:03 作者:一起拼課學(xué)習(xí)下_ | 我要投稿
程序內(nèi)部的速度U,V,W已經(jīng)按照x方向利用切片分快了,我需要將導(dǎo)入數(shù)組對應(yīng)CPU分塊
先要確認使用多少個核心(假如14核心,256節(jié)點)并行計算,并輸出每個核心在x方向上的上下限
if (0<=comm_rank<=3):
? ? ?i=comm_rank*19
? ? ?j=(comm_rank+1)*19
? ? ?U[0]=u1[i:j,:,:]
? ? ?U[1]=v1[i:j,:,:]
? ? ?U[2]=w1[i:j,:,:]
elif (4<=comm_rank<=13):
? ? ? a=comm_rank*18+4
? ? ? b=(comm_rank+1)*18+4
? ? ??U[0]=u1[a:b,:,:]
? ? ? U[1]=v1[a:b,:,:]
? ? ? U[2]=w1[a:b,:,:]
標簽: