我從外部導(dǎo)入進(jìn)去的三維數(shù)組如下
u=loadtxt("u_fla.dat", dtype=float, unpack=True)
v=loadtxt("v_fla.dat", dtype=float, unpack=True)
w=loadtxt("w_fla.dat", dtype=float, unpack=True)
u1=u.reshape(256,256,256)
v1=v.reshape(256,256,256)
w1=w.reshape(256,256,256)
2.程序內(nèi)部的速度U,V,W已經(jīng)按照x方向利用切片分快了危喉,我需要將導(dǎo)入數(shù)組對(duì)應(yīng)CPU分塊
先要確認(rèn)使用多少個(gè)核心(假如14核心翼悴,256節(jié)點(diǎn))并行計(jì)算雹姊,并輸出每個(gè)核心在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,:,:]