! ! Parallel Sparse BLAS version 3.5 ! (C) Copyright 2006-2018 ! Salvatore Filippone ! Alfredo Buttari ! ! Redistribution and use in source and binary forms, with or without ! modification, are permitted provided that the following conditions ! are met: ! 1. Redistributions of source code must retain the above copyright ! notice, this list of conditions and the following disclaimer. ! 2. Redistributions in binary form must reproduce the above copyright ! notice, this list of conditions, and the following disclaimer in the ! documentation and/or other materials provided with the distribution. ! 3. The name of the PSBLAS group or the names of its contributors may ! not be used to endorse or promote products derived from this ! software without specific written permission. ! ! THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS ! ``AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED ! TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR ! PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE PSBLAS GROUP OR ITS CONTRIBUTORS ! BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR ! CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF ! SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS ! INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN ! CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ! ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE ! POSSIBILITY OF SUCH DAMAGE. ! ! ! File: psb_zsphalo.f90 ! ! Subroutine: psb_zsphalo psb_lzsphalo ! This routine does the retrieval of remote matrix rows. ! Retrieval is done through GETROW, therefore it should work ! for any matrix format in A; as for the output, default is CSR. ! ! There is also a specialized version lz_CSR whose interface ! is adapted for the needs of z_par_csr_spspmm. ! ! There are three possible exchange algorithms: ! 1. Use MPI_Alltoallv ! 2. Use psb_simple_a2av ! 3. Use psb_simple_triad_a2av ! Default choice is 3. The MPI variant has proved to be inefficient; ! that is because it is not persistent, therefore you pay the initialization price ! every time, and it is not optimized for a sparse communication pattern, ! most MPI implementations assume that all communications are non-empty. ! The PSB_SIMPLE variants reuse the same communicator, and go for a simplistic ! sequence of sends/receive that is quite efficient for a sparse communication ! pattern. To be refined/reviewed in the future to compare with neighbour ! persistent collectives. ! ! Arguments: ! a - type(psb_zspmat_type) The local part of input matrix A ! desc_a - type(psb_desc_type). The communication descriptor. ! blck - type(psb_zspmat_type) The local part of output matrix BLCK ! info - integer. Return code ! rowcnv - logical Should row/col indices be converted ! colcnv - logical to/from global numbering when sent/received? ! default is .TRUE. ! rowscale - logical Should row/col indices on output be remapped ! colscale - logical from MIN:MAX to 1:(MAX-MIN+1) ? ! default is .FALSE. ! (commmon use is ROWSCALE=.TRUE., COLSCALE=.FALSE.) ! data - integer Which index list in desc_a should be used to retrieve ! rows, default psb_comm_halo_ ! psb_comm_halo_ use halo_index ! psb_comm_ext_ use ext_index ! psb_comm_ovrl_ DISABLED for this routine. ! Subroutine psb_zsphalo(a,desc_a,blk,info,rowcnv,colcnv,& & rowscale,colscale,outfmt,data) use psb_base_mod, psb_protect_name => psb_zsphalo #ifdef MPI_MOD use mpi #endif Implicit None #ifdef MPI_H include 'mpif.h' #endif Type(psb_zspmat_type),Intent(in) :: a Type(psb_zspmat_type),Intent(inout) :: blk Type(psb_desc_type),Intent(in), target :: desc_a integer(psb_ipk_), intent(out) :: info logical, optional, intent(in) :: rowcnv,colcnv,rowscale,colscale character(len=5), optional :: outfmt integer(psb_ipk_), intent(in), optional :: data ! ...local scalars.... type(psb_ctxt_type) :: ctxt integer(psb_ipk_) :: np,me integer(psb_ipk_) :: counter,proc,i, & & n_el_send,k,n_el_recv,idx, r, tot_elem,& & n_elem, j, ipx,mat_recv, iszs, iszr,idxs,idxr,nz,& & irmin,icmin,irmax,icmax,data_,ngtz,totxch,nxs, nxr,& & l1, err_act, nsnds, nrcvs, lnr, lnc, lnnz integer(psb_mpk_) :: icomm, minfo integer(psb_mpk_), allocatable :: brvindx(:), & & rvsz(:), bsdindx(:),sdsz(:) #if defined(IPK4) && defined(LPK8) ! If globals are 8 bytes but locals are 4, things get tricky integer(psb_ipk_), allocatable :: liasnd(:), ljasnd(:) integer(psb_lpk_), allocatable :: iasnd(:), jasnd(:), iarcv(:), jarcv(:) #else integer(psb_ipk_), allocatable :: iasnd(:), jasnd(:) #endif complex(psb_dpk_), allocatable :: valsnd(:) type(psb_z_coo_sparse_mat), allocatable :: acoo integer(psb_ipk_), pointer :: idxv(:) class(psb_i_base_vect_type), pointer :: pdxv integer(psb_ipk_), allocatable :: ipdxv(:) logical :: rowcnv_,colcnv_,rowscale_,colscale_ character(len=5) :: outfmt_ integer(psb_ipk_) :: debug_level, debug_unit character(len=20) :: name, ch_err info=psb_success_ name='psb_zsphalo' call psb_erractionsave(err_act) if (psb_errstatus_fatal()) then info = psb_err_internal_error_ ; goto 9999 end if debug_unit = psb_get_debug_unit() debug_level = psb_get_debug_level() ctxt = desc_a%get_context() icomm = desc_a%get_mpic() Call psb_info(ctxt, me, np) if (debug_level >= psb_debug_outer_) & & write(debug_unit,*) me,' ',trim(name),': Start' if (present(rowcnv)) then rowcnv_ = rowcnv else rowcnv_ = .true. endif if (present(colcnv)) then colcnv_ = colcnv else colcnv_ = .true. endif if (present(rowscale)) then rowscale_ = rowscale else rowscale_ = .false. endif if (present(colscale)) then colscale_ = colscale else colscale_ = .false. endif if (present(data)) then data_ = data else data_ = psb_comm_halo_ endif if (present(outfmt)) then outfmt_ = psb_toupper(outfmt) else outfmt_ = 'CSR' endif Allocate(brvindx(np+1),& & rvsz(np),sdsz(np),bsdindx(np+1), acoo,stat=info) if (info /= psb_success_) then info=psb_err_alloc_dealloc_ call psb_errpush(info,name) goto 9999 end if If (debug_level >= psb_debug_outer_)& & write(debug_unit,*) me,' ',trim(name),': Data selector',data_ select case(data_) case(psb_comm_halo_,psb_comm_ext_ ) ! Do not accept OVRLAP_INDEX any longer. case default call psb_errpush(psb_err_from_subroutine_,name,a_err='wrong Data selector') goto 9999 end select sdsz(:)=0 rvsz(:)=0 l1 = 0 ipx = 1 brvindx(ipx) = 0 bsdindx(ipx) = 0 counter=1 idx = 0 idxs = 0 idxr = 0 call desc_a%get_list(data_,pdxv,totxch,nxr,nxs,info) ipdxv = pdxv%get_vect() ! For all rows in the halo descriptor, extract the row size lnr = 0 Do proc=ipdxv(counter) if (proc == -1) exit n_el_recv = ipdxv(counter+psb_n_elem_recv_) counter = counter+n_el_recv n_el_send = ipdxv(counter+psb_n_elem_send_) tot_elem = 0 Do j=0,n_el_send-1 idx = ipdxv(counter+psb_elem_send_+j) n_elem = a%get_nz_row(idx) tot_elem = tot_elem+n_elem Enddo sdsz(proc+1) = tot_elem lnr = lnr + n_el_recv counter = counter+n_el_send+3 Enddo ! ! Exchange row sizes, so as to know sends/receives. ! This is different from the halo exchange because the ! size of the rows may vary, as opposed to fixed ! (multi) vector row size. ! call mpi_alltoall(sdsz,1,psb_mpi_mpk_,& & rvsz,1,psb_mpi_mpk_,icomm,minfo) if (info /= psb_success_) then info=psb_err_from_subroutine_ call psb_errpush(info,name,a_err='mpi_alltoall') goto 9999 end if nsnds = count(sdsz /= 0) nrcvs = count(rvsz /= 0) if (debug_level >= psb_debug_outer_)& & write(debug_unit,*) me,' ',trim(name),': Done initial alltoall',nsnds,nrcvs idxs = 0 idxr = 0 counter = 1 Do proc=ipdxv(counter) if (proc == -1) exit n_el_recv = ipdxv(counter+psb_n_elem_recv_) counter = counter+n_el_recv n_el_send = ipdxv(counter+psb_n_elem_send_) bsdindx(proc+1) = idxs idxs = idxs + sdsz(proc+1) brvindx(proc+1) = idxr idxr = idxr + rvsz(proc+1) counter = counter+n_el_send+3 Enddo iszr = sum(rvsz) mat_recv = iszr iszs = sum(sdsz) lnnz = max(iszr,iszs,ione) lnc = a%get_ncols() call acoo%allocate(lnr,lnc,lnnz) if (debug_level >= psb_debug_outer_)& & write(debug_unit,*) me,' ',trim(name),': Sizes:',acoo%get_size(),& & ' Send:',sdsz(:),' Receive:',rvsz(:) call psb_ensure_size(max(iszs,1),iasnd,info) if (info == psb_success_) call psb_ensure_size(max(iszs,1),jasnd,info) if (info == psb_success_) call psb_ensure_size(max(iszs,1),valsnd,info) #if defined(IPK4) && defined(LPK8) ! If globals are 8 bytes but locals are not, things get tricky if (info == psb_success_) call psb_ensure_size(max(iszs,1),liasnd,info) if (info == psb_success_) call psb_ensure_size(max(iszs,1),ljasnd,info) if (info == psb_success_) call psb_ensure_size(max(iszr,1),iarcv,info) if (info == psb_success_) call psb_ensure_size(max(iszr,1),jarcv,info) if (info /= psb_success_) then info=psb_err_from_subroutine_; ch_err='psb_sp_reall' call psb_errpush(info,name,a_err=ch_err); goto 9999 end if l1 = 0 ipx = 1 counter=1 idx = 0 tot_elem=0 Do proc=ipdxv(counter) if (proc == -1) exit n_el_recv=ipdxv(counter+psb_n_elem_recv_) counter=counter+n_el_recv n_el_send=ipdxv(counter+psb_n_elem_send_) Do j=0,n_el_send-1 idx = ipdxv(counter+psb_elem_send_+j) call a%csget(idx,idx,ngtz,liasnd,ljasnd,valsnd,info,& & append=.true.,nzin=tot_elem) if (info /= psb_success_) then info=psb_err_from_subroutine_ ch_err='psb_sp_getrow' call psb_errpush(info,name,a_err=ch_err) goto 9999 end if tot_elem=tot_elem+ngtz Enddo ipx = ipx + 1 counter = counter+n_el_send+3 Enddo nz = tot_elem if (rowcnv_) then call psb_loc_to_glob(liasnd(1:nz),iasnd(1:nz),desc_a,info,iact='I') else iasnd(1:nz) = liasnd(1:nz) end if if (colcnv_) then call psb_loc_to_glob(ljasnd(1:nz),jasnd(1:nz),desc_a,info,iact='I') else jasnd(1:nz) = ljasnd(1:nz) end if if (info /= psb_success_) then info=psb_err_from_subroutine_; ch_err='psb_loc_to_glob' call psb_errpush(info,name,a_err=ch_err); goto 9999 end if select case(psb_get_sp_a2av_alg()) case(psb_sp_a2av_smpl_triad_) call psb_simple_triad_a2av(valsnd,iasnd,jasnd,sdsz,bsdindx,& & acoo%val,iarcv,jarcv,rvsz,brvindx,ctxt,info) case(psb_sp_a2av_smpl_v_) call psb_simple_a2av(valsnd,sdsz,bsdindx,& & acoo%val,rvsz,brvindx,ctxt,info) if (info == psb_success_) call psb_simple_a2av(iasnd,sdsz,bsdindx,& & iarcv,rvsz,brvindx,ctxt,info) if (info == psb_success_) call psb_simple_a2av(jasnd,sdsz,bsdindx,& & jarcv,rvsz,brvindx,ctxt,info) case(psb_sp_a2av_mpi_) call mpi_alltoallv(valsnd,sdsz,bsdindx,psb_mpi_c_dpk_,& & acoo%val,rvsz,brvindx,psb_mpi_c_dpk_,icomm,minfo) if (minfo == mpi_success) & & call mpi_alltoallv(iasnd,sdsz,bsdindx,psb_mpi_lpk_,& & iarcv,rvsz,brvindx,psb_mpi_lpk_,icomm,minfo) if (minfo == mpi_success) & & call mpi_alltoallv(jasnd,sdsz,bsdindx,psb_mpi_lpk_,& & jarcv,rvsz,brvindx,psb_mpi_lpk_,icomm,minfo) if (minfo /= mpi_success) info = minfo case default info = psb_err_internal_error_ call psb_errpush(info,name,a_err='wrong A2AV alg selector') goto 9999 end select if (info /= psb_success_) then info=psb_err_from_subroutine_ ch_err='alltoallv' call psb_errpush(info,name,a_err=ch_err) goto 9999 end if ! ! Convert into local numbering ! if (rowcnv_) then call psb_glob_to_loc(iarcv(1:iszr),acoo%ia(1:iszr),desc_a,info,iact='I') else acoo%ia(1:iszr) = iarcv(1:iszr) end if if (colcnv_) then call psb_glob_to_loc(jarcv(1:iszr),acoo%ja(1:iszr),desc_a,info,iact='I') else acoo%ja(1:iszr) = jarcv(1:iszr) end if #else if (info /= psb_success_) then info=psb_err_from_subroutine_; ch_err='psb_sp_reall' call psb_errpush(info,name,a_err=ch_err); goto 9999 end if l1 = 0 ipx = 1 counter=1 idx = 0 ! ! Make sure to get all columns in csget. ! This is necessary when sphalo is used to compute a transpose, ! as opposed to just gathering halo for spspmm purposes. ! tot_elem = 0 Do proc = ipdxv(counter) if (proc == -1) exit n_el_recv = ipdxv(counter+psb_n_elem_recv_) counter = counter+n_el_recv n_el_send = ipdxv(counter+psb_n_elem_send_) Do j=0,n_el_send-1 idx = ipdxv(counter+psb_elem_send_+j) call a%csget(idx,idx,ngtz,iasnd,jasnd,valsnd,info,& & append=.true.,nzin=tot_elem) if (info /= psb_success_) then info=psb_err_from_subroutine_ call psb_errpush(info,name,a_err='psb_sp_getrow') goto 9999 end if tot_elem = tot_elem+ngtz Enddo ipx = ipx + 1 counter = counter+n_el_send+3 Enddo nz = tot_elem if (rowcnv_) call psb_loc_to_glob(iasnd(1:nz),desc_a,info,iact='I') if (colcnv_) call psb_loc_to_glob(jasnd(1:nz),desc_a,info,iact='I') if (info /= psb_success_) then info=psb_err_from_subroutine_ call psb_errpush(info,name,a_err='psb_loc_to_glob') goto 9999 end if select case(psb_get_sp_a2av_alg()) case(psb_sp_a2av_smpl_triad_) call psb_simple_triad_a2av(valsnd,iasnd,jasnd,sdsz,bsdindx,& & acoo%val,acoo%ia,acoo%ja,rvsz,brvindx,ctxt,info) case(psb_sp_a2av_smpl_v_) call psb_simple_a2av(valsnd,sdsz,bsdindx,& & acoo%val,rvsz,brvindx,ctxt,info) if (info == psb_success_) call psb_simple_a2av(iasnd,sdsz,bsdindx,& & acoo%ia,rvsz,brvindx,ctxt,info) if (info == psb_success_) call psb_simple_a2av(jasnd,sdsz,bsdindx,& & acoo%ja,rvsz,brvindx,ctxt,info) case(psb_sp_a2av_mpi_) call mpi_alltoallv(valsnd,sdsz,bsdindx,psb_mpi_c_dpk_,& & acoo%val,rvsz,brvindx,psb_mpi_c_dpk_,icomm,minfo) if (minfo == mpi_success) & & call mpi_alltoallv(iasnd,sdsz,bsdindx,psb_mpi_lpk_,& & acoo%ia,rvsz,brvindx,psb_mpi_lpk_,icomm,minfo) if (minfo == mpi_success) & & call mpi_alltoallv(jasnd,sdsz,bsdindx,psb_mpi_lpk_,& & acoo%ja,rvsz,brvindx,psb_mpi_lpk_,icomm,minfo) if (minfo /= mpi_success) info = minfo case default info = psb_err_internal_error_ call psb_errpush(info,name,a_err='wrong A2AV alg selector') goto 9999 end select if (info /= psb_success_) then info=psb_err_from_subroutine_ ch_err='alltoallv' call psb_errpush(info,name,a_err=ch_err) goto 9999 end if ! ! Convert into local numbering ! if (rowcnv_) call psb_glob_to_loc(acoo%ia(1:iszr),desc_a,info,iact='I') if (colcnv_) call psb_glob_to_loc(acoo%ja(1:iszr),desc_a,info,iact='I') #endif if (info /= psb_success_) then info=psb_err_from_subroutine_ ch_err='psbglob_to_loc' call psb_errpush(info,name,a_err=ch_err) goto 9999 end if l1 = 0 call acoo%set_nrows(izero) ! irmin = huge(irmin) icmin = huge(icmin) irmax = 0 icmax = 0 Do i=1,iszr r=(acoo%ia(i)) k=(acoo%ja(i)) ! Just in case some of the conversions were out-of-range If ((r>0).and.(k>0)) Then l1=l1+1 acoo%val(l1) = acoo%val(i) acoo%ia(l1) = r acoo%ja(l1) = k irmin = min(irmin,r) irmax = max(irmax,r) icmin = min(icmin,k) icmax = max(icmax,k) End If Enddo if (rowscale_) then call acoo%set_nrows(max(irmax-irmin+1,0)) acoo%ia(1:l1) = acoo%ia(1:l1) - irmin + 1 else call acoo%set_nrows(irmax) end if if (colscale_) then call acoo%set_ncols(max(icmax-icmin+1,0)) acoo%ja(1:l1) = acoo%ja(1:l1) - icmin + 1 else call acoo%set_ncols(icmax) end if call acoo%set_nzeros(l1) call acoo%set_sorted(.false.) if (debug_level >= psb_debug_outer_)& & write(debug_unit,*) me,' ',trim(name),& & ': End data exchange',counter,l1 call move_alloc(acoo,blk%a) ! Do we expect any duplicates to appear???? call blk%cscnv(info,type=outfmt_,dupl=psb_dupl_add_) if (info /= psb_success_) then info=psb_err_from_subroutine_ ch_err='psb_spcnv' call psb_errpush(info,name,a_err=ch_err) goto 9999 end if Deallocate(brvindx,bsdindx,rvsz,sdsz,& & iasnd,jasnd,valsnd,stat=info) if (debug_level >= psb_debug_outer_)& & write(debug_unit,*) me,' ',trim(name),': Done' call psb_erractionrestore(err_act) return 9999 call psb_error_handler(ctxt,err_act) return End Subroutine psb_zsphalo Subroutine psb_lzsphalo(a,desc_a,blk,info,rowcnv,colcnv,& & rowscale,colscale,outfmt,data) use psb_base_mod, psb_protect_name => psb_lzsphalo #ifdef MPI_MOD use mpi #endif Implicit None #ifdef MPI_H include 'mpif.h' #endif Type(psb_lzspmat_type),Intent(in) :: a Type(psb_lzspmat_type),Intent(inout) :: blk Type(psb_desc_type),Intent(in), target :: desc_a integer(psb_ipk_), intent(out) :: info logical, optional, intent(in) :: rowcnv,colcnv,rowscale,colscale character(len=5), optional :: outfmt integer(psb_ipk_), intent(in), optional :: data ! ...local scalars.... type(psb_ctxt_type) :: ctxt integer(psb_ipk_) :: np,me integer(psb_ipk_) :: counter, proc, i, n_el_send,n_el_recv, & & n_elem, j, ipx,mat_recv, idxs,idxr,nz,& & data_,totxch,nxs, nxr, ncg integer(psb_lpk_) :: r, k, irmin, irmax, icmin, icmax, iszs, iszr, & & lidx, l1, lnr, lnc, lnnz, idx, ngtz, tot_elem integer(psb_mpk_) :: icomm, minfo integer(psb_mpk_), allocatable :: brvindx(:), & & rvsz(:), bsdindx(:),sdsz(:) integer(psb_lpk_), allocatable :: iasnd(:), jasnd(:) complex(psb_dpk_), allocatable :: valsnd(:) type(psb_lz_coo_sparse_mat), allocatable :: acoo integer(psb_ipk_), pointer :: idxv(:) class(psb_i_base_vect_type), pointer :: pdxv integer(psb_ipk_), allocatable :: ipdxv(:) logical :: rowcnv_,colcnv_,rowscale_,colscale_ character(len=5) :: outfmt_ integer(psb_ipk_) :: debug_level, debug_unit, err_act character(len=20) :: name, ch_err info=psb_success_ name='psb_zsphalo' call psb_erractionsave(err_act) if (psb_errstatus_fatal()) then info = psb_err_internal_error_ ; goto 9999 end if debug_unit = psb_get_debug_unit() debug_level = psb_get_debug_level() ctxt = desc_a%get_context() icomm = desc_a%get_mpic() Call psb_info(ctxt, me, np) if (debug_level >= psb_debug_outer_) & & write(debug_unit,*) me,' ',trim(name),': Start' if (present(rowcnv)) then rowcnv_ = rowcnv else rowcnv_ = .true. endif if (present(colcnv)) then colcnv_ = colcnv else colcnv_ = .true. endif if (present(rowscale)) then rowscale_ = rowscale else rowscale_ = .false. endif if (present(colscale)) then colscale_ = colscale else colscale_ = .false. endif if (present(data)) then data_ = data else data_ = psb_comm_halo_ endif if (present(outfmt)) then outfmt_ = psb_toupper(outfmt) else outfmt_ = 'CSR' endif Allocate(brvindx(np+1),& & rvsz(np),sdsz(np),bsdindx(np+1), acoo,stat=info) if (info /= psb_success_) then info=psb_err_alloc_dealloc_ call psb_errpush(info,name) goto 9999 end if If (debug_level >= psb_debug_outer_)& & write(debug_unit,*) me,' ',trim(name),': Data selector',data_ select case(data_) case(psb_comm_halo_,psb_comm_ext_ ) ! Do not accept OVRLAP_INDEX any longer. case default call psb_errpush(psb_err_from_subroutine_,name,a_err='wrong Data selector') goto 9999 end select sdsz(:)=0 rvsz(:)=0 l1 = 0 ipx = 1 brvindx(ipx) = 0 bsdindx(ipx) = 0 counter=1 idx = 0 idxs = 0 idxr = 0 call desc_a%get_list(data_,pdxv,totxch,nxr,nxs,info) ipdxv = pdxv%get_vect() ! For all rows in the halo descriptor, extract and send/receive. lnr = 0 Do proc=ipdxv(counter) if (proc == -1) exit n_el_recv = ipdxv(counter+psb_n_elem_recv_) counter = counter+n_el_recv n_el_send = ipdxv(counter+psb_n_elem_send_) tot_elem = 0 Do j=0,n_el_send-1 idx = ipdxv(counter+psb_elem_send_+j) n_elem = a%get_nz_row(idx) tot_elem = tot_elem+n_elem Enddo sdsz(proc+1) = tot_elem lnr = lnr + n_el_recv counter = counter+n_el_send+3 Enddo ! ! Exchange row sizes, so as to know sends/receives. ! This is different from the halo exchange because the ! size of the rows may vary, as opposed to fixed ! (multi) vector row size. ! call mpi_alltoall(sdsz,1,psb_mpi_mpk_,& & rvsz,1,psb_mpi_mpk_,icomm,minfo) if (info /= psb_success_) then info=psb_err_from_subroutine_ call psb_errpush(info,name,a_err='mpi_alltoall') goto 9999 end if idxs = 0 idxr = 0 counter = 1 Do proc=ipdxv(counter) if (proc == -1) exit n_el_recv = ipdxv(counter+psb_n_elem_recv_) counter = counter+n_el_recv n_el_send = ipdxv(counter+psb_n_elem_send_) bsdindx(proc+1) = idxs idxs = idxs + sdsz(proc+1) brvindx(proc+1) = idxr idxr = idxr + rvsz(proc+1) counter = counter+n_el_send+3 Enddo iszr=sum(1_psb_lpk_*rvsz) mat_recv = iszr iszs=sum(1_psb_lpk_*sdsz) lnnz = max(iszr,iszs,lone) lnc = a%get_ncols() call acoo%allocate(lnr,lnc,lnnz) if (debug_level >= psb_debug_outer_)& & write(debug_unit,*) me,' ',trim(name),': Sizes:',acoo%get_size(),& & ' Send:',sdsz(:),' Receive:',rvsz(:) if (info == psb_success_) call psb_ensure_size(max(iszs,1),iasnd,info) if (info == psb_success_) call psb_ensure_size(max(iszs,1),jasnd,info) if (info == psb_success_) call psb_ensure_size(max(iszs,1),valsnd,info) if (info /= psb_success_) then info=psb_err_from_subroutine_ call psb_errpush(info,name,a_err='ensure_size') goto 9999 end if l1 = 0 ipx = 1 counter=1 idx = 0 ncg = huge(ncg) tot_elem=0 Do proc=ipdxv(counter) if (proc == -1) exit n_el_recv=ipdxv(counter+psb_n_elem_recv_) counter=counter+n_el_recv n_el_send=ipdxv(counter+psb_n_elem_send_) Do j=0,n_el_send-1 idx = ipdxv(counter+psb_elem_send_+j) call a%csget(idx,idx,ngtz,iasnd,jasnd,valsnd,info,& & append=.true.,nzin=tot_elem) if (info /= psb_success_) then info=psb_err_from_subroutine_ call psb_errpush(info,name,a_err='psb_sp_getrow') goto 9999 end if tot_elem=tot_elem+ngtz Enddo ipx = ipx + 1 counter = counter+n_el_send+3 Enddo nz = tot_elem if (rowcnv_) call psb_loc_to_glob(iasnd(1:nz),desc_a,info,iact='I') if (colcnv_) call psb_loc_to_glob(jasnd(1:nz),desc_a,info,iact='I') if (info /= psb_success_) then info=psb_err_from_subroutine_ call psb_errpush(info,name,a_err='psb_loc_to_glob') goto 9999 end if select case(psb_get_sp_a2av_alg()) case(psb_sp_a2av_smpl_triad_) call psb_simple_triad_a2av(valsnd,iasnd,jasnd,sdsz,bsdindx,& & acoo%val,acoo%ia,acoo%ja,rvsz,brvindx,ctxt,info) case(psb_sp_a2av_smpl_v_) call psb_simple_a2av(valsnd,sdsz,bsdindx,& & acoo%val,rvsz,brvindx,ctxt,info) if (info == psb_success_) call psb_simple_a2av(iasnd,sdsz,bsdindx,& & acoo%ia,rvsz,brvindx,ctxt,info) if (info == psb_success_) call psb_simple_a2av(jasnd,sdsz,bsdindx,& & acoo%ja,rvsz,brvindx,ctxt,info) case(psb_sp_a2av_mpi_) call mpi_alltoallv(valsnd,sdsz,bsdindx,psb_mpi_c_dpk_,& & acoo%val,rvsz,brvindx,psb_mpi_c_dpk_,icomm,minfo) if (minfo == mpi_success) & & call mpi_alltoallv(iasnd,sdsz,bsdindx,psb_mpi_lpk_,& & acoo%ia,rvsz,brvindx,psb_mpi_lpk_,icomm,minfo) if (minfo == mpi_success) & & call mpi_alltoallv(jasnd,sdsz,bsdindx,psb_mpi_lpk_,& & acoo%ja,rvsz,brvindx,psb_mpi_lpk_,icomm,minfo) if (minfo /= mpi_success) info = minfo case default info = psb_err_internal_error_ call psb_errpush(info,name,a_err='wrong A2AV alg selector') goto 9999 end select if (info /= psb_success_) then info=psb_err_from_subroutine_ call psb_errpush(info,name,a_err='alltoallv') goto 9999 end if ! ! Convert into local numbering ! if (rowcnv_) call psb_glob_to_loc(acoo%ia(1:iszr),desc_a,info,iact='I') if (colcnv_) call psb_glob_to_loc(acoo%ja(1:iszr),desc_a,info,iact='I') if (info /= psb_success_) then info=psb_err_from_subroutine_ call psb_errpush(info,name,a_err='psbglob_to_loc') goto 9999 end if l1 = 0 call acoo%set_nrows(lzero) ! irmin = huge(irmin) icmin = huge(icmin) irmax = 0 icmax = 0 Do i=1,iszr r=(acoo%ia(i)) k=(acoo%ja(i)) ! Just in case some of the conversions were out-of-range If ((r>0).and.(k>0)) Then l1=l1+1 acoo%val(l1) = acoo%val(i) acoo%ia(l1) = r acoo%ja(l1) = k irmin = min(irmin,r) irmax = max(irmax,r) icmin = min(icmin,k) icmax = max(icmax,k) End If Enddo if (rowscale_) then call acoo%set_nrows(max(irmax-irmin+1,0)) acoo%ia(1:l1) = acoo%ia(1:l1) - irmin + 1 else call acoo%set_nrows(irmax) end if if (colscale_) then call acoo%set_ncols(max(icmax-icmin+1,0)) acoo%ja(1:l1) = acoo%ja(1:l1) - icmin + 1 else call acoo%set_ncols(icmax) end if call acoo%set_nzeros(l1) call acoo%set_sorted(.false.) if (debug_level >= psb_debug_outer_)& & write(debug_unit,*) me,' ',trim(name),& & ': End data exchange',counter,l1 call move_alloc(acoo,blk%a) ! Do we expect any duplicates to appear???? call blk%cscnv(info,type=outfmt_,dupl=psb_dupl_add_) if (info /= psb_success_) then info=psb_err_from_subroutine_ call psb_errpush(info,name,a_err='psb_spcnv') goto 9999 end if Deallocate(brvindx,bsdindx,rvsz,sdsz,& & iasnd,jasnd,valsnd,stat=info) if (debug_level >= psb_debug_outer_)& & write(debug_unit,*) me,' ',trim(name),': Done' call psb_erractionrestore(err_act) return 9999 call psb_error_handler(ctxt,err_act) return End Subroutine psb_lzsphalo Subroutine psb_lz_csr_halo(a,desc_a,blk,info,rowcnv,colcnv,& & rowscale,colscale,data,outcol_glob,col_desc) use psb_base_mod, psb_protect_name => psb_lz_csr_halo #ifdef MPI_MOD use mpi #endif Implicit None #ifdef MPI_H include 'mpif.h' #endif type(psb_lz_csr_sparse_mat),Intent(in) :: a type(psb_lz_csr_sparse_mat),Intent(inout) :: blk type(psb_desc_type),intent(in), target :: desc_a integer(psb_ipk_), intent(out) :: info logical, optional, intent(in) :: rowcnv,colcnv,rowscale,colscale,outcol_glob integer(psb_ipk_), intent(in), optional :: data type(psb_desc_type),Intent(in), optional, target :: col_desc ! ...local scalars.... type(psb_ctxt_type) :: ctxt integer(psb_ipk_) :: np,me integer(psb_ipk_) :: counter,proc,i, n_el_send,n_el_recv,& & n_elem, j,ipx,mat_recv, iszs, iszr,idxs,idxr,nz,& & data_,totxch,nxs, nxr, err_act, nsnds, nrcvs integer(psb_lpk_) :: ngtz,irmax,icmax,irmin,icmin,l1, lnr, lnc, lnnz, & & r, k, ncg, jpx, idx, tot_elem integer(psb_mpk_) :: icomm, minfo integer(psb_mpk_), allocatable :: brvindx(:), & & rvsz(:), bsdindx(:),sdsz(:) integer(psb_lpk_), allocatable :: iasnd(:), jasnd(:) complex(psb_dpk_), allocatable :: valsnd(:) type(psb_lz_coo_sparse_mat), allocatable :: acoo class(psb_i_base_vect_type), pointer :: pdxv integer(psb_ipk_), allocatable :: ipdxv(:) logical :: rowcnv_,colcnv_,rowscale_,colscale_,outcol_glob_ Type(psb_desc_type), pointer :: col_desc_ character(len=5) :: outfmt_ integer(psb_ipk_) :: debug_level, debug_unit character(len=20) :: name, ch_err if(psb_get_errstatus() /= 0) return info=psb_success_ name='psb_lz_csr_sphalo' call psb_erractionsave(err_act) if (psb_errstatus_fatal()) then info = psb_err_internal_error_ ; goto 9999 end if debug_unit = psb_get_debug_unit() debug_level = psb_get_debug_level() ctxt = desc_a%get_context() icomm = desc_a%get_mpic() Call psb_info(ctxt, me, np) if (debug_level >= psb_debug_outer_) & & write(debug_unit,*) me,' ',trim(name),': Start' if (present(rowcnv)) then rowcnv_ = rowcnv else rowcnv_ = .true. endif if (present(colcnv)) then colcnv_ = colcnv else colcnv_ = .true. endif if (present(rowscale)) then rowscale_ = rowscale else rowscale_ = .false. endif if (present(colscale)) then colscale_ = colscale else colscale_ = .false. endif if (present(data)) then data_ = data else data_ = psb_comm_halo_ endif if (present(outcol_glob)) then outcol_glob_ = outcol_glob else outcol_glob_ = .false. endif if (present(col_desc)) then col_desc_ => col_desc else col_desc_ => desc_a end if Allocate(brvindx(np+1),& & rvsz(np),sdsz(np),bsdindx(np+1), acoo,stat=info) if (info /= psb_success_) then info=psb_err_alloc_dealloc_ call psb_errpush(info,name) goto 9999 end if If (debug_level >= psb_debug_outer_)& & write(debug_unit,*) me,' ',trim(name),': Data selector',data_ select case(data_) case(psb_comm_halo_,psb_comm_ext_ ) ! Do not accept OVRLAP_INDEX any longer. case default call psb_errpush(psb_err_from_subroutine_,name,a_err='wrong Data selector') goto 9999 end select sdsz(:)=0 rvsz(:)=0 l1 = 0 brvindx(1) = 0 bsdindx(1) = 0 counter=1 idx = 0 idxs = 0 idxr = 0 call desc_a%get_list(data_,pdxv,totxch,nxr,nxs,info) ipdxv = pdxv%get_vect() ! For all rows in the halo descriptor, extract the row size lnr = 0 Do proc=ipdxv(counter) if (proc == -1) exit n_el_recv = ipdxv(counter+psb_n_elem_recv_) counter = counter+n_el_recv n_el_send = ipdxv(counter+psb_n_elem_send_) tot_elem = 0 Do j=0,n_el_send-1 idx = ipdxv(counter+psb_elem_send_+j) n_elem = a%get_nz_row(idx) tot_elem = tot_elem+n_elem Enddo sdsz(proc+1) = tot_elem lnr = lnr + n_el_recv counter = counter+n_el_send+3 Enddo ! ! Exchange row sizes, so as to know sends/receives. ! This is different from the halo exchange because the ! size of the rows may vary, as opposed to fixed ! (multi) vector row size. ! call mpi_alltoall(sdsz,1,psb_mpi_mpk_,& & rvsz,1,psb_mpi_mpk_,icomm,minfo) if (info /= psb_success_) then info=psb_err_from_subroutine_ call psb_errpush(info,name,a_err='mpi_alltoall') goto 9999 end if nsnds = count(sdsz /= 0) nrcvs = count(rvsz /= 0) if (debug_level >= psb_debug_outer_)& & write(debug_unit,*) me,' ',trim(name),': Done initial alltoall',nsnds,nrcvs idxs = 0 idxr = 0 counter = 1 Do proc=ipdxv(counter) if (proc == -1) exit n_el_recv = ipdxv(counter+psb_n_elem_recv_) counter = counter+n_el_recv n_el_send = ipdxv(counter+psb_n_elem_send_) bsdindx(proc+1) = idxs idxs = idxs + sdsz(proc+1) brvindx(proc+1) = idxr idxr = idxr + rvsz(proc+1) counter = counter+n_el_send+3 Enddo iszr = sum(rvsz) mat_recv = iszr iszs = sum(sdsz) lnnz = max(iszr,iszs,ione) lnc = a%get_ncols() call acoo%allocate(lnr,lnc,lnnz) if (debug_level >= psb_debug_outer_)& & write(debug_unit,*) me,' ',trim(name),': Sizes:',acoo%get_size(),& & ' Send:',sdsz(:),' Receive:',rvsz(:) call psb_ensure_size(max(iszs,1),iasnd,info) if (info == psb_success_) call psb_ensure_size(max(iszs,1),jasnd,info) if (info == psb_success_) call psb_ensure_size(max(iszs,1),valsnd,info) if (info /= psb_success_) then info=psb_err_from_subroutine_ call psb_errpush(info,name,a_err='ensure_size') goto 9999 end if l1 = 0 ipx = 1 counter=1 idx = 0 ! ! Make sure to get all columns in csget. ! This is necessary when sphalo is used to compute a transpose, ! as opposed to just gathering halo for spspmm purposes. ! ncg = huge(ncg) tot_elem = 0 Do proc = ipdxv(counter) if (proc == -1) exit n_el_recv = ipdxv(counter+psb_n_elem_recv_) counter = counter+n_el_recv n_el_send = ipdxv(counter+psb_n_elem_send_) Do j=0,n_el_send-1 idx = ipdxv(counter+psb_elem_send_+j) call a%csget(idx,idx,ngtz,iasnd,jasnd,valsnd,info,& & append=.true.,nzin=tot_elem,jmax=ncg) if (info /= psb_success_) then info=psb_err_from_subroutine_ call psb_errpush(info,name,a_err='psb_sp_getrow') goto 9999 end if tot_elem = tot_elem+ngtz Enddo counter = counter+n_el_send+3 Enddo nz = tot_elem if (debug_level >= psb_debug_outer_)& & write(debug_unit,*) me,' ',trim(name),': Going for alltoallv',iszs,iszr if (rowcnv_) call psb_loc_to_glob(iasnd(1:nz),desc_a,info,iact='I') if (colcnv_) call psb_loc_to_glob(jasnd(1:nz),col_desc_,info,iact='I') if (info /= psb_success_) then info=psb_err_from_subroutine_ call psb_errpush(info,name,a_err='psb_loc_to_glob') goto 9999 end if select case(psb_get_sp_a2av_alg()) case(psb_sp_a2av_smpl_triad_) call psb_simple_triad_a2av(valsnd,iasnd,jasnd,sdsz,bsdindx,& & acoo%val,acoo%ia,acoo%ja,rvsz,brvindx,ctxt,info) case(psb_sp_a2av_smpl_v_) call psb_simple_a2av(valsnd,sdsz,bsdindx,& & acoo%val,rvsz,brvindx,ctxt,info) if (info == psb_success_) call psb_simple_a2av(iasnd,sdsz,bsdindx,& & acoo%ia,rvsz,brvindx,ctxt,info) if (info == psb_success_) call psb_simple_a2av(jasnd,sdsz,bsdindx,& & acoo%ja,rvsz,brvindx,ctxt,info) case(psb_sp_a2av_mpi_) call mpi_alltoallv(valsnd,sdsz,bsdindx,psb_mpi_c_dpk_,& & acoo%val,rvsz,brvindx,psb_mpi_c_dpk_,icomm,minfo) if (minfo == mpi_success) & & call mpi_alltoallv(iasnd,sdsz,bsdindx,psb_mpi_lpk_,& & acoo%ia,rvsz,brvindx,psb_mpi_lpk_,icomm,minfo) if (minfo == mpi_success) & & call mpi_alltoallv(jasnd,sdsz,bsdindx,psb_mpi_lpk_,& & acoo%ja,rvsz,brvindx,psb_mpi_lpk_,icomm,minfo) if (minfo /= mpi_success) info = minfo case default info = psb_err_internal_error_ call psb_errpush(info,name,a_err='wrong A2AV alg selector') goto 9999 end select if (info /= psb_success_) then info=psb_err_from_subroutine_ call psb_errpush(info,name,a_err='alltoallv') goto 9999 end if if (debug_level >= psb_debug_outer_)& & write(debug_unit,*) me,' ',trim(name),': Done alltoallv' ! ! Convert into local numbering ! if (rowcnv_) call psb_glob_to_loc(acoo%ia(1:iszr),desc_a,info,iact='I') ! ! This seems to be the correct output condition ! if (colcnv_.and.(.not.outcol_glob_)) & & call psb_glob_to_loc(acoo%ja(1:iszr),col_desc_,info,iact='I') if (info /= psb_success_) then info=psb_err_from_subroutine_ call psb_errpush(info,name,a_err='psbglob_to_loc') goto 9999 end if l1 = 0 call acoo%set_nrows(lzero) ! irmin = huge(irmin) icmin = huge(icmin) irmax = 0 icmax = 0 Do i=1,iszr r=(acoo%ia(i)) k=(acoo%ja(i)) ! Just in case some of the conversions were out-of-range If ((r>0).and.(k>0)) Then l1=l1+1 acoo%val(l1) = acoo%val(i) acoo%ia(l1) = r acoo%ja(l1) = k irmin = min(irmin,r) irmax = max(irmax,r) icmin = min(icmin,k) icmax = max(icmax,k) End If Enddo if (rowscale_) then call acoo%set_nrows(max(irmax-irmin+1,0)) acoo%ia(1:l1) = acoo%ia(1:l1) - irmin + 1 else call acoo%set_nrows(irmax) end if if (colscale_) then call acoo%set_ncols(max(icmax-icmin+1,0)) acoo%ja(1:l1) = acoo%ja(1:l1) - icmin + 1 else call acoo%set_ncols(icmax) end if call acoo%set_nzeros(l1) call acoo%set_sorted(.false.) if (debug_level >= psb_debug_outer_)& & write(debug_unit,*) me,' ',trim(name),& & ': End data exchange',counter,l1 call acoo%fix(info) if (info == psb_success_) call acoo%mv_to_fmt(blk,info) if (info /= psb_success_) then info=psb_err_from_subroutine_ call psb_errpush(info,name,a_err='psb_spcnv') goto 9999 end if Deallocate(brvindx,bsdindx,rvsz,sdsz,& & iasnd,jasnd,valsnd,stat=info) if (debug_level >= psb_debug_outer_)& & write(debug_unit,*) me,' ',trim(name),': Done' call psb_erractionrestore(err_act) return 9999 call psb_error_handler(ctxt,err_act) return End Subroutine psb_lz_csr_halo Subroutine psb_z_lz_csr_halo(a,desc_a,blk,info,rowcnv,colcnv,& & rowscale,colscale,data,outcol_glob,col_desc) use psb_base_mod, psb_protect_name => psb_z_lz_csr_halo #ifdef MPI_MOD use mpi #endif Implicit None #ifdef MPI_H include 'mpif.h' #endif type(psb_z_csr_sparse_mat),Intent(in) :: a type(psb_lz_csr_sparse_mat),Intent(inout) :: blk type(psb_desc_type),intent(in), target :: desc_a integer(psb_ipk_), intent(out) :: info logical, optional, intent(in) :: rowcnv,colcnv,rowscale,colscale,outcol_glob integer(psb_ipk_), intent(in), optional :: data type(psb_desc_type),Intent(in), optional, target :: col_desc ! ...local scalars.... type(psb_ctxt_type) :: ctxt integer(psb_ipk_) :: np,me integer(psb_ipk_) :: counter,proc,i, n_el_send,n_el_recv,& & n_elem, j,ipx,mat_recv, iszs, iszr,idxs,idxr,nz,& & data_,totxch,ngtz, idx, nxs, nxr, err_act, & & nsnds, nrcvs, ncg, jpx, tot_elem integer(psb_lpk_) :: irmax,icmax,irmin,icmin,l1, lnr, lnc, lnnz, & & r, k integer(psb_mpk_) :: icomm, minfo integer(psb_mpk_), allocatable :: brvindx(:), & & rvsz(:), bsdindx(:),sdsz(:) integer(psb_ipk_), allocatable :: iasnd(:), jasnd(:) integer(psb_lpk_), allocatable :: liasnd(:), ljasnd(:) complex(psb_dpk_), allocatable :: valsnd(:) type(psb_lz_coo_sparse_mat), allocatable :: acoo class(psb_i_base_vect_type), pointer :: pdxv integer(psb_ipk_), allocatable :: ipdxv(:) logical :: rowcnv_,colcnv_,rowscale_,colscale_,outcol_glob_ Type(psb_desc_type), pointer :: col_desc_ character(len=5) :: outfmt_ integer(psb_ipk_) :: debug_level, debug_unit character(len=20) :: name, ch_err if(psb_get_errstatus() /= 0) return info=psb_success_ name='psb_z_lz_csr_sphalo' call psb_erractionsave(err_act) if (psb_errstatus_fatal()) then info = psb_err_internal_error_ ; goto 9999 end if debug_unit = psb_get_debug_unit() debug_level = psb_get_debug_level() ctxt = desc_a%get_context() icomm = desc_a%get_mpic() Call psb_info(ctxt, me, np) if (debug_level >= psb_debug_outer_) & & write(debug_unit,*) me,' ',trim(name),': Start' if (present(rowcnv)) then rowcnv_ = rowcnv else rowcnv_ = .true. endif if (present(colcnv)) then colcnv_ = colcnv else colcnv_ = .true. endif if (present(rowscale)) then rowscale_ = rowscale else rowscale_ = .false. endif if (present(colscale)) then colscale_ = colscale else colscale_ = .false. endif if (present(data)) then data_ = data else data_ = psb_comm_halo_ endif if (present(outcol_glob)) then outcol_glob_ = outcol_glob else outcol_glob_ = .false. endif if (present(col_desc)) then col_desc_ => col_desc else col_desc_ => desc_a end if Allocate(brvindx(np+1),& & rvsz(np),sdsz(np),bsdindx(np+1), acoo,stat=info) if (info /= psb_success_) then info=psb_err_alloc_dealloc_ call psb_errpush(info,name) goto 9999 end if If (debug_level >= psb_debug_outer_)& & write(debug_unit,*) me,' ',trim(name),': Data selector',data_ select case(data_) case(psb_comm_halo_,psb_comm_ext_ ) ! Do not accept OVRLAP_INDEX any longer. case default call psb_errpush(psb_err_from_subroutine_,name,a_err='wrong Data selector') goto 9999 end select sdsz(:)=0 rvsz(:)=0 l1 = 0 brvindx(1) = 0 bsdindx(1) = 0 counter=1 idx = 0 idxs = 0 idxr = 0 call desc_a%get_list(data_,pdxv,totxch,nxr,nxs,info) ipdxv = pdxv%get_vect() ! For all rows in the halo descriptor, extract the row size lnr = 0 Do proc=ipdxv(counter) if (proc == -1) exit n_el_recv = ipdxv(counter+psb_n_elem_recv_) counter = counter+n_el_recv n_el_send = ipdxv(counter+psb_n_elem_send_) tot_elem = 0 Do j=0,n_el_send-1 idx = ipdxv(counter+psb_elem_send_+j) n_elem = a%get_nz_row(idx) tot_elem = tot_elem+n_elem Enddo sdsz(proc+1) = tot_elem lnr = lnr + n_el_recv counter = counter+n_el_send+3 Enddo ! ! Exchange row sizes, so as to know sends/receives. ! This is different from the halo exchange because the ! size of the rows may vary, as opposed to fixed ! (multi) vector row size. ! call mpi_alltoall(sdsz,1,psb_mpi_mpk_,& & rvsz,1,psb_mpi_mpk_,icomm,minfo) if (info /= psb_success_) then info=psb_err_from_subroutine_ call psb_errpush(info,name,a_err='mpi_alltoall') goto 9999 end if nsnds = count(sdsz /= 0) nrcvs = count(rvsz /= 0) if (debug_level >= psb_debug_outer_)& & write(debug_unit,*) me,' ',trim(name),': Done initial alltoall',nsnds,nrcvs idxs = 0 idxr = 0 counter = 1 Do proc=ipdxv(counter) if (proc == -1) exit n_el_recv = ipdxv(counter+psb_n_elem_recv_) counter = counter+n_el_recv n_el_send = ipdxv(counter+psb_n_elem_send_) bsdindx(proc+1) = idxs idxs = idxs + sdsz(proc+1) brvindx(proc+1) = idxr idxr = idxr + rvsz(proc+1) counter = counter+n_el_send+3 Enddo iszr = sum(rvsz) mat_recv = iszr iszs = sum(sdsz) lnnz = max(iszr,iszs,ione) lnc = a%get_ncols() call acoo%allocate(lnr,lnc,lnnz) if (debug_level >= psb_debug_outer_)& & write(debug_unit,*) me,' ',trim(name),': Sizes:',acoo%get_size(),& & ' Send:',sdsz(:),' Receive:',rvsz(:) call psb_ensure_size(max(iszs,1),iasnd,info) if (info == psb_success_) call psb_ensure_size(max(iszs,1),jasnd,info) if (info == psb_success_) call psb_ensure_size(max(iszs,1),liasnd,info) if (info == psb_success_) call psb_ensure_size(max(iszs,1),ljasnd,info) if (info == psb_success_) call psb_ensure_size(max(iszs,1),valsnd,info) if (info /= psb_success_) then info=psb_err_from_subroutine_ call psb_errpush(info,name,a_err='ensure_size') goto 9999 end if l1 = 0 ipx = 1 counter=1 idx = 0 ! ! Make sure to get all columns in csget. ! This is necessary when sphalo is used to compute a transpose, ! as opposed to just gathering halo for spspmm purposes. ! ncg = huge(ncg) tot_elem = 0 Do proc = ipdxv(counter) if (proc == -1) exit n_el_recv = ipdxv(counter+psb_n_elem_recv_) counter = counter+n_el_recv n_el_send = ipdxv(counter+psb_n_elem_send_) Do j=0,n_el_send-1 idx = ipdxv(counter+psb_elem_send_+j) call a%csget(idx,idx,ngtz,iasnd,jasnd,valsnd,info,& & append=.true.,nzin=tot_elem,jmax=ncg) if (info /= psb_success_) then info=psb_err_from_subroutine_ call psb_errpush(info,name,a_err='psb_sp_getrow') goto 9999 end if tot_elem = tot_elem+ngtz Enddo counter = counter+n_el_send+3 Enddo nz = tot_elem if (debug_level >= psb_debug_outer_)& & write(debug_unit,*) me,' ',trim(name),': Going for alltoallv',iszs,iszr if (rowcnv_) then call psb_loc_to_glob(iasnd(1:nz),liasnd(1:nz),desc_a,info,iact='I') else liasnd(1:nz) = iasnd(1:nz) end if if (colcnv_) then call psb_loc_to_glob(jasnd(1:nz),ljasnd(1:nz),col_desc_,info,iact='I') else ljasnd(1:nz) = jasnd(1:nz) end if if (info /= psb_success_) then info=psb_err_from_subroutine_ call psb_errpush(info,name,a_err='psb_loc_to_glob') goto 9999 end if select case(psb_get_sp_a2av_alg()) case(psb_sp_a2av_smpl_triad_) call psb_simple_triad_a2av(valsnd,liasnd,ljasnd,sdsz,bsdindx,& & acoo%val,acoo%ia,acoo%ja,rvsz,brvindx,ctxt,info) case(psb_sp_a2av_smpl_v_) call psb_simple_a2av(valsnd,sdsz,bsdindx,& & acoo%val,rvsz,brvindx,ctxt,info) if (info == psb_success_) call psb_simple_a2av(liasnd,sdsz,bsdindx,& & acoo%ia,rvsz,brvindx,ctxt,info) if (info == psb_success_) call psb_simple_a2av(ljasnd,sdsz,bsdindx,& & acoo%ja,rvsz,brvindx,ctxt,info) case(psb_sp_a2av_mpi_) call mpi_alltoallv(valsnd,sdsz,bsdindx,psb_mpi_c_dpk_,& & acoo%val,rvsz,brvindx,psb_mpi_c_dpk_,icomm,minfo) if (minfo == mpi_success) & & call mpi_alltoallv(liasnd,sdsz,bsdindx,psb_mpi_lpk_,& & acoo%ia,rvsz,brvindx,psb_mpi_lpk_,icomm,minfo) if (minfo == mpi_success) & & call mpi_alltoallv(ljasnd,sdsz,bsdindx,psb_mpi_lpk_,& & acoo%ja,rvsz,brvindx,psb_mpi_lpk_,icomm,minfo) if (minfo /= mpi_success) info = minfo case default info = psb_err_internal_error_ call psb_errpush(info,name,a_err='wrong A2AV alg selector') goto 9999 end select if (info /= psb_success_) then info=psb_err_from_subroutine_ call psb_errpush(info,name,a_err='alltoallv') goto 9999 end if if (debug_level >= psb_debug_outer_)& & write(debug_unit,*) me,' ',trim(name),': Done alltoallv' ! ! Convert into local numbering ! if (rowcnv_) call psb_glob_to_loc(acoo%ia(1:iszr),desc_a,info,iact='I') ! ! This seems to be the correct output condition ! if (colcnv_.and.(.not.outcol_glob_)) & & call psb_glob_to_loc(acoo%ja(1:iszr),col_desc_,info,iact='I') if (info /= psb_success_) then info=psb_err_from_subroutine_ call psb_errpush(info,name,a_err='psbglob_to_loc') goto 9999 end if l1 = 0 call acoo%set_nrows(lzero) ! irmin = huge(irmin) icmin = huge(icmin) irmax = 0 icmax = 0 Do i=1,iszr r=(acoo%ia(i)) k=(acoo%ja(i)) ! Just in case some of the conversions were out-of-range If ((r>0).and.(k>0)) Then l1=l1+1 acoo%val(l1) = acoo%val(i) acoo%ia(l1) = r acoo%ja(l1) = k irmin = min(irmin,r) irmax = max(irmax,r) icmin = min(icmin,k) icmax = max(icmax,k) End If Enddo if (rowscale_) then call acoo%set_nrows(max(irmax-irmin+1,0)) acoo%ia(1:l1) = acoo%ia(1:l1) - irmin + 1 else call acoo%set_nrows(irmax) end if if (colscale_) then call acoo%set_ncols(max(icmax-icmin+1,0)) acoo%ja(1:l1) = acoo%ja(1:l1) - icmin + 1 else call acoo%set_ncols(icmax) end if call acoo%set_nzeros(l1) call acoo%set_sorted(.false.) if (debug_level >= psb_debug_outer_)& & write(debug_unit,*) me,' ',trim(name),& & ': End data exchange',counter,l1 call acoo%fix(info) if (info == psb_success_) call acoo%mv_to_fmt(blk,info) if (info /= psb_success_) then info=psb_err_from_subroutine_ call psb_errpush(info,name,a_err='psb_spcnv') goto 9999 end if Deallocate(brvindx,bsdindx,rvsz,sdsz,& & iasnd,jasnd,valsnd,stat=info) if (debug_level >= psb_debug_outer_)& & write(debug_unit,*) me,' ',trim(name),': Done' call psb_erractionrestore(err_act) return 9999 call psb_error_handler(ctxt,err_act) return End Subroutine psb_z_lz_csr_halo