!!$ !!$ 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