|
|
|
@ -268,6 +268,49 @@ contains
|
|
|
|
|
!
|
|
|
|
|
call psb_cdall(ctxt,desc_a,info,vl=myidx)
|
|
|
|
|
|
|
|
|
|
!
|
|
|
|
|
! Specify process topology
|
|
|
|
|
!
|
|
|
|
|
block
|
|
|
|
|
!
|
|
|
|
|
! Use adjcncy methods
|
|
|
|
|
!
|
|
|
|
|
integer(psb_mpk_), allocatable :: neighbours(:)
|
|
|
|
|
integer(psb_mpk_) :: cnt
|
|
|
|
|
logical, parameter :: debug_adj=.true.
|
|
|
|
|
if (debug_adj.and.(np > 1)) then
|
|
|
|
|
cnt = 0
|
|
|
|
|
allocate(neighbours(np))
|
|
|
|
|
if (iamx < npx-1) then
|
|
|
|
|
cnt = cnt + 1
|
|
|
|
|
call ijk2idx(neighbours(cnt),iamx+1,iamy,iamz,npx,npy,npz,base=0)
|
|
|
|
|
end if
|
|
|
|
|
if (iamy < npy-1) then
|
|
|
|
|
cnt = cnt + 1
|
|
|
|
|
call ijk2idx(neighbours(cnt),iamx,iamy+1,iamz,npx,npy,npz,base=0)
|
|
|
|
|
end if
|
|
|
|
|
if (iamz < npz-1) then
|
|
|
|
|
cnt = cnt + 1
|
|
|
|
|
call ijk2idx(neighbours(cnt),iamx,iamy,iamz+1,npx,npy,npz,base=0)
|
|
|
|
|
end if
|
|
|
|
|
if (iamx >0) then
|
|
|
|
|
cnt = cnt + 1
|
|
|
|
|
call ijk2idx(neighbours(cnt),iamx-1,iamy,iamz,npx,npy,npz,base=0)
|
|
|
|
|
end if
|
|
|
|
|
if (iamy >0) then
|
|
|
|
|
cnt = cnt + 1
|
|
|
|
|
call ijk2idx(neighbours(cnt),iamx,iamy-1,iamz,npx,npy,npz,base=0)
|
|
|
|
|
end if
|
|
|
|
|
if (iamz >0) then
|
|
|
|
|
cnt = cnt + 1
|
|
|
|
|
call ijk2idx(neighbours(cnt),iamx,iamy,iamz-1,npx,npy,npz,base=0)
|
|
|
|
|
end if
|
|
|
|
|
call psb_realloc(cnt, neighbours,info)
|
|
|
|
|
call desc_a%set_p_adjcncy(neighbours)
|
|
|
|
|
!write(0,*) iam,' Check on neighbours: ',desc_a%get_p_adjcncy()
|
|
|
|
|
end if
|
|
|
|
|
end block
|
|
|
|
|
|
|
|
|
|
case default
|
|
|
|
|
write(psb_err_unit,*) iam, 'Initialization error: should not get here'
|
|
|
|
|
info = -1
|
|
|
|
@ -674,6 +717,41 @@ contains
|
|
|
|
|
!
|
|
|
|
|
call psb_cdall(ctxt,desc_a,info,vl=myidx)
|
|
|
|
|
|
|
|
|
|
!
|
|
|
|
|
! Specify process topology
|
|
|
|
|
!
|
|
|
|
|
block
|
|
|
|
|
!
|
|
|
|
|
! Use adjcncy methods
|
|
|
|
|
!
|
|
|
|
|
integer(psb_mpk_), allocatable :: neighbours(:)
|
|
|
|
|
integer(psb_mpk_) :: cnt
|
|
|
|
|
logical, parameter :: debug_adj=.true.
|
|
|
|
|
if (debug_adj.and.(np > 1)) then
|
|
|
|
|
cnt = 0
|
|
|
|
|
allocate(neighbours(np))
|
|
|
|
|
if (iamx < npx-1) then
|
|
|
|
|
cnt = cnt + 1
|
|
|
|
|
call ijk2idx(neighbours(cnt),iamx+1,iamy,npx,npy,base=0)
|
|
|
|
|
end if
|
|
|
|
|
if (iamy < npy-1) then
|
|
|
|
|
cnt = cnt + 1
|
|
|
|
|
call ijk2idx(neighbours(cnt),iamx,iamy+1,npx,npy,base=0)
|
|
|
|
|
end if
|
|
|
|
|
if (iamx >0) then
|
|
|
|
|
cnt = cnt + 1
|
|
|
|
|
call ijk2idx(neighbours(cnt),iamx-1,iamy,npx,npy,base=0)
|
|
|
|
|
end if
|
|
|
|
|
if (iamy >0) then
|
|
|
|
|
cnt = cnt + 1
|
|
|
|
|
call ijk2idx(neighbours(cnt),iamx,iamy-1,npx,npy,base=0)
|
|
|
|
|
end if
|
|
|
|
|
call psb_realloc(cnt, neighbours,info)
|
|
|
|
|
call desc_a%set_p_adjcncy(neighbours)
|
|
|
|
|
!write(0,*) iam,' Check on neighbours: ',desc_a%get_p_adjcncy()
|
|
|
|
|
end if
|
|
|
|
|
end block
|
|
|
|
|
|
|
|
|
|
case default
|
|
|
|
|
write(psb_err_unit,*) iam, 'Initialization error: should not get here'
|
|
|
|
|
info = -1
|
|
|
|
|