!!$ !!$ Parallel Sparse BLAS version 3.0 !!$ (C) Copyright 2010 !!$ Salvatore Filippone University of Rome Tor Vergata !!$ Alfredo Buttari CNRS-IRIT, Toulouse !!$ !!$ 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. !!$ !!$ module psb_iv_tools_mod use psb_const_mod interface psb_geall subroutine psb_ialloc(x, desc_a, info,n, lb) use psb_descriptor_type integer, allocatable, intent(out) :: x(:,:) type(psb_desc_type), intent(in) :: desc_a integer, intent(out) :: info integer, optional, intent(in) :: n, lb end subroutine psb_ialloc subroutine psb_iallocv(x, desc_a,info,n) use psb_descriptor_type integer, allocatable, intent(out) :: x(:) type(psb_desc_type), intent(in) :: desc_a integer, intent(out) :: info integer, optional, intent(in) :: n end subroutine psb_iallocv end interface interface psb_geasb subroutine psb_iasb(x, desc_a, info) use psb_descriptor_type type(psb_desc_type), intent(in) :: desc_a integer, allocatable, intent(inout) :: x(:,:) integer, intent(out) :: info end subroutine psb_iasb subroutine psb_iasbv(x, desc_a, info) use psb_descriptor_type type(psb_desc_type), intent(in) :: desc_a integer, allocatable, intent(inout) :: x(:) integer, intent(out) :: info end subroutine psb_iasbv end interface interface psb_gefree subroutine psb_ifree(x, desc_a, info) use psb_descriptor_type integer,allocatable, intent(inout) :: x(:,:) type(psb_desc_type), intent(in) :: desc_a integer, intent(out) :: info end subroutine psb_ifree subroutine psb_ifreev(x, desc_a, info) use psb_descriptor_type integer, allocatable, intent(inout) :: x(:) type(psb_desc_type), intent(in) :: desc_a integer, intent(out) :: info end subroutine psb_ifreev end interface interface psb_geins subroutine psb_iinsi(m,irw,val, x,desc_a,info,dupl) use psb_descriptor_type integer, intent(in) :: m type(psb_desc_type), intent(in) :: desc_a integer,intent(inout) :: x(:,:) integer, intent(in) :: irw(:) integer, intent(in) :: val(:,:) integer, intent(out) :: info integer, optional, intent(in) :: dupl end subroutine psb_iinsi subroutine psb_iinsvi(m, irw,val, x,desc_a,info,dupl) use psb_descriptor_type integer, intent(in) :: m type(psb_desc_type), intent(in) :: desc_a integer,intent(inout) :: x(:) integer, intent(in) :: irw(:) integer, intent(in) :: val(:) integer, intent(out) :: info integer, optional, intent(in) :: dupl end subroutine psb_iinsvi end interface interface psb_glob_to_loc subroutine psb_glob_to_loc2(x,y,desc_a,info,iact,owned) use psb_descriptor_type type(psb_desc_type), intent(in) :: desc_a integer,intent(in) :: x(:) integer,intent(out) :: y(:) integer, intent(out) :: info logical, intent(in), optional :: owned character, intent(in), optional :: iact end subroutine psb_glob_to_loc2 subroutine psb_glob_to_loc(x,desc_a,info,iact,owned) use psb_descriptor_type type(psb_desc_type), intent(in) :: desc_a integer,intent(inout) :: x(:) integer, intent(out) :: info logical, intent(in), optional :: owned character, intent(in), optional :: iact end subroutine psb_glob_to_loc subroutine psb_glob_to_loc2s(x,y,desc_a,info,iact,owned) use psb_descriptor_type implicit none type(psb_desc_type), intent(in) :: desc_a integer,intent(in) :: x integer,intent(out) :: y integer, intent(out) :: info character, intent(in), optional :: iact logical, intent(in), optional :: owned end subroutine psb_glob_to_loc2s subroutine psb_glob_to_locs(x,desc_a,info,iact,owned) use psb_descriptor_type implicit none type(psb_desc_type), intent(in) :: desc_a integer,intent(inout) :: x integer, intent(out) :: info character, intent(in), optional :: iact logical, intent(in), optional :: owned end subroutine psb_glob_to_locs end interface interface psb_loc_to_glob subroutine psb_loc_to_glob2(x,y,desc_a,info,iact) use psb_descriptor_type type(psb_desc_type), intent(in) :: desc_a integer,intent(in) :: x(:) integer,intent(out) :: y(:) integer, intent(out) :: info character, intent(in), optional :: iact end subroutine psb_loc_to_glob2 subroutine psb_loc_to_glob(x,desc_a,info,iact) use psb_descriptor_type type(psb_desc_type), intent(in) :: desc_a integer,intent(inout) :: x(:) integer, intent(out) :: info character, intent(in), optional :: iact end subroutine psb_loc_to_glob subroutine psb_loc_to_glob2s(x,y,desc_a,info,iact) use psb_descriptor_type implicit none type(psb_desc_type), intent(in) :: desc_a integer,intent(in) :: x integer,intent(out) :: y integer, intent(out) :: info character, intent(in), optional :: iact end subroutine psb_loc_to_glob2s subroutine psb_loc_to_globs(x,desc_a,info,iact) use psb_descriptor_type type(psb_desc_type), intent(in) :: desc_a integer,intent(inout) :: x integer, intent(out) :: info character, intent(in), optional :: iact end subroutine psb_loc_to_globs end interface interface psb_is_owned module procedure psb_is_owned end interface interface psb_is_local module procedure psb_is_local end interface interface psb_owned_index module procedure psb_owned_index, psb_owned_index_v end interface interface psb_local_index module procedure psb_local_index, psb_local_index_v end interface contains function psb_is_owned(idx,desc) use psb_descriptor_type implicit none integer, intent(in) :: idx type(psb_desc_type), intent(in) :: desc logical :: psb_is_owned logical :: res integer :: info call psb_owned_index(res,idx,desc,info) if (info /= psb_success_) res=.false. psb_is_owned = res end function psb_is_owned function psb_is_local(idx,desc) use psb_descriptor_type implicit none integer, intent(in) :: idx type(psb_desc_type), intent(in) :: desc logical :: psb_is_local logical :: res integer :: info call psb_local_index(res,idx,desc,info) if (info /= psb_success_) res=.false. psb_is_local = res end function psb_is_local subroutine psb_owned_index(res,idx,desc,info) use psb_descriptor_type implicit none integer, intent(in) :: idx type(psb_desc_type), intent(in) :: desc logical, intent(out) :: res integer, intent(out) :: info integer :: lx call psb_glob_to_loc(idx,lx,desc,info,iact='I',owned=.true.) res = (lx>0) end subroutine psb_owned_index subroutine psb_owned_index_v(res,idx,desc,info) use psb_descriptor_type implicit none integer, intent(in) :: idx(:) type(psb_desc_type), intent(in) :: desc logical, intent(out) :: res(:) integer, intent(out) :: info integer, allocatable :: lx(:) allocate(lx(size(idx)),stat=info) res=.false. if (info /= psb_success_) return call psb_glob_to_loc(idx,lx,desc,info,iact='I',owned=.true.) res = (lx>0) end subroutine psb_owned_index_v subroutine psb_local_index(res,idx,desc,info) use psb_descriptor_type implicit none integer, intent(in) :: idx type(psb_desc_type), intent(in) :: desc logical, intent(out) :: res integer, intent(out) :: info integer :: lx call psb_glob_to_loc(idx,lx,desc,info,iact='I',owned=.false.) res = (lx>0) end subroutine psb_local_index subroutine psb_local_index_v(res,idx,desc,info) use psb_descriptor_type implicit none integer, intent(in) :: idx(:) type(psb_desc_type), intent(in) :: desc logical, intent(out) :: res(:) integer, intent(out) :: info integer, allocatable :: lx(:) allocate(lx(size(idx)),stat=info) res=.false. if (info /= psb_success_) return call psb_glob_to_loc(idx,lx,desc,info,iact='I',owned=.false.) res = (lx>0) end subroutine psb_local_index_v end module psb_iv_tools_mod module psb_cd_if_tools_mod use psb_const_mod interface psb_cd_set_bld subroutine psb_cd_set_bld(desc,info) use psb_descriptor_type type(psb_desc_type), intent(inout) :: desc integer :: info end subroutine psb_cd_set_bld end interface interface psb_cd_set_ovl_bld subroutine psb_cd_set_ovl_bld(desc,info) use psb_descriptor_type type(psb_desc_type), intent(inout) :: desc integer :: info end subroutine psb_cd_set_ovl_bld end interface interface psb_cd_reinit Subroutine psb_cd_reinit(desc,info) use psb_descriptor_type Implicit None ! .. Array Arguments .. Type(psb_desc_type), Intent(inout) :: desc integer, intent(out) :: info end Subroutine psb_cd_reinit end interface interface psb_cdcpy subroutine psb_cdcpy(desc_in, desc_out, info) use psb_descriptor_type implicit none !....parameters... type(psb_desc_type), intent(in) :: desc_in type(psb_desc_type), intent(out) :: desc_out integer, intent(out) :: info end subroutine psb_cdcpy end interface interface psb_cdprt subroutine psb_cdprt(iout,desc_p,glob,short) use psb_const_mod use psb_descriptor_type implicit none type(psb_desc_type), intent(in) :: desc_p integer, intent(in) :: iout logical, intent(in), optional :: glob,short end subroutine psb_cdprt end interface interface psb_cdins subroutine psb_cdinsrc(nz,ia,ja,desc_a,info,ila,jla) use psb_descriptor_type type(psb_desc_type), intent(inout) :: desc_a integer, intent(in) :: nz,ia(:),ja(:) integer, intent(out) :: info integer, optional, intent(out) :: ila(:), jla(:) end subroutine psb_cdinsrc subroutine psb_cdinsc(nz,ja,desc,info,jla,mask) use psb_descriptor_type type(psb_desc_type), intent(inout) :: desc integer, intent(in) :: nz,ja(:) integer, intent(out) :: info integer, optional, intent(out) :: jla(:) logical, optional, target, intent(in) :: mask(:) end subroutine psb_cdinsc end interface interface psb_cdbldext Subroutine psb_cd_lstext(desc_a,in_list,desc_ov,info, mask,extype) use psb_descriptor_type Implicit None Type(psb_desc_type), Intent(in), target :: desc_a integer, intent(in) :: in_list(:) Type(psb_desc_type), Intent(out) :: desc_ov integer, intent(out) :: info logical, intent(in), optional, target :: mask(:) integer, intent(in),optional :: extype end Subroutine psb_cd_lstext end interface interface psb_cdren subroutine psb_cdren(trans,iperm,desc_a,info) use psb_descriptor_type type(psb_desc_type), intent(inout) :: desc_a integer, intent(inout) :: iperm(:) character, intent(in) :: trans integer, intent(out) :: info end subroutine psb_cdren end interface interface psb_get_overlap subroutine psb_get_ovrlap(ovrel,desc,info) use psb_descriptor_type implicit none integer, allocatable, intent(out) :: ovrel(:) type(psb_desc_type), intent(in) :: desc integer, intent(out) :: info end subroutine psb_get_ovrlap end interface interface psb_icdasb subroutine psb_icdasb(desc,info,ext_hv) use psb_descriptor_type Type(psb_desc_type), intent(inout) :: desc integer, intent(out) :: info logical, intent(in),optional :: ext_hv end subroutine psb_icdasb end interface end module psb_cd_if_tools_mod module psb_cd_tools_mod use psb_const_mod use psb_cd_if_tools_mod interface psb_cdall subroutine psb_cdall(ictxt, desc, info,mg,ng,parts,vg,vl,flag,nl,repl, globalcheck) use psb_descriptor_type implicit None include 'parts.fh' Integer, intent(in) :: mg,ng,ictxt, vg(:), vl(:),nl integer, intent(in) :: flag logical, intent(in) :: repl, globalcheck integer, intent(out) :: info type(psb_desc_type), intent(out) :: desc optional :: mg,ng,parts,vg,vl,flag,nl,repl, globalcheck end subroutine psb_cdall end interface interface psb_cdasb module procedure psb_cdasb end interface interface psb_get_boundary module procedure psb_get_boundary end interface contains subroutine psb_get_boundary(bndel,desc,info) use psb_descriptor_type use psi_mod, only : psi_crea_bnd_elem implicit none integer, allocatable, intent(out) :: bndel(:) type(psb_desc_type), intent(in) :: desc integer, intent(out) :: info call psi_crea_bnd_elem(bndel,desc,info) end subroutine psb_get_boundary subroutine psb_cdasb(desc,info) use psb_descriptor_type Type(psb_desc_type), intent(inout) :: desc integer, intent(out) :: info call psb_icdasb(desc,info,ext_hv=.false.) end subroutine psb_cdasb end module psb_cd_tools_mod module psb_base_tools_mod use psb_iv_tools_mod use psb_cd_tools_mod end module psb_base_tools_mod subroutine psb_cdall(ictxt, desc, info,mg,ng,parts,vg,vl,flag,nl,repl, globalcheck) use psb_descriptor_type use psb_serial_mod use psb_const_mod use psb_error_mod use psb_penv_mod implicit None include 'parts.fh' Integer, intent(in) :: mg,ng,ictxt, vg(:), vl(:),nl integer, intent(in) :: flag logical, intent(in) :: repl, globalcheck integer, intent(out) :: info type(psb_desc_type), intent(out) :: desc optional :: mg,ng,parts,vg,vl,flag,nl,repl, globalcheck interface subroutine psb_cdals(m, n, parts, ictxt, desc, info) use psb_descriptor_type include 'parts.fh' Integer, intent(in) :: m,n,ictxt Type(psb_desc_type), intent(out) :: desc integer, intent(out) :: info end subroutine psb_cdals subroutine psb_cdalv(v, ictxt, desc, info, flag) use psb_descriptor_type Integer, intent(in) :: ictxt, v(:) integer, intent(in), optional :: flag integer, intent(out) :: info Type(psb_desc_type), intent(out) :: desc end subroutine psb_cdalv subroutine psb_cd_inloc(v, ictxt, desc, info, globalcheck) use psb_descriptor_type implicit None Integer, intent(in) :: ictxt, v(:) integer, intent(out) :: info type(psb_desc_type), intent(out) :: desc logical, intent(in), optional :: globalcheck end subroutine psb_cd_inloc subroutine psb_cdrep(m, ictxt, desc,info) use psb_descriptor_type Integer, intent(in) :: m,ictxt Type(psb_desc_type), intent(out) :: desc integer, intent(out) :: info end subroutine psb_cdrep end interface character(len=20) :: name integer :: err_act, n_, flag_, i, me, np, nlp, nnv integer, allocatable :: itmpsz(:) if (psb_get_errstatus() /= 0) return info=psb_success_ name = 'psb_cdall' call psb_erractionsave(err_act) call psb_info(ictxt, me, np) if (count((/ present(vg),present(vl),& & present(parts),present(nl), present(repl) /)) /= 1) then info=psb_err_no_optional_arg_ call psb_errpush(info,name,a_err=" vg, vl, parts, nl, repl") goto 999 endif desc%base_desc => null() if (present(parts)) then if (.not.present(mg)) then info=psb_err_no_optional_arg_ call psb_errpush(info,name) goto 999 end if if (present(ng)) then n_ = ng else n_ = mg endif call psb_cdals(mg, n_, parts, ictxt, desc, info) else if (present(repl)) then if (.not.present(mg)) then info=psb_err_no_optional_arg_ call psb_errpush(info,name) goto 999 end if if (.not.repl) then info=psb_err_no_optional_arg_ call psb_errpush(info,name) goto 999 end if call psb_cdrep(mg, ictxt, desc, info) else if (present(vg)) then if (present(flag)) then flag_=flag else flag_=0 endif if (present(mg)) then nnv = min(mg,size(vg)) else nnv = size(vg) end if call psb_cdalv(vg(1:nnv), ictxt, desc, info, flag=flag_) else if (present(vl)) then if (present(nl)) then nnv = min(nl,size(vl)) else nnv = size(vl) end if call psb_cd_inloc(vl(1:nnv),ictxt,desc,info, globalcheck=globalcheck) else if (present(nl)) then allocate(itmpsz(0:np-1),stat=info) if (info /= psb_success_) then info = psb_err_alloc_dealloc_ call psb_errpush(info,name) goto 999 endif itmpsz = 0 itmpsz(me) = nl call psb_sum(ictxt,itmpsz) nlp=0 do i=0, me-1 nlp = nlp + itmpsz(i) end do call psb_cd_inloc((/(i,i=nlp+1,nlp+nl)/),ictxt,desc,info,globalcheck=.false.) endif if (info /= psb_success_) goto 999 call psb_erractionrestore(err_act) return 999 continue call psb_erractionrestore(err_act) if (err_act == psb_act_abort_) then call psb_error(ictxt) return end if return end subroutine psb_cdall