mirror of
https://github.com/sfilippone/psblas3.git
synced 2026-10-07 07:04:57 +00:00
New GLOBTRIL e GLOBTRIU functions
This commit is contained in:
sfilippone
@@ -435,6 +435,30 @@ Module psb_d_tools_mod
|
||||
end subroutine psb_d_simple_glob_transpose_ip
|
||||
end interface psb_glob_transpose
|
||||
|
||||
interface psb_globtril
|
||||
Subroutine psb_dglobtril(a,desc_a,b,info,diag,imin,imax,jmin,jmax)
|
||||
import
|
||||
implicit none
|
||||
Type(psb_dspmat_type), Intent(inout) :: a
|
||||
Type(psb_desc_type), Intent(inout), target :: desc_a
|
||||
Type(psb_dspmat_type), Intent(out) :: b
|
||||
integer(psb_ipk_), intent(out) :: info
|
||||
integer(psb_lpk_), intent(in), optional :: diag,imin,imax,jmin,jmax
|
||||
end Subroutine psb_dglobtril
|
||||
end interface psb_globtril
|
||||
|
||||
interface psb_globtriu
|
||||
Subroutine psb_dglobtriu(a,desc_a,b,info,diag,imin,imax,jmin,jmax)
|
||||
import
|
||||
implicit none
|
||||
Type(psb_dspmat_type), Intent(inout) :: a
|
||||
Type(psb_desc_type), Intent(inout), target :: desc_a
|
||||
Type(psb_dspmat_type), Intent(out) :: b
|
||||
integer(psb_ipk_), intent(out) :: info
|
||||
integer(psb_lpk_), intent(in), optional :: diag,imin,imax,jmin,jmax
|
||||
end Subroutine psb_dglobtriu
|
||||
end interface psb_globtriu
|
||||
|
||||
interface psb_getelem
|
||||
function psb_d_getelem(x,index,desc_a,info) result(res)
|
||||
import
|
||||
|
||||
@@ -28,6 +28,7 @@ FOBJS = psb_cdall.o psb_cdals.o psb_cdalv.o psb_cd_inloc.o psb_cdins.o psb_cdprt
|
||||
psb_s_map.o psb_d_map.o psb_c_map.o psb_z_map.o \
|
||||
psb_s_par_csr_spspmm.o psb_d_par_csr_spspmm.o psb_c_par_csr_spspmm.o psb_z_par_csr_spspmm.o \
|
||||
psb_s_glob_transpose.o psb_d_glob_transpose.o psb_c_glob_transpose.o psb_z_glob_transpose.o \
|
||||
psb_dglobtril.o psb_dglobtriu.o \
|
||||
psb_cgetelem.o psb_dgetelem.o psb_sgetelem.o psb_zgetelem.o
|
||||
|
||||
MPFOBJS = psb_icdasb.o psb_ssphalo.o psb_dsphalo.o psb_csphalo.o psb_zsphalo.o \
|
||||
|
||||
@@ -0,0 +1,110 @@
|
||||
!
|
||||
! 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 prior 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_dglobtril.f90
|
||||
!
|
||||
! Subroutine: psb_dglobtril
|
||||
!
|
||||
Subroutine psb_dglobtril(a,desc_a,b,info,diag,imin,imax,jmin,jmax)
|
||||
|
||||
use psb_base_mod, psb_protect_name => psb_dglobtril
|
||||
use psi_mod
|
||||
|
||||
Implicit None
|
||||
|
||||
! .. Array Arguments ..
|
||||
Type(psb_dspmat_type), Intent(inout) :: a
|
||||
Type(psb_desc_type), Intent(inout), target :: desc_a
|
||||
Type(psb_dspmat_type), Intent(out) :: b
|
||||
integer(psb_ipk_), intent(out) :: info
|
||||
integer(psb_lpk_), intent(in), optional :: diag,imin,imax,jmin,jmax
|
||||
|
||||
! .. Local Scalars ..
|
||||
integer(psb_ipk_) :: i, j, err_act,m,&
|
||||
& nz
|
||||
integer(psb_lpk_) :: gidx, lnz
|
||||
type(psb_ctxt_type) :: ctxt
|
||||
integer(psb_ipk_) :: me, np
|
||||
integer(psb_mpk_) :: icomm, minfo
|
||||
|
||||
type(psb_d_coo_sparse_mat) :: dtcoo
|
||||
type(psb_ld_coo_sparse_mat) :: ldtcoo
|
||||
type(psb_ldspmat_type) :: ldglob,ldtril
|
||||
integer(psb_ipk_) :: debug_level, debug_unit
|
||||
integer(psb_ipk_) :: ierr(5)
|
||||
character(len=20) :: name, ch_err
|
||||
|
||||
name='psb_dglobtril'
|
||||
info = psb_success_
|
||||
if (psb_errstatus_fatal()) return
|
||||
call psb_erractionsave(err_act)
|
||||
debug_unit = psb_get_debug_unit()
|
||||
debug_level = psb_get_debug_level()
|
||||
|
||||
if (.not.desc_a%is_ok()) then
|
||||
info = psb_err_invalid_cd_state_
|
||||
call psb_errpush(info,name)
|
||||
goto 9999
|
||||
end if
|
||||
ctxt = desc_a%get_context()
|
||||
icomm = ctxt%get_mpic()
|
||||
Call psb_info(ctxt, me, np)
|
||||
|
||||
If (debug_level >= psb_debug_outer_) &
|
||||
& Write(debug_unit,*) me,' ',trim(name),&
|
||||
& ': start',diag
|
||||
|
||||
call a%a%cp_to_lcoo(ldtcoo,info)
|
||||
lnz = ldtcoo%get_nzeros()
|
||||
call desc_a%l2gip(ldtcoo%ia(1:lnz),info,owned=.false.)
|
||||
call desc_a%l2gip(ldtcoo%ja(1:lnz),info,owned=.false.)
|
||||
call ldglob%mv_from(ldtcoo)
|
||||
call ldglob%tril(ldtril,info,&
|
||||
& diag=diag,imin=imin,imax=imax,jmin=jmin,jmax=jmax)
|
||||
call ldglob%free()
|
||||
call ldglob%mv_to(ldtcoo)
|
||||
lnz = ldtcoo%get_nzeros()
|
||||
call desc_a%g2lip(ldtcoo%ia(1:lnz),info,owned=.false.)
|
||||
call desc_a%g2lip(ldtcoo%ja(1:lnz),info,owned=.false.)
|
||||
call b%mv_from_lb(ldtcoo)
|
||||
|
||||
if (debug_level >= psb_debug_outer_) &
|
||||
& write(debug_unit,*) me,' ',trim(name),': end'
|
||||
|
||||
call psb_erractionrestore(err_act)
|
||||
return
|
||||
|
||||
9999 call psb_error_handler(ctxt,err_act)
|
||||
|
||||
return
|
||||
|
||||
End Subroutine psb_dglobtril
|
||||
|
||||
@@ -0,0 +1,110 @@
|
||||
!
|
||||
! 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 prior 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_dglobtriu.f90
|
||||
!
|
||||
! Subroutine: psb_dglobtriu
|
||||
!
|
||||
Subroutine psb_dglobtriu(a,desc_a,b,info,diag,imin,imax,jmin,jmax)
|
||||
|
||||
use psb_base_mod, psb_protect_name => psb_dglobtriu
|
||||
use psi_mod
|
||||
|
||||
Implicit None
|
||||
|
||||
! .. Array Arguments ..
|
||||
Type(psb_dspmat_type), Intent(inout) :: a
|
||||
Type(psb_desc_type), Intent(inout), target :: desc_a
|
||||
Type(psb_dspmat_type), Intent(out) :: b
|
||||
integer(psb_ipk_), intent(out) :: info
|
||||
integer(psb_lpk_), intent(in), optional :: diag,imin,imax,jmin,jmax
|
||||
|
||||
! .. Local Scalars ..
|
||||
integer(psb_ipk_) :: i, j, err_act,m,&
|
||||
& nz
|
||||
integer(psb_lpk_) :: gidx, lnz
|
||||
type(psb_ctxt_type) :: ctxt
|
||||
integer(psb_ipk_) :: me, np
|
||||
integer(psb_mpk_) :: icomm, minfo
|
||||
|
||||
type(psb_d_coo_sparse_mat) :: dtcoo
|
||||
type(psb_ld_coo_sparse_mat) :: ldtcoo
|
||||
type(psb_ldspmat_type) :: ldglob,ldtriu
|
||||
integer(psb_ipk_) :: debug_level, debug_unit
|
||||
integer(psb_ipk_) :: ierr(5)
|
||||
character(len=20) :: name, ch_err
|
||||
|
||||
name='psb_dglobtriu'
|
||||
info = psb_success_
|
||||
if (psb_errstatus_fatal()) return
|
||||
call psb_erractionsave(err_act)
|
||||
debug_unit = psb_get_debug_unit()
|
||||
debug_level = psb_get_debug_level()
|
||||
|
||||
if (.not.desc_a%is_ok()) then
|
||||
info = psb_err_invalid_cd_state_
|
||||
call psb_errpush(info,name)
|
||||
goto 9999
|
||||
end if
|
||||
ctxt = desc_a%get_context()
|
||||
icomm = ctxt%get_mpic()
|
||||
Call psb_info(ctxt, me, np)
|
||||
|
||||
If (debug_level >= psb_debug_outer_) &
|
||||
& Write(debug_unit,*) me,' ',trim(name),&
|
||||
& ': start',diag
|
||||
|
||||
call a%a%cp_to_lcoo(ldtcoo,info)
|
||||
lnz = ldtcoo%get_nzeros()
|
||||
call desc_a%l2gip(ldtcoo%ia(1:lnz),info,owned=.false.)
|
||||
call desc_a%l2gip(ldtcoo%ja(1:lnz),info,owned=.false.)
|
||||
call ldglob%mv_from(ldtcoo)
|
||||
call ldglob%triu(ldtriu,info,&
|
||||
& diag=diag,imin=imin,imax=imax,jmin=jmin,jmax=jmax)
|
||||
call ldglob%free()
|
||||
call ldglob%mv_to(ldtcoo)
|
||||
lnz = ldtcoo%get_nzeros()
|
||||
call desc_a%g2lip(ldtcoo%ia(1:lnz),info,owned=.false.)
|
||||
call desc_a%g2lip(ldtcoo%ja(1:lnz),info,owned=.false.)
|
||||
call b%mv_from_lb(ldtcoo)
|
||||
|
||||
if (debug_level >= psb_debug_outer_) &
|
||||
& write(debug_unit,*) me,' ',trim(name),': end'
|
||||
|
||||
call psb_erractionrestore(err_act)
|
||||
return
|
||||
|
||||
9999 call psb_error_handler(ctxt,err_act)
|
||||
|
||||
return
|
||||
|
||||
End Subroutine psb_dglobtriu
|
||||
|
||||
Reference in New Issue
Block a user