psblas3-type-index:
Salvatore Filippone
13 years ago
commit
dacc4e566d
@ -0,0 +1,213 @@
|
||||
!!$
|
||||
!!$ Parallel Sparse BLAS version 3.0
|
||||
!!$ (C) Copyright 2006, 2007, 2008, 2009, 2010, 2012
|
||||
!!$ 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_cd_tools_mod
|
||||
|
||||
use psb_const_mod
|
||||
use psb_descriptor_type
|
||||
use psb_gen_block_map_mod
|
||||
use psb_list_map_mod
|
||||
use psb_glist_map_mod
|
||||
use psb_hash_map_mod
|
||||
use psb_repl_map_mod
|
||||
|
||||
interface psb_cd_set_bld
|
||||
subroutine psb_cd_set_bld(desc,info)
|
||||
import :: psb_ipk_, psb_desc_type
|
||||
type(psb_desc_type), intent(inout) :: desc
|
||||
integer(psb_ipk_) :: info
|
||||
end subroutine psb_cd_set_bld
|
||||
end interface
|
||||
|
||||
interface psb_cd_set_ovl_bld
|
||||
subroutine psb_cd_set_ovl_bld(desc,info)
|
||||
import :: psb_ipk_, psb_desc_type
|
||||
type(psb_desc_type), intent(inout) :: desc
|
||||
integer(psb_ipk_) :: info
|
||||
end subroutine psb_cd_set_ovl_bld
|
||||
end interface
|
||||
|
||||
interface psb_cd_reinit
|
||||
Subroutine psb_cd_reinit(desc,info)
|
||||
import :: psb_ipk_, psb_desc_type
|
||||
Implicit None
|
||||
|
||||
! .. Array Arguments ..
|
||||
Type(psb_desc_type), Intent(inout) :: desc
|
||||
integer(psb_ipk_), intent(out) :: info
|
||||
end Subroutine psb_cd_reinit
|
||||
end interface
|
||||
|
||||
interface psb_cdcpy
|
||||
subroutine psb_cdcpy(desc_in, desc_out, info)
|
||||
import :: psb_ipk_, psb_desc_type
|
||||
|
||||
implicit none
|
||||
!....parameters...
|
||||
|
||||
type(psb_desc_type), intent(in) :: desc_in
|
||||
type(psb_desc_type), intent(out) :: desc_out
|
||||
integer(psb_ipk_), intent(out) :: info
|
||||
end subroutine psb_cdcpy
|
||||
end interface
|
||||
|
||||
|
||||
interface psb_cdprt
|
||||
subroutine psb_cdprt(iout,desc_p,glob,short)
|
||||
import :: psb_ipk_, psb_desc_type
|
||||
implicit none
|
||||
type(psb_desc_type), intent(in) :: desc_p
|
||||
integer(psb_ipk_), 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)
|
||||
import :: psb_ipk_, psb_desc_type
|
||||
type(psb_desc_type), intent(inout) :: desc_a
|
||||
integer(psb_ipk_), intent(in) :: nz,ia(:),ja(:)
|
||||
integer(psb_ipk_), intent(out) :: info
|
||||
integer(psb_ipk_), optional, intent(out) :: ila(:), jla(:)
|
||||
end subroutine psb_cdinsrc
|
||||
subroutine psb_cdinsc(nz,ja,desc,info,jla,mask,lidx)
|
||||
import :: psb_ipk_, psb_desc_type
|
||||
type(psb_desc_type), intent(inout) :: desc
|
||||
integer(psb_ipk_), intent(in) :: nz,ja(:)
|
||||
integer(psb_ipk_), intent(out) :: info
|
||||
integer(psb_ipk_), optional, intent(out) :: jla(:)
|
||||
logical, optional, target, intent(in) :: mask(:)
|
||||
integer(psb_ipk_), intent(in), optional :: lidx(:)
|
||||
end subroutine psb_cdinsc
|
||||
end interface
|
||||
|
||||
interface psb_cdbldext
|
||||
Subroutine psb_cd_lstext(desc_a,in_list,desc_ov,info, mask,extype)
|
||||
import :: psb_ipk_, psb_desc_type
|
||||
Implicit None
|
||||
Type(psb_desc_type), Intent(in), target :: desc_a
|
||||
integer(psb_ipk_), intent(in) :: in_list(:)
|
||||
Type(psb_desc_type), Intent(out) :: desc_ov
|
||||
integer(psb_ipk_), intent(out) :: info
|
||||
logical, intent(in), optional, target :: mask(:)
|
||||
integer(psb_ipk_), intent(in),optional :: extype
|
||||
end Subroutine psb_cd_lstext
|
||||
end interface
|
||||
|
||||
|
||||
interface psb_cdren
|
||||
subroutine psb_cdren(trans,iperm,desc_a,info)
|
||||
import :: psb_ipk_, psb_desc_type
|
||||
type(psb_desc_type), intent(inout) :: desc_a
|
||||
integer(psb_ipk_), intent(inout) :: iperm(:)
|
||||
character, intent(in) :: trans
|
||||
integer(psb_ipk_), intent(out) :: info
|
||||
end subroutine psb_cdren
|
||||
end interface
|
||||
|
||||
interface psb_get_overlap
|
||||
subroutine psb_get_ovrlap(ovrel,desc,info)
|
||||
import :: psb_ipk_, psb_desc_type
|
||||
implicit none
|
||||
integer(psb_ipk_), allocatable, intent(out) :: ovrel(:)
|
||||
type(psb_desc_type), intent(in) :: desc
|
||||
integer(psb_ipk_), intent(out) :: info
|
||||
end subroutine psb_get_ovrlap
|
||||
end interface
|
||||
|
||||
interface psb_icdasb
|
||||
subroutine psb_icdasb(desc,info,ext_hv)
|
||||
import :: psb_ipk_, psb_desc_type
|
||||
Type(psb_desc_type), intent(inout) :: desc
|
||||
integer(psb_ipk_), intent(out) :: info
|
||||
logical, intent(in),optional :: ext_hv
|
||||
end subroutine psb_icdasb
|
||||
end interface
|
||||
|
||||
interface psb_cdall
|
||||
|
||||
subroutine psb_cdall(ictxt, desc, info,mg,ng,parts,vg,vl,flag,nl,repl,&
|
||||
& globalcheck,lidx)
|
||||
import :: psb_ipk_, psb_desc_type, psb_parts
|
||||
implicit None
|
||||
procedure(psb_parts) :: parts
|
||||
integer(psb_ipk_), intent(in) :: mg,ng,ictxt, vg(:), vl(:),nl,lidx(:)
|
||||
integer(psb_ipk_), intent(in) :: flag
|
||||
logical, intent(in) :: repl, globalcheck
|
||||
integer(psb_ipk_), intent(out) :: info
|
||||
type(psb_desc_type), intent(out) :: desc
|
||||
optional :: mg,ng,parts,vg,vl,flag,nl,repl, globalcheck,lidx
|
||||
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
|
||||
|
||||
interface
|
||||
subroutine psb_cd_switch_ovl_indxmap(desc,info)
|
||||
import :: psb_ipk_, psb_desc_type
|
||||
implicit None
|
||||
include 'parts.fh'
|
||||
type(psb_desc_type), intent(inout) :: desc
|
||||
integer(psb_ipk_), intent(out) :: info
|
||||
end subroutine psb_cd_switch_ovl_indxmap
|
||||
end interface
|
||||
|
||||
contains
|
||||
|
||||
subroutine psb_get_boundary(bndel,desc,info)
|
||||
use psi_mod, only : psi_crea_bnd_elem
|
||||
implicit none
|
||||
integer(psb_ipk_), allocatable, intent(out) :: bndel(:)
|
||||
type(psb_desc_type), intent(in) :: desc
|
||||
integer(psb_ipk_), intent(out) :: info
|
||||
|
||||
call psi_crea_bnd_elem(bndel,desc,info)
|
||||
|
||||
end subroutine psb_get_boundary
|
||||
|
||||
subroutine psb_cdasb(desc,info)
|
||||
|
||||
Type(psb_desc_type), intent(inout) :: desc
|
||||
integer(psb_ipk_), intent(out) :: info
|
||||
|
||||
call psb_icdasb(desc,info,ext_hv=.false.)
|
||||
end subroutine psb_cdasb
|
||||
|
||||
end module psb_cd_tools_mod
|
||||
|
||||
|
||||
@ -0,0 +1,820 @@
|
||||
!!$
|
||||
!!$ Parallel Sparse BLAS version 3.0
|
||||
!!$ (C) Copyright 2006, 2007, 2008, 2009, 2010, 2012
|
||||
!!$ 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.
|
||||
!!$
|
||||
!!$
|
||||
!
|
||||
! package: psb_i_base_vect_mod
|
||||
!
|
||||
! This module contains the definition of the psb_i_base_vect type which
|
||||
! is a container for dense vectors.
|
||||
! This is encapsulated instead of being just a simple array to allow for
|
||||
! more complicated situations, such as GPU programming, where the memory
|
||||
! area we are interested in is not easily accessible from the host/Fortran
|
||||
! side. It is also meant to be encapsulated in an outer type, to allow
|
||||
! runtime switching as per the STATE design pattern, similar to the
|
||||
! sparse matrix types.
|
||||
!
|
||||
!
|
||||
module psb_i_base_vect_mod
|
||||
|
||||
use psb_const_mod
|
||||
use psb_error_mod
|
||||
|
||||
|
||||
!> \namespace psb_base_mod \class psb_i_base_vect_type
|
||||
!! The psb_i_base_vect_type
|
||||
!! defines a middle level integer(psb_ipk_) encapsulated dense vector.
|
||||
!! The encapsulation is needed, in place of a simple array, to allow
|
||||
!! for complicated situations, such as GPU programming, where the memory
|
||||
!! area we are interested in is not easily accessible from the host/Fortran
|
||||
!! side. It is also meant to be encapsulated in an outer type, to allow
|
||||
!! runtime switching as per the STATE design pattern, similar to the
|
||||
!! sparse matrix types.
|
||||
!!
|
||||
type psb_i_base_vect_type
|
||||
integer(psb_ipk_), allocatable :: v(:)
|
||||
contains
|
||||
!
|
||||
! Constructors/allocators
|
||||
!
|
||||
procedure, pass(x) :: bld_x => i_base_bld_x
|
||||
procedure, pass(x) :: bld_n => i_base_bld_n
|
||||
generic, public :: bld => bld_x, bld_n
|
||||
procedure, pass(x) :: all => i_base_all
|
||||
procedure, pass(x) :: mold => i_base_mold
|
||||
!
|
||||
! Insert/set. Assembly and free.
|
||||
! Assembly does almost nothing here, but is important
|
||||
! in derived classes.
|
||||
!
|
||||
procedure, pass(x) :: ins => i_base_ins
|
||||
procedure, pass(x) :: zero => i_base_zero
|
||||
procedure, pass(x) :: asb => i_base_asb
|
||||
procedure, pass(x) :: free => i_base_free
|
||||
!
|
||||
! Sync: centerpiece of handling of external storage.
|
||||
! Any derived class having extra storage upon sync
|
||||
! will guarantee that both fortran/host side and
|
||||
! external side contain the same data. The base
|
||||
! version is only a placeholder.
|
||||
!
|
||||
procedure, pass(x) :: sync => i_base_sync
|
||||
procedure, pass(x) :: is_host => i_base_is_host
|
||||
procedure, pass(x) :: is_dev => i_base_is_dev
|
||||
procedure, pass(x) :: is_sync => i_base_is_sync
|
||||
procedure, pass(x) :: set_host => i_base_set_host
|
||||
procedure, pass(x) :: set_dev => i_base_set_dev
|
||||
procedure, pass(x) :: set_sync => i_base_set_sync
|
||||
|
||||
!
|
||||
! Basic info
|
||||
procedure, pass(x) :: get_nrows => i_base_get_nrows
|
||||
procedure, pass(x) :: sizeof => i_base_sizeof
|
||||
!
|
||||
! Set/get data from/to an external array; also
|
||||
! overload assignment.
|
||||
!
|
||||
procedure, pass(x) :: get_vect => i_base_get_vect
|
||||
procedure, pass(x) :: set_scal => i_base_set_scal
|
||||
procedure, pass(x) :: set_vect => i_base_set_vect
|
||||
generic, public :: set => set_vect, set_scal
|
||||
|
||||
!
|
||||
! Dot product and AXPBY
|
||||
!
|
||||
procedure, pass(x) :: dot_v => i_base_dot_v
|
||||
procedure, pass(x) :: dot_a => i_base_dot_a
|
||||
generic, public :: dot => dot_v, dot_a
|
||||
procedure, pass(y) :: axpby_v => i_base_axpby_v
|
||||
procedure, pass(y) :: axpby_a => i_base_axpby_a
|
||||
generic, public :: axpby => axpby_v, axpby_a
|
||||
!
|
||||
! Vector by vector multiplication. Need all variants
|
||||
! to handle multiple requirements from preconditioners
|
||||
!
|
||||
procedure, pass(y) :: mlt_v => i_base_mlt_v
|
||||
procedure, pass(y) :: mlt_a => i_base_mlt_a
|
||||
procedure, pass(z) :: mlt_a_2 => i_base_mlt_a_2
|
||||
procedure, pass(z) :: mlt_v_2 => i_base_mlt_v_2
|
||||
procedure, pass(z) :: mlt_va => i_base_mlt_va
|
||||
procedure, pass(z) :: mlt_av => i_base_mlt_av
|
||||
generic, public :: mlt => mlt_v, mlt_a, mlt_a_2, mlt_v_2, mlt_av, mlt_va
|
||||
!
|
||||
! Scaling and norms
|
||||
!
|
||||
procedure, pass(x) :: scal => i_base_scal
|
||||
procedure, pass(x) :: nrm2 => i_base_nrm2
|
||||
procedure, pass(x) :: amax => i_base_amax
|
||||
procedure, pass(x) :: asum => i_base_asum
|
||||
!
|
||||
! Gather/scatter. These are needed for MPI interfacing.
|
||||
! May have to be reworked.
|
||||
!
|
||||
procedure, pass(x) :: gthab => i_base_gthab
|
||||
procedure, pass(x) :: gthzv => i_base_gthzv
|
||||
generic, public :: gth => gthab, gthzv
|
||||
procedure, pass(y) :: sctb => i_base_sctb
|
||||
generic, public :: sct => sctb
|
||||
end type psb_i_base_vect_type
|
||||
|
||||
public :: psb_i_base_vect
|
||||
private :: constructor, size_const
|
||||
interface psb_i_base_vect
|
||||
module procedure constructor, size_const
|
||||
end interface psb_i_base_vect
|
||||
|
||||
contains
|
||||
|
||||
!
|
||||
! Constructors.
|
||||
!
|
||||
|
||||
function constructor(x) result(this)
|
||||
integer(psb_ipk_) :: x(:)
|
||||
type(psb_i_base_vect_type) :: this
|
||||
integer(psb_ipk_) :: info
|
||||
|
||||
this%v = x
|
||||
call this%asb(size(x,kind=psb_ipk_),info)
|
||||
end function constructor
|
||||
|
||||
|
||||
function size_const(n) result(this)
|
||||
integer(psb_ipk_), intent(in) :: n
|
||||
type(psb_i_base_vect_type) :: this
|
||||
integer(psb_ipk_) :: info
|
||||
|
||||
call this%asb(n,info)
|
||||
|
||||
end function size_const
|
||||
|
||||
!
|
||||
! Build from a sample
|
||||
!
|
||||
|
||||
subroutine i_base_bld_x(x,this)
|
||||
use psb_realloc_mod
|
||||
integer(psb_ipk_), intent(in) :: this(:)
|
||||
class(psb_i_base_vect_type), intent(inout) :: x
|
||||
integer(psb_ipk_) :: info
|
||||
|
||||
call psb_realloc(size(this),x%v,info)
|
||||
if (info /= 0) then
|
||||
call psb_errpush(psb_err_alloc_dealloc_,'base_vect_bld')
|
||||
return
|
||||
end if
|
||||
x%v(:) = this(:)
|
||||
|
||||
end subroutine i_base_bld_x
|
||||
|
||||
!
|
||||
! Create with size, but no initialization
|
||||
!
|
||||
subroutine i_base_bld_n(x,n)
|
||||
use psb_realloc_mod
|
||||
integer(psb_ipk_), intent(in) :: n
|
||||
class(psb_i_base_vect_type), intent(inout) :: x
|
||||
integer(psb_ipk_) :: info
|
||||
|
||||
call psb_realloc(n,x%v,info)
|
||||
call x%asb(n,info)
|
||||
|
||||
end subroutine i_base_bld_n
|
||||
|
||||
subroutine i_base_all(n, x, info)
|
||||
use psi_serial_mod
|
||||
use psb_realloc_mod
|
||||
implicit none
|
||||
integer(psb_ipk_), intent(in) :: n
|
||||
class(psb_i_base_vect_type), intent(out) :: x
|
||||
integer(psb_ipk_), intent(out) :: info
|
||||
|
||||
call psb_realloc(n,x%v,info)
|
||||
|
||||
end subroutine i_base_all
|
||||
|
||||
subroutine i_base_mold(x, y, info)
|
||||
use psi_serial_mod
|
||||
use psb_realloc_mod
|
||||
implicit none
|
||||
class(psb_i_base_vect_type), intent(in) :: x
|
||||
class(psb_i_base_vect_type), intent(out), allocatable :: y
|
||||
integer(psb_ipk_), intent(out) :: info
|
||||
|
||||
allocate(psb_i_base_vect_type :: y, stat=info)
|
||||
|
||||
end subroutine i_base_mold
|
||||
|
||||
!
|
||||
! Insert a bunch of values at specified positions.
|
||||
!
|
||||
subroutine i_base_ins(n,irl,val,dupl,x,info)
|
||||
use psi_serial_mod
|
||||
implicit none
|
||||
class(psb_i_base_vect_type), intent(inout) :: x
|
||||
integer(psb_ipk_), intent(in) :: n, dupl
|
||||
integer(psb_ipk_), intent(in) :: irl(:)
|
||||
integer(psb_ipk_), intent(in) :: val(:)
|
||||
integer(psb_ipk_), intent(out) :: info
|
||||
|
||||
integer(psb_ipk_) :: i
|
||||
|
||||
info = 0
|
||||
if (psb_errstatus_fatal()) return
|
||||
|
||||
if (.not.allocated(x%v)) then
|
||||
info = psb_err_invalid_vect_state_
|
||||
else if (n > min(size(irl),size(val))) then
|
||||
info = psb_err_invalid_input_
|
||||
|
||||
else
|
||||
select case(dupl)
|
||||
case(psb_dupl_ovwrt_)
|
||||
do i = 1, n
|
||||
!loop over all val's rows
|
||||
|
||||
! row actual block row
|
||||
if (irl(i) > 0) then
|
||||
! this row belongs to me
|
||||
! copy i-th row of block val in x
|
||||
x%v(irl(i)) = val(i)
|
||||
end if
|
||||
enddo
|
||||
|
||||
case(psb_dupl_add_)
|
||||
|
||||
do i = 1, n
|
||||
!loop over all val's rows
|
||||
|
||||
if (irl(i) > 0) then
|
||||
! this row belongs to me
|
||||
! copy i-th row of block val in x
|
||||
x%v(irl(i)) = x%v(irl(i)) + val(i)
|
||||
end if
|
||||
enddo
|
||||
|
||||
case default
|
||||
info = 321
|
||||
!!$ call psb_errpush(info,name)
|
||||
!!$ goto 9999
|
||||
end select
|
||||
end if
|
||||
if (info /= 0) then
|
||||
call psb_errpush(info,'base_vect_ins')
|
||||
return
|
||||
end if
|
||||
|
||||
end subroutine i_base_ins
|
||||
|
||||
!
|
||||
subroutine i_base_zero(x)
|
||||
use psi_serial_mod
|
||||
implicit none
|
||||
class(psb_i_base_vect_type), intent(inout) :: x
|
||||
|
||||
if (allocated(x%v)) x%v=izero
|
||||
|
||||
end subroutine i_base_zero
|
||||
|
||||
|
||||
!
|
||||
! Assembly.
|
||||
! For derived classes: after this the vector
|
||||
! storage is supposed to be in sync.
|
||||
!
|
||||
|
||||
subroutine i_base_asb(n, x, info)
|
||||
use psi_serial_mod
|
||||
use psb_realloc_mod
|
||||
implicit none
|
||||
integer(psb_ipk_), intent(in) :: n
|
||||
class(psb_i_base_vect_type), intent(inout) :: x
|
||||
integer(psb_ipk_), intent(out) :: info
|
||||
|
||||
if (x%get_nrows() < n) &
|
||||
& call psb_realloc(n,x%v,info)
|
||||
if (info /= 0) &
|
||||
& call psb_errpush(psb_err_alloc_dealloc_,'vect_asb')
|
||||
|
||||
end subroutine i_base_asb
|
||||
|
||||
|
||||
subroutine i_base_free(x, info)
|
||||
use psi_serial_mod
|
||||
use psb_realloc_mod
|
||||
implicit none
|
||||
class(psb_i_base_vect_type), intent(inout) :: x
|
||||
integer(psb_ipk_), intent(out) :: info
|
||||
|
||||
info = 0
|
||||
if (allocated(x%v)) deallocate(x%v, stat=info)
|
||||
if (info /= 0) call &
|
||||
& psb_errpush(psb_err_alloc_dealloc_,'vect_free')
|
||||
|
||||
end subroutine i_base_free
|
||||
|
||||
|
||||
|
||||
!
|
||||
! The base version of SYNC & friends does nothing, it's just
|
||||
! a placeholder.
|
||||
!
|
||||
subroutine i_base_sync(x)
|
||||
implicit none
|
||||
class(psb_i_base_vect_type), intent(inout) :: x
|
||||
|
||||
end subroutine i_base_sync
|
||||
|
||||
|
||||
subroutine i_base_set_host(x)
|
||||
implicit none
|
||||
class(psb_i_base_vect_type), intent(inout) :: x
|
||||
|
||||
end subroutine i_base_set_host
|
||||
|
||||
subroutine i_base_set_dev(x)
|
||||
implicit none
|
||||
class(psb_i_base_vect_type), intent(inout) :: x
|
||||
|
||||
end subroutine i_base_set_dev
|
||||
|
||||
subroutine i_base_set_sync(x)
|
||||
implicit none
|
||||
class(psb_i_base_vect_type), intent(inout) :: x
|
||||
|
||||
end subroutine i_base_set_sync
|
||||
|
||||
function i_base_is_dev(x) result(res)
|
||||
implicit none
|
||||
class(psb_i_base_vect_type), intent(in) :: x
|
||||
logical :: res
|
||||
|
||||
res = .false.
|
||||
end function i_base_is_dev
|
||||
|
||||
function i_base_is_host(x) result(res)
|
||||
implicit none
|
||||
class(psb_i_base_vect_type), intent(in) :: x
|
||||
logical :: res
|
||||
|
||||
res = .true.
|
||||
end function i_base_is_host
|
||||
|
||||
function i_base_is_sync(x) result(res)
|
||||
implicit none
|
||||
class(psb_i_base_vect_type), intent(in) :: x
|
||||
logical :: res
|
||||
|
||||
res = .true.
|
||||
end function i_base_is_sync
|
||||
|
||||
|
||||
!
|
||||
! Size info.
|
||||
!
|
||||
|
||||
function i_base_get_nrows(x) result(res)
|
||||
implicit none
|
||||
class(psb_i_base_vect_type), intent(in) :: x
|
||||
integer(psb_ipk_) :: res
|
||||
|
||||
res = 0
|
||||
if (allocated(x%v)) res = size(x%v)
|
||||
|
||||
end function i_base_get_nrows
|
||||
|
||||
function i_base_sizeof(x) result(res)
|
||||
implicit none
|
||||
class(psb_i_base_vect_type), intent(in) :: x
|
||||
integer(psb_long_int_k_) :: res
|
||||
|
||||
! Force 8-byte integers.
|
||||
res = (1_psb_long_int_k_ * psb_sizeof_int) * x%get_nrows()
|
||||
|
||||
end function i_base_sizeof
|
||||
|
||||
|
||||
!
|
||||
! Two versions of extracting an array: one of them
|
||||
! overload the assignment.
|
||||
!
|
||||
|
||||
function i_base_get_vect(x) result(res)
|
||||
class(psb_i_base_vect_type), intent(inout) :: x
|
||||
integer(psb_ipk_), allocatable :: res(:)
|
||||
integer(psb_ipk_) :: info
|
||||
|
||||
if (.not.allocated(x%v)) return
|
||||
call x%sync()
|
||||
allocate(res(x%get_nrows()),stat=info)
|
||||
if (info /= 0) then
|
||||
call psb_errpush(psb_err_alloc_dealloc_,'base_get_vect')
|
||||
return
|
||||
end if
|
||||
res(:) = x%v(:)
|
||||
end function i_base_get_vect
|
||||
|
||||
!
|
||||
! Reset all values
|
||||
!
|
||||
subroutine i_base_set_scal(x,val)
|
||||
class(psb_i_base_vect_type), intent(inout) :: x
|
||||
integer(psb_ipk_), intent(in) :: val
|
||||
|
||||
integer(psb_ipk_) :: info
|
||||
x%v = val
|
||||
|
||||
end subroutine i_base_set_scal
|
||||
|
||||
subroutine i_base_set_vect(x,val)
|
||||
class(psb_i_base_vect_type), intent(inout) :: x
|
||||
integer(psb_ipk_), intent(in) :: val(:)
|
||||
integer(psb_ipk_) :: nr
|
||||
integer(psb_ipk_) :: info
|
||||
|
||||
if (allocated(x%v)) then
|
||||
nr = min(size(x%v),size(val))
|
||||
x%v(1:nr) = val(1:nr)
|
||||
else
|
||||
x%v = val
|
||||
end if
|
||||
|
||||
end subroutine i_base_set_vect
|
||||
|
||||
!
|
||||
! Dot products
|
||||
!
|
||||
function i_base_dot_v(n,x,y) result(res)
|
||||
implicit none
|
||||
class(psb_i_base_vect_type), intent(inout) :: x, y
|
||||
integer(psb_ipk_), intent(in) :: n
|
||||
integer(psb_ipk_) :: res
|
||||
integer(psb_ipk_), external :: idot
|
||||
|
||||
res = izero
|
||||
!
|
||||
! Note: this is the base implementation.
|
||||
! When we get here, we are sure that X is of
|
||||
! TYPE psb_i_base_vect.
|
||||
! If Y is not, throw the burden on it, implicitly
|
||||
! calling dot_a
|
||||
!
|
||||
select type(yy => y)
|
||||
type is (psb_i_base_vect_type)
|
||||
res = idot(n,x%v,1,y%v,1)
|
||||
class default
|
||||
res = y%dot(n,x%v)
|
||||
end select
|
||||
|
||||
end function i_base_dot_v
|
||||
|
||||
!
|
||||
! Base workhorse is good old BLAS1
|
||||
!
|
||||
function i_base_dot_a(n,x,y) result(res)
|
||||
implicit none
|
||||
class(psb_i_base_vect_type), intent(inout) :: x
|
||||
integer(psb_ipk_), intent(in) :: y(:)
|
||||
integer(psb_ipk_), intent(in) :: n
|
||||
integer(psb_ipk_) :: res
|
||||
integer(psb_ipk_), external :: idot
|
||||
|
||||
res = idot(n,y,1,x%v,1)
|
||||
|
||||
end function i_base_dot_a
|
||||
|
||||
!
|
||||
! AXPBY is invoked via Y, hence the structure below.
|
||||
!
|
||||
subroutine i_base_axpby_v(m,alpha, x, beta, y, info)
|
||||
use psi_serial_mod
|
||||
implicit none
|
||||
integer(psb_ipk_), intent(in) :: m
|
||||
class(psb_i_base_vect_type), intent(inout) :: x
|
||||
class(psb_i_base_vect_type), intent(inout) :: y
|
||||
integer(psb_ipk_), intent (in) :: alpha, beta
|
||||
integer(psb_ipk_), intent(out) :: info
|
||||
|
||||
select type(xx => x)
|
||||
type is (psb_i_base_vect_type)
|
||||
call psb_geaxpby(m,alpha,x%v,beta,y%v,info)
|
||||
class default
|
||||
call y%axpby(m,alpha,x%v,beta,info)
|
||||
end select
|
||||
|
||||
end subroutine i_base_axpby_v
|
||||
|
||||
subroutine i_base_axpby_a(m,alpha, x, beta, y, info)
|
||||
use psi_serial_mod
|
||||
implicit none
|
||||
integer(psb_ipk_), intent(in) :: m
|
||||
integer(psb_ipk_), intent(in) :: x(:)
|
||||
class(psb_i_base_vect_type), intent(inout) :: y
|
||||
integer(psb_ipk_), intent (in) :: alpha, beta
|
||||
integer(psb_ipk_), intent(out) :: info
|
||||
|
||||
call psb_geaxpby(m,alpha,x,beta,y%v,info)
|
||||
|
||||
end subroutine i_base_axpby_a
|
||||
|
||||
|
||||
!
|
||||
! Multiple variants of two operations:
|
||||
! Simple multiplication Y(:) = X(:)*Y(:)
|
||||
! blas-like: Z(:) = alpha*X(:)*Y(:)+beta*Z(:)
|
||||
!
|
||||
! Variants expanded according to the dynamic type
|
||||
! of the involved entities
|
||||
!
|
||||
|
||||
subroutine i_base_mlt_v(x, y, info)
|
||||
use psi_serial_mod
|
||||
implicit none
|
||||
class(psb_i_base_vect_type), intent(inout) :: x
|
||||
class(psb_i_base_vect_type), intent(inout) :: y
|
||||
integer(psb_ipk_), intent(out) :: info
|
||||
integer(psb_ipk_) :: i, n
|
||||
|
||||
info = 0
|
||||
select type(xx => x)
|
||||
type is (psb_i_base_vect_type)
|
||||
n = min(size(y%v), size(xx%v))
|
||||
do i=1, n
|
||||
y%v(i) = y%v(i)*xx%v(i)
|
||||
end do
|
||||
class default
|
||||
call y%mlt(x%v,info)
|
||||
end select
|
||||
|
||||
end subroutine i_base_mlt_v
|
||||
|
||||
subroutine i_base_mlt_a(x, y, info)
|
||||
use psi_serial_mod
|
||||
implicit none
|
||||
integer(psb_ipk_), intent(in) :: x(:)
|
||||
class(psb_i_base_vect_type), intent(inout) :: y
|
||||
integer(psb_ipk_), intent(out) :: info
|
||||
integer(psb_ipk_) :: i, n
|
||||
|
||||
info = 0
|
||||
n = min(size(y%v), size(x))
|
||||
do i=1, n
|
||||
y%v(i) = y%v(i)*x(i)
|
||||
end do
|
||||
|
||||
end subroutine i_base_mlt_a
|
||||
|
||||
|
||||
subroutine i_base_mlt_a_2(alpha,x,y,beta,z,info)
|
||||
use psi_serial_mod
|
||||
implicit none
|
||||
integer(psb_ipk_), intent(in) :: alpha,beta
|
||||
integer(psb_ipk_), intent(in) :: y(:)
|
||||
integer(psb_ipk_), intent(in) :: x(:)
|
||||
class(psb_i_base_vect_type), intent(inout) :: z
|
||||
integer(psb_ipk_), intent(out) :: info
|
||||
integer(psb_ipk_) :: i, n
|
||||
|
||||
info = 0
|
||||
n = min(size(z%v), size(x), size(y))
|
||||
!!$ write(0,*) 'Mlt_a_2: ',n
|
||||
if (alpha == izero) then
|
||||
if (beta == ione) then
|
||||
return
|
||||
else
|
||||
do i=1, n
|
||||
z%v(i) = beta*z%v(i)
|
||||
end do
|
||||
end if
|
||||
else
|
||||
if (alpha == ione) then
|
||||
if (beta == izero) then
|
||||
do i=1, n
|
||||
z%v(i) = y(i)*x(i)
|
||||
end do
|
||||
else if (beta == ione) then
|
||||
do i=1, n
|
||||
z%v(i) = z%v(i) + y(i)*x(i)
|
||||
end do
|
||||
else
|
||||
do i=1, n
|
||||
z%v(i) = beta*z%v(i) + y(i)*x(i)
|
||||
end do
|
||||
end if
|
||||
else if (alpha == -ione) then
|
||||
if (beta == izero) then
|
||||
do i=1, n
|
||||
z%v(i) = -y(i)*x(i)
|
||||
end do
|
||||
else if (beta == ione) then
|
||||
do i=1, n
|
||||
z%v(i) = z%v(i) - y(i)*x(i)
|
||||
end do
|
||||
else
|
||||
do i=1, n
|
||||
z%v(i) = beta*z%v(i) - y(i)*x(i)
|
||||
end do
|
||||
end if
|
||||
else
|
||||
if (beta == izero) then
|
||||
do i=1, n
|
||||
z%v(i) = alpha*y(i)*x(i)
|
||||
end do
|
||||
else if (beta == ione) then
|
||||
do i=1, n
|
||||
z%v(i) = z%v(i) + alpha*y(i)*x(i)
|
||||
end do
|
||||
else
|
||||
do i=1, n
|
||||
z%v(i) = beta*z%v(i) + alpha*y(i)*x(i)
|
||||
end do
|
||||
end if
|
||||
end if
|
||||
end if
|
||||
end subroutine i_base_mlt_a_2
|
||||
|
||||
subroutine i_base_mlt_v_2(alpha,x,y,beta,z,info,conjgx,conjgy)
|
||||
use psi_serial_mod
|
||||
use psb_string_mod
|
||||
implicit none
|
||||
integer(psb_ipk_), intent(in) :: alpha,beta
|
||||
class(psb_i_base_vect_type), intent(inout) :: x
|
||||
class(psb_i_base_vect_type), intent(inout) :: y
|
||||
class(psb_i_base_vect_type), intent(inout) :: z
|
||||
integer(psb_ipk_), intent(out) :: info
|
||||
character(len=1), intent(in), optional :: conjgx, conjgy
|
||||
integer(psb_ipk_) :: i, n
|
||||
logical :: conjgx_, conjgy_
|
||||
|
||||
info = 0
|
||||
if (.not.psb_i_is_complex_) then
|
||||
call z%mlt(alpha,x%v,y%v,beta,info)
|
||||
else
|
||||
conjgx_=.false.
|
||||
if (present(conjgx)) conjgx_ = (psb_toupper(conjgx)=='C')
|
||||
conjgy_=.false.
|
||||
if (present(conjgy)) conjgy_ = (psb_toupper(conjgy)=='C')
|
||||
if (conjgx_) x%v=(x%v)
|
||||
if (conjgy_) y%v=(y%v)
|
||||
call z%mlt(alpha,x%v,y%v,beta,info)
|
||||
if (conjgx_) x%v=(x%v)
|
||||
if (conjgy_) y%v=(y%v)
|
||||
end if
|
||||
end subroutine i_base_mlt_v_2
|
||||
|
||||
subroutine i_base_mlt_av(alpha,x,y,beta,z,info)
|
||||
use psi_serial_mod
|
||||
implicit none
|
||||
integer(psb_ipk_), intent(in) :: alpha,beta
|
||||
integer(psb_ipk_), intent(in) :: x(:)
|
||||
class(psb_i_base_vect_type), intent(inout) :: y
|
||||
class(psb_i_base_vect_type), intent(inout) :: z
|
||||
integer(psb_ipk_), intent(out) :: info
|
||||
integer(psb_ipk_) :: i, n
|
||||
|
||||
info = 0
|
||||
|
||||
call z%mlt(alpha,x,y%v,beta,info)
|
||||
|
||||
end subroutine i_base_mlt_av
|
||||
|
||||
subroutine i_base_mlt_va(alpha,x,y,beta,z,info)
|
||||
use psi_serial_mod
|
||||
implicit none
|
||||
integer(psb_ipk_), intent(in) :: alpha,beta
|
||||
integer(psb_ipk_), intent(in) :: y(:)
|
||||
class(psb_i_base_vect_type), intent(inout) :: x
|
||||
class(psb_i_base_vect_type), intent(inout) :: z
|
||||
integer(psb_ipk_), intent(out) :: info
|
||||
integer(psb_ipk_) :: i, n
|
||||
|
||||
info = 0
|
||||
|
||||
call z%mlt(alpha,y,x,beta,info)
|
||||
|
||||
end subroutine i_base_mlt_va
|
||||
|
||||
|
||||
!
|
||||
! Simple scaling
|
||||
!
|
||||
|
||||
subroutine i_base_scal(alpha, x)
|
||||
use psi_serial_mod
|
||||
implicit none
|
||||
class(psb_i_base_vect_type), intent(inout) :: x
|
||||
integer(psb_ipk_), intent (in) :: alpha
|
||||
|
||||
if (allocated(x%v)) x%v = alpha*x%v
|
||||
|
||||
end subroutine i_base_scal
|
||||
|
||||
!
|
||||
! Norms 1, 2 and infinity
|
||||
!
|
||||
|
||||
function i_base_nrm2(n,x) result(res)
|
||||
implicit none
|
||||
class(psb_i_base_vect_type), intent(inout) :: x
|
||||
integer(psb_ipk_), intent(in) :: n
|
||||
integer(psb_ipk_) :: res
|
||||
integer(psb_ipk_), external :: inrm2
|
||||
|
||||
res = inrm2(n,x%v,1)
|
||||
|
||||
end function i_base_nrm2
|
||||
|
||||
function i_base_amax(n,x) result(res)
|
||||
implicit none
|
||||
class(psb_i_base_vect_type), intent(inout) :: x
|
||||
integer(psb_ipk_), intent(in) :: n
|
||||
integer(psb_ipk_) :: res
|
||||
|
||||
res = maxval(abs(x%v(1:n)))
|
||||
|
||||
end function i_base_amax
|
||||
|
||||
function i_base_asum(n,x) result(res)
|
||||
implicit none
|
||||
class(psb_i_base_vect_type), intent(inout) :: x
|
||||
integer(psb_ipk_), intent(in) :: n
|
||||
integer(psb_ipk_) :: res
|
||||
|
||||
res = sum(abs(x%v(1:n)))
|
||||
|
||||
end function i_base_asum
|
||||
|
||||
|
||||
!
|
||||
! Gather: Y = beta * Y + alpha * X(IDX(:))
|
||||
!
|
||||
|
||||
subroutine i_base_gthab(n,idx,alpha,x,beta,y)
|
||||
use psi_serial_mod
|
||||
integer(psb_ipk_) :: n, idx(:)
|
||||
integer(psb_ipk_) :: alpha, beta, y(:)
|
||||
class(psb_i_base_vect_type) :: x
|
||||
|
||||
call x%sync()
|
||||
call psi_gth(n,idx,alpha,x%v,beta,y)
|
||||
|
||||
end subroutine i_base_gthab
|
||||
!
|
||||
! shortcut alpha=1 beta=0
|
||||
!
|
||||
subroutine i_base_gthzv(n,idx,x,y)
|
||||
use psi_serial_mod
|
||||
integer(psb_ipk_) :: n, idx(:)
|
||||
integer(psb_ipk_) :: y(:)
|
||||
class(psb_i_base_vect_type) :: x
|
||||
|
||||
call x%sync()
|
||||
call psi_gth(n,idx,x%v,y)
|
||||
|
||||
end subroutine i_base_gthzv
|
||||
|
||||
!
|
||||
! Scatter:
|
||||
! Y(IDX(:)) = beta*Y(IDX(:)) + X(:)
|
||||
!
|
||||
|
||||
subroutine i_base_sctb(n,idx,x,beta,y)
|
||||
use psi_serial_mod
|
||||
integer(psb_ipk_) :: n, idx(:)
|
||||
integer(psb_ipk_) :: beta, x(:)
|
||||
class(psb_i_base_vect_type) :: y
|
||||
|
||||
call y%sync()
|
||||
call psi_sct(n,idx,x,beta,y%v)
|
||||
call y%set_host()
|
||||
|
||||
end subroutine i_base_sctb
|
||||
|
||||
end module psb_i_base_vect_mod
|
||||
@ -0,0 +1,558 @@
|
||||
!!$
|
||||
!!$ Parallel Sparse BLAS version 3.0
|
||||
!!$ (C) Copyright 2006, 2007, 2008, 2009, 2010, 2012
|
||||
!!$ 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.
|
||||
!!$
|
||||
!!$
|
||||
!
|
||||
! package: psb_i_vect_mod
|
||||
!
|
||||
! This module contains the definition of the psb_i_vect type which
|
||||
! is the outer container for dense vectors.
|
||||
! Therefore all methods simply invoke the corresponding methods of the
|
||||
! inner component.
|
||||
!
|
||||
module psb_i_vect_mod
|
||||
|
||||
use psb_i_base_vect_mod
|
||||
|
||||
type psb_i_vect_type
|
||||
class(psb_i_base_vect_type), allocatable :: v
|
||||
contains
|
||||
procedure, pass(x) :: get_nrows => i_vect_get_nrows
|
||||
procedure, pass(x) :: sizeof => i_vect_sizeof
|
||||
procedure, pass(x) :: dot_v => i_vect_dot_v
|
||||
procedure, pass(x) :: dot_a => i_vect_dot_a
|
||||
generic, public :: dot => dot_v, dot_a
|
||||
procedure, pass(y) :: axpby_v => i_vect_axpby_v
|
||||
procedure, pass(y) :: axpby_a => i_vect_axpby_a
|
||||
generic, public :: axpby => axpby_v, axpby_a
|
||||
procedure, pass(y) :: mlt_v => i_vect_mlt_v
|
||||
procedure, pass(y) :: mlt_a => i_vect_mlt_a
|
||||
procedure, pass(z) :: mlt_a_2 => i_vect_mlt_a_2
|
||||
procedure, pass(z) :: mlt_v_2 => i_vect_mlt_v_2
|
||||
procedure, pass(z) :: mlt_va => i_vect_mlt_va
|
||||
procedure, pass(z) :: mlt_av => i_vect_mlt_av
|
||||
generic, public :: mlt => mlt_v, mlt_a, mlt_a_2,&
|
||||
& mlt_v_2, mlt_av, mlt_va
|
||||
procedure, pass(x) :: scal => i_vect_scal
|
||||
procedure, pass(x) :: nrm2 => i_vect_nrm2
|
||||
procedure, pass(x) :: amax => i_vect_amax
|
||||
procedure, pass(x) :: asum => i_vect_asum
|
||||
procedure, pass(x) :: all => i_vect_all
|
||||
procedure, pass(x) :: zero => i_vect_zero
|
||||
procedure, pass(x) :: asb => i_vect_asb
|
||||
procedure, pass(x) :: sync => i_vect_sync
|
||||
procedure, pass(x) :: gthab => i_vect_gthab
|
||||
procedure, pass(x) :: gthzv => i_vect_gthzv
|
||||
generic, public :: gth => gthab, gthzv
|
||||
procedure, pass(y) :: sctb => i_vect_sctb
|
||||
generic, public :: sct => sctb
|
||||
procedure, pass(x) :: free => i_vect_free
|
||||
procedure, pass(x) :: ins => i_vect_ins
|
||||
procedure, pass(x) :: bld_x => i_vect_bld_x
|
||||
procedure, pass(x) :: bld_n => i_vect_bld_n
|
||||
generic, public :: bld => bld_x, bld_n
|
||||
procedure, pass(x) :: get_vect => i_vect_get_vect
|
||||
procedure, pass(x) :: cnv => i_vect_cnv
|
||||
procedure, pass(x) :: set_scal => i_vect_set_scal
|
||||
procedure, pass(x) :: set_vect => i_vect_set_vect
|
||||
generic, public :: set => set_vect, set_scal
|
||||
end type psb_i_vect_type
|
||||
|
||||
public :: psb_i_vect
|
||||
private :: constructor, size_const
|
||||
interface psb_i_vect
|
||||
module procedure constructor, size_const
|
||||
end interface psb_i_vect
|
||||
|
||||
contains
|
||||
|
||||
subroutine i_vect_bld_x(x,invect,mold)
|
||||
integer(psb_ipk_), intent(in) :: invect(:)
|
||||
class(psb_i_vect_type), intent(out) :: x
|
||||
class(psb_i_base_vect_type), intent(in), optional :: mold
|
||||
integer(psb_ipk_) :: info
|
||||
|
||||
if (present(mold)) then
|
||||
#ifdef HAVE_MOLD
|
||||
allocate(x%v,stat=info,mold=mold)
|
||||
#else
|
||||
call mold%mold(x%v,info)
|
||||
#endif
|
||||
else
|
||||
allocate(psb_i_base_vect_type :: x%v,stat=info)
|
||||
endif
|
||||
|
||||
if (info == psb_success_) call x%v%bld(invect)
|
||||
|
||||
end subroutine i_vect_bld_x
|
||||
|
||||
|
||||
subroutine i_vect_bld_n(x,n,mold)
|
||||
integer(psb_ipk_), intent(in) :: n
|
||||
class(psb_i_vect_type), intent(out) :: x
|
||||
class(psb_i_base_vect_type), intent(in), optional :: mold
|
||||
integer(psb_ipk_) :: info
|
||||
|
||||
if (present(mold)) then
|
||||
#ifdef HAVE_MOLD
|
||||
allocate(x%v,stat=info,mold=mold)
|
||||
#else
|
||||
call mold%mold(x%v,info)
|
||||
#endif
|
||||
else
|
||||
allocate(psb_i_base_vect_type :: x%v,stat=info)
|
||||
endif
|
||||
if (info == psb_success_) call x%v%bld(n)
|
||||
|
||||
end subroutine i_vect_bld_n
|
||||
|
||||
function i_vect_get_vect(x) result(res)
|
||||
class(psb_i_vect_type), intent(inout) :: x
|
||||
integer(psb_ipk_), allocatable :: res(:)
|
||||
integer(psb_ipk_) :: info
|
||||
|
||||
if (allocated(x%v)) then
|
||||
res = x%v%get_vect()
|
||||
end if
|
||||
end function i_vect_get_vect
|
||||
|
||||
subroutine i_vect_set_scal(x,val)
|
||||
class(psb_i_vect_type), intent(inout) :: x
|
||||
integer(psb_ipk_), intent(in) :: val
|
||||
|
||||
integer(psb_ipk_) :: info
|
||||
if (allocated(x%v)) call x%v%set(val)
|
||||
|
||||
end subroutine i_vect_set_scal
|
||||
|
||||
subroutine i_vect_set_vect(x,val)
|
||||
class(psb_i_vect_type), intent(inout) :: x
|
||||
integer(psb_ipk_), intent(in) :: val(:)
|
||||
|
||||
integer(psb_ipk_) :: info
|
||||
if (allocated(x%v)) call x%v%set(val)
|
||||
|
||||
end subroutine i_vect_set_vect
|
||||
|
||||
|
||||
function constructor(x) result(this)
|
||||
integer(psb_ipk_) :: x(:)
|
||||
type(psb_i_vect_type) :: this
|
||||
integer(psb_ipk_) :: info
|
||||
|
||||
allocate(psb_i_base_vect_type :: this%v, stat=info)
|
||||
|
||||
if (info == 0) call this%v%bld(x)
|
||||
|
||||
call this%asb(size(x,kind=psb_ipk_),info)
|
||||
|
||||
end function constructor
|
||||
|
||||
|
||||
function size_const(n) result(this)
|
||||
integer(psb_ipk_), intent(in) :: n
|
||||
type(psb_i_vect_type) :: this
|
||||
integer(psb_ipk_) :: info
|
||||
|
||||
allocate(psb_i_base_vect_type :: this%v, stat=info)
|
||||
call this%asb(n,info)
|
||||
|
||||
end function size_const
|
||||
|
||||
function i_vect_get_nrows(x) result(res)
|
||||
implicit none
|
||||
class(psb_i_vect_type), intent(in) :: x
|
||||
integer(psb_ipk_) :: res
|
||||
res = 0
|
||||
if (allocated(x%v)) res = x%v%get_nrows()
|
||||
end function i_vect_get_nrows
|
||||
|
||||
function i_vect_sizeof(x) result(res)
|
||||
implicit none
|
||||
class(psb_i_vect_type), intent(in) :: x
|
||||
integer(psb_long_int_k_) :: res
|
||||
res = 0
|
||||
if (allocated(x%v)) res = x%v%sizeof()
|
||||
end function i_vect_sizeof
|
||||
|
||||
function i_vect_dot_v(n,x,y) result(res)
|
||||
implicit none
|
||||
class(psb_i_vect_type), intent(inout) :: x, y
|
||||
integer(psb_ipk_), intent(in) :: n
|
||||
integer(psb_ipk_) :: res
|
||||
|
||||
res = izero
|
||||
if (allocated(x%v).and.allocated(y%v)) &
|
||||
& res = x%v%dot(n,y%v)
|
||||
|
||||
end function i_vect_dot_v
|
||||
|
||||
function i_vect_dot_a(n,x,y) result(res)
|
||||
implicit none
|
||||
class(psb_i_vect_type), intent(inout) :: x
|
||||
integer(psb_ipk_), intent(in) :: y(:)
|
||||
integer(psb_ipk_), intent(in) :: n
|
||||
integer(psb_ipk_) :: res
|
||||
|
||||
res = izero
|
||||
if (allocated(x%v)) &
|
||||
& res = x%v%dot(n,y)
|
||||
|
||||
end function i_vect_dot_a
|
||||
|
||||
subroutine i_vect_axpby_v(m,alpha, x, beta, y, info)
|
||||
use psi_serial_mod
|
||||
implicit none
|
||||
integer(psb_ipk_), intent(in) :: m
|
||||
class(psb_i_vect_type), intent(inout) :: x
|
||||
class(psb_i_vect_type), intent(inout) :: y
|
||||
integer(psb_ipk_), intent (in) :: alpha, beta
|
||||
integer(psb_ipk_), intent(out) :: info
|
||||
|
||||
if (allocated(x%v).and.allocated(y%v)) then
|
||||
call y%v%axpby(m,alpha,x%v,beta,info)
|
||||
else
|
||||
info = psb_err_invalid_vect_state_
|
||||
end if
|
||||
|
||||
end subroutine i_vect_axpby_v
|
||||
|
||||
subroutine i_vect_axpby_a(m,alpha, x, beta, y, info)
|
||||
use psi_serial_mod
|
||||
implicit none
|
||||
integer(psb_ipk_), intent(in) :: m
|
||||
integer(psb_ipk_), intent(in) :: x(:)
|
||||
class(psb_i_vect_type), intent(inout) :: y
|
||||
integer(psb_ipk_), intent (in) :: alpha, beta
|
||||
integer(psb_ipk_), intent(out) :: info
|
||||
|
||||
if (allocated(y%v)) &
|
||||
& call y%v%axpby(m,alpha,x,beta,info)
|
||||
|
||||
end subroutine i_vect_axpby_a
|
||||
|
||||
|
||||
subroutine i_vect_mlt_v(x, y, info)
|
||||
use psi_serial_mod
|
||||
implicit none
|
||||
class(psb_i_vect_type), intent(inout) :: x
|
||||
class(psb_i_vect_type), intent(inout) :: y
|
||||
integer(psb_ipk_), intent(out) :: info
|
||||
integer(psb_ipk_) :: i, n
|
||||
|
||||
info = 0
|
||||
if (allocated(x%v).and.allocated(y%v)) &
|
||||
& call y%v%mlt(x%v,info)
|
||||
|
||||
end subroutine i_vect_mlt_v
|
||||
|
||||
subroutine i_vect_mlt_a(x, y, info)
|
||||
use psi_serial_mod
|
||||
implicit none
|
||||
integer(psb_ipk_), intent(in) :: x(:)
|
||||
class(psb_i_vect_type), intent(inout) :: y
|
||||
integer(psb_ipk_), intent(out) :: info
|
||||
integer(psb_ipk_) :: i, n
|
||||
|
||||
|
||||
info = 0
|
||||
if (allocated(y%v)) &
|
||||
& call y%v%mlt(x,info)
|
||||
|
||||
end subroutine i_vect_mlt_a
|
||||
|
||||
|
||||
subroutine i_vect_mlt_a_2(alpha,x,y,beta,z,info)
|
||||
use psi_serial_mod
|
||||
implicit none
|
||||
integer(psb_ipk_), intent(in) :: alpha,beta
|
||||
integer(psb_ipk_), intent(in) :: y(:)
|
||||
integer(psb_ipk_), intent(in) :: x(:)
|
||||
class(psb_i_vect_type), intent(inout) :: z
|
||||
integer(psb_ipk_), intent(out) :: info
|
||||
integer(psb_ipk_) :: i, n
|
||||
|
||||
info = 0
|
||||
if (allocated(z%v)) &
|
||||
& call z%v%mlt(alpha,x,y,beta,info)
|
||||
|
||||
end subroutine i_vect_mlt_a_2
|
||||
|
||||
subroutine i_vect_mlt_v_2(alpha,x,y,beta,z,info,conjgx,conjgy)
|
||||
use psi_serial_mod
|
||||
implicit none
|
||||
integer(psb_ipk_), intent(in) :: alpha,beta
|
||||
class(psb_i_vect_type), intent(inout) :: x
|
||||
class(psb_i_vect_type), intent(inout) :: y
|
||||
class(psb_i_vect_type), intent(inout) :: z
|
||||
integer(psb_ipk_), intent(out) :: info
|
||||
character(len=1), intent(in), optional :: conjgx, conjgy
|
||||
|
||||
integer(psb_ipk_) :: i, n
|
||||
|
||||
info = 0
|
||||
if (allocated(x%v).and.allocated(y%v).and.&
|
||||
& allocated(z%v)) &
|
||||
& call z%v%mlt(alpha,x%v,y%v,beta,info,conjgx,conjgy)
|
||||
|
||||
end subroutine i_vect_mlt_v_2
|
||||
|
||||
subroutine i_vect_mlt_av(alpha,x,y,beta,z,info)
|
||||
use psi_serial_mod
|
||||
implicit none
|
||||
integer(psb_ipk_), intent(in) :: alpha,beta
|
||||
integer(psb_ipk_), intent(in) :: x(:)
|
||||
class(psb_i_vect_type), intent(inout) :: y
|
||||
class(psb_i_vect_type), intent(inout) :: z
|
||||
integer(psb_ipk_), intent(out) :: info
|
||||
integer(psb_ipk_) :: i, n
|
||||
|
||||
info = 0
|
||||
if (allocated(z%v).and.allocated(y%v)) &
|
||||
& call z%v%mlt(alpha,x,y%v,beta,info)
|
||||
|
||||
end subroutine i_vect_mlt_av
|
||||
|
||||
subroutine i_vect_mlt_va(alpha,x,y,beta,z,info)
|
||||
use psi_serial_mod
|
||||
implicit none
|
||||
integer(psb_ipk_), intent(in) :: alpha,beta
|
||||
integer(psb_ipk_), intent(in) :: y(:)
|
||||
class(psb_i_vect_type), intent(inout) :: x
|
||||
class(psb_i_vect_type), intent(inout) :: z
|
||||
integer(psb_ipk_), intent(out) :: info
|
||||
integer(psb_ipk_) :: i, n
|
||||
|
||||
info = 0
|
||||
|
||||
if (allocated(z%v).and.allocated(x%v)) &
|
||||
& call z%v%mlt(alpha,x%v,y,beta,info)
|
||||
|
||||
end subroutine i_vect_mlt_va
|
||||
|
||||
subroutine i_vect_scal(alpha, x)
|
||||
use psi_serial_mod
|
||||
implicit none
|
||||
class(psb_i_vect_type), intent(inout) :: x
|
||||
integer(psb_ipk_), intent (in) :: alpha
|
||||
|
||||
if (allocated(x%v)) call x%v%scal(alpha)
|
||||
|
||||
end subroutine i_vect_scal
|
||||
|
||||
|
||||
function i_vect_nrm2(n,x) result(res)
|
||||
implicit none
|
||||
class(psb_i_vect_type), intent(inout) :: x
|
||||
integer(psb_ipk_), intent(in) :: n
|
||||
integer(psb_ipk_) :: res
|
||||
|
||||
if (allocated(x%v)) then
|
||||
res = x%v%nrm2(n)
|
||||
else
|
||||
res = izero
|
||||
end if
|
||||
|
||||
end function i_vect_nrm2
|
||||
|
||||
function i_vect_amax(n,x) result(res)
|
||||
implicit none
|
||||
class(psb_i_vect_type), intent(inout) :: x
|
||||
integer(psb_ipk_), intent(in) :: n
|
||||
integer(psb_ipk_) :: res
|
||||
|
||||
if (allocated(x%v)) then
|
||||
res = x%v%amax(n)
|
||||
else
|
||||
res = izero
|
||||
end if
|
||||
|
||||
end function i_vect_amax
|
||||
|
||||
function i_vect_asum(n,x) result(res)
|
||||
implicit none
|
||||
class(psb_i_vect_type), intent(inout) :: x
|
||||
integer(psb_ipk_), intent(in) :: n
|
||||
integer(psb_ipk_) :: res
|
||||
|
||||
if (allocated(x%v)) then
|
||||
res = x%v%asum(n)
|
||||
else
|
||||
res = izero
|
||||
end if
|
||||
|
||||
end function i_vect_asum
|
||||
|
||||
subroutine i_vect_all(n, x, info, mold)
|
||||
|
||||
implicit none
|
||||
integer(psb_ipk_), intent(in) :: n
|
||||
class(psb_i_vect_type), intent(out) :: x
|
||||
class(psb_i_base_vect_type), intent(in), optional :: mold
|
||||
integer(psb_ipk_), intent(out) :: info
|
||||
|
||||
if (present(mold)) then
|
||||
#ifdef HAVE_MOLD
|
||||
allocate(x%v,stat=info,mold=mold)
|
||||
#else
|
||||
call mold%mold(x%v,info)
|
||||
#endif
|
||||
else
|
||||
allocate(psb_i_base_vect_type :: x%v,stat=info)
|
||||
endif
|
||||
if (info == 0) then
|
||||
call x%v%all(n,info)
|
||||
else
|
||||
info = psb_err_alloc_dealloc_
|
||||
end if
|
||||
|
||||
end subroutine i_vect_all
|
||||
|
||||
subroutine i_vect_zero(x)
|
||||
use psi_serial_mod
|
||||
implicit none
|
||||
class(psb_i_vect_type), intent(inout) :: x
|
||||
|
||||
if (allocated(x%v)) call x%v%zero()
|
||||
|
||||
end subroutine i_vect_zero
|
||||
|
||||
subroutine i_vect_asb(n, x, info)
|
||||
use psi_serial_mod
|
||||
use psb_realloc_mod
|
||||
implicit none
|
||||
integer(psb_ipk_), intent(in) :: n
|
||||
class(psb_i_vect_type), intent(inout) :: x
|
||||
integer(psb_ipk_), intent(out) :: info
|
||||
|
||||
if (allocated(x%v)) &
|
||||
& call x%v%asb(n,info)
|
||||
|
||||
end subroutine i_vect_asb
|
||||
|
||||
subroutine i_vect_sync(x)
|
||||
implicit none
|
||||
class(psb_i_vect_type), intent(inout) :: x
|
||||
|
||||
if (allocated(x%v)) &
|
||||
& call x%v%sync()
|
||||
|
||||
end subroutine i_vect_sync
|
||||
|
||||
subroutine i_vect_gthab(n,idx,alpha,x,beta,y)
|
||||
use psi_serial_mod
|
||||
integer(psb_ipk_) :: n, idx(:)
|
||||
integer(psb_ipk_) :: alpha, beta, y(:)
|
||||
class(psb_i_vect_type) :: x
|
||||
|
||||
if (allocated(x%v)) &
|
||||
& call x%v%gth(n,idx,alpha,beta,y)
|
||||
|
||||
end subroutine i_vect_gthab
|
||||
|
||||
subroutine i_vect_gthzv(n,idx,x,y)
|
||||
use psi_serial_mod
|
||||
integer(psb_ipk_) :: n, idx(:)
|
||||
integer(psb_ipk_) :: y(:)
|
||||
class(psb_i_vect_type) :: x
|
||||
|
||||
if (allocated(x%v)) &
|
||||
& call x%v%gth(n,idx,y)
|
||||
|
||||
end subroutine i_vect_gthzv
|
||||
|
||||
subroutine i_vect_sctb(n,idx,x,beta,y)
|
||||
use psi_serial_mod
|
||||
integer(psb_ipk_) :: n, idx(:)
|
||||
integer(psb_ipk_) :: beta, x(:)
|
||||
class(psb_i_vect_type) :: y
|
||||
|
||||
if (allocated(y%v)) &
|
||||
& call y%v%sct(n,idx,x,beta)
|
||||
|
||||
end subroutine i_vect_sctb
|
||||
|
||||
subroutine i_vect_free(x, info)
|
||||
use psi_serial_mod
|
||||
use psb_realloc_mod
|
||||
implicit none
|
||||
class(psb_i_vect_type), intent(inout) :: x
|
||||
integer(psb_ipk_), intent(out) :: info
|
||||
|
||||
info = 0
|
||||
if (allocated(x%v)) then
|
||||
call x%v%free(info)
|
||||
if (info == 0) deallocate(x%v,stat=info)
|
||||
end if
|
||||
|
||||
end subroutine i_vect_free
|
||||
|
||||
subroutine i_vect_ins(n,irl,val,dupl,x,info)
|
||||
use psi_serial_mod
|
||||
implicit none
|
||||
class(psb_i_vect_type), intent(inout) :: x
|
||||
integer(psb_ipk_), intent(in) :: n, dupl
|
||||
integer(psb_ipk_), intent(in) :: irl(:)
|
||||
integer(psb_ipk_), intent(in) :: val(:)
|
||||
integer(psb_ipk_), intent(out) :: info
|
||||
|
||||
integer(psb_ipk_) :: i
|
||||
|
||||
info = 0
|
||||
if (.not.allocated(x%v)) then
|
||||
info = psb_err_invalid_vect_state_
|
||||
return
|
||||
end if
|
||||
|
||||
call x%v%ins(n,irl,val,dupl,info)
|
||||
|
||||
end subroutine i_vect_ins
|
||||
|
||||
|
||||
subroutine i_vect_cnv(x,mold)
|
||||
class(psb_i_vect_type), intent(inout) :: x
|
||||
class(psb_i_base_vect_type), intent(in) :: mold
|
||||
class(psb_i_base_vect_type), allocatable :: tmp
|
||||
integer(psb_ipk_), allocatable :: invect(:)
|
||||
integer(psb_ipk_) :: info
|
||||
|
||||
#ifdef HAVE_MOLD
|
||||
allocate(tmp,stat=info,mold=mold)
|
||||
#else
|
||||
call mold%mold(tmp,info)
|
||||
#endif
|
||||
call x%v%sync()
|
||||
if (info == psb_success_) call tmp%bld(x%v%v)
|
||||
call x%v%free(info)
|
||||
call move_alloc(tmp,x%v)
|
||||
|
||||
end subroutine i_vect_cnv
|
||||
|
||||
end module psb_i_vect_mod
|
||||
Some files were not shown because too many files have changed in this diff Show More
Loading…
Reference in New Issue