You cannot select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
psblas3/base/modules/auxil/psb_m_realloc_mod.F90

1128 lines
29 KiB
Fortran

!
! Parallel Sparse BLAS version 3.5
! (C) Copyright 2006-2018
! Salvatore Filippone
! Alfredo Buttari
!
! Redistribution and use in source and binary forms, with or without
! modification, are permitted provided that the following conditions
! are met:
! 1. Redistributions of source code must retain the above copyright
! notice, this list of conditions and the following disclaimer.
! 2. Redistributions in binary form must reproduce the above copyright
! notice, this list of conditions, and the following disclaimer in the
! documentation and/or other materials provided with the distribution.
! 3. The name of the PSBLAS group or the names of its contributors may
! not be used to endorse or promote products derived from this
! software without specific written permission.
!
! THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
! ``AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED
! TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
! PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE PSBLAS GROUP OR ITS CONTRIBUTORS
! BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
! CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
! SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
! INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
! CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
! ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
! POSSIBILITY OF SUCH DAMAGE.
!
!
module psb_m_realloc_mod
use psb_const_mod
implicit none
!
! psb_realloc will reallocate the input array to have exactly
! the size specified, possibly shortening it.
!
Interface psb_realloc
module procedure psb_r_m_s
module procedure psb_r_m_m_rk1
module procedure psb_r_m_m_rk2
module procedure psb_r_e_m_rk1
module procedure psb_r_e_m_rk2
module procedure psb_r_me_m_rk2
module procedure psb_r_em_m_rk2
module procedure psb_r_m_2_m_rk1
module procedure psb_r_e_2_m_rk1
end Interface psb_realloc
interface psb_move_alloc
module procedure psb_move_alloc_m_rk1, psb_move_alloc_m_rk2
end interface psb_move_alloc
Interface psb_safe_ab_cpy
module procedure psb_ab_cpy_m_s, psb_ab_cpy_m_rk1, psb_ab_cpy_m_rk2
end Interface psb_safe_ab_cpy
Interface psb_safe_cpy
module procedure psb_cpy_m_rk1, psb_cpy_m_rk2
end Interface psb_safe_cpy
!
! psb_ensure_size will reallocate the input array if necessary
! to guarantee that its size is at least as large as the
! value required, usually with some room to spare.
!
interface psb_ensure_size
module procedure psb_ensure_m_sz_m_rk1, psb_ensure_e_sz_m_rk1
end Interface psb_ensure_size
!
! psb_size returns 0 if argument is not allocated.
!
interface psb_size
module procedure psb_size_m_rk1, psb_size_m_rk2
end interface psb_size
Contains
Subroutine psb_r_m_s(rrax,info)
use psb_error_mod
! ...Subroutine Arguments
integer(psb_mpk_), allocatable, intent(inout) :: rrax
integer(psb_ipk_) :: info
! ...Local Variables
integer(psb_ipk_) :: err_act,err
character(len=30) :: name
logical, parameter :: debug=.false.
name='psb_r_m_s'
call psb_erractionsave(err_act)
info=psb_success_
if (.not.allocated(rrax)) then
Allocate(rrax,stat=info)
if (info /= psb_success_) then
err=4025
call psb_errpush(err,name, l_err=(/1_psb_lpk_/), &
& a_err='integer(psb_mpk_)')
goto 9999
end if
endif
call psb_erractionrestore(err_act)
return
9999 continue
info = err
call psb_error_handler(err_act)
return
End Subroutine psb_r_m_s
Subroutine psb_r_m_m_rk1(len,rrax,info,pad,lb)
use psb_error_mod
! ...Subroutine Arguments
integer(psb_mpk_),Intent(in) :: len
integer(psb_mpk_), allocatable, intent(inout) :: rrax(:)
integer(psb_ipk_) :: info
integer(psb_mpk_), optional, intent(in) :: pad
integer(psb_mpk_), optional, intent(in) :: lb
! ...Local Variables
integer(psb_mpk_),allocatable :: tmp(:)
integer(psb_mpk_) :: dim, lb_, lbi,ub_, i
integer(psb_ipk_) :: err_act,err
character(len=30) :: name
logical, parameter :: debug=.false.
name='psb_r_m_m_rk1'
call psb_erractionsave(err_act)
info=psb_success_
if (debug) write(psb_err_unit,*) 'reallocate D',len
if (present(lb)) then
lb_ = lb
else
lb_ = 1
endif
if ((len<0)) then
err=4025
call psb_errpush(err,name, l_err=(/len*1_psb_lpk_/), &
& a_err='integer(psb_mpk_)')
goto 9999
end if
ub_ = lb_ + len-1
if (allocated(rrax)) then
dim = size(rrax)
lbi = lbound(rrax,1)
If ((dim /= len).or.(lbi /= lb_)) Then
Allocate(tmp(lb_:ub_),stat=info)
if (info /= psb_success_) then
err=4025
call psb_errpush(err,name, l_err=(/len*1_psb_lpk_/), &
& a_err='integer(psb_mpk_)')
goto 9999
end if
tmp(lb_:lb_-1+min(len,dim))=rrax(lbi:lbi-1+min(len,dim))
call psb_move_alloc(tmp,rrax,info)
End If
else
dim = 0
Allocate(rrax(lb_:ub_),stat=info)
if (info /= psb_success_) then
err=4025
call psb_errpush(err,name, l_err=(/len*1_psb_lpk_/), &
& a_err='integer(psb_mpk_)')
goto 9999
end if
endif
if (present(pad)) then
!$omp parallel do private(i) shared(dim,len)
do i=lb_-1+dim+1,lb_-1+len
rrax(i) = pad
end do
endif
call psb_erractionrestore(err_act)
return
9999 continue
info = err
call psb_error_handler(err_act)
return
End Subroutine psb_r_m_m_rk1
Subroutine psb_r_m_m_rk2(len1,len2,rrax,info,pad,lb1,lb2)
use psb_error_mod
! ...Subroutine Arguments
integer(psb_mpk_),Intent(in) :: len1,len2
integer(psb_mpk_),allocatable :: rrax(:,:)
integer(psb_ipk_) :: info
integer(psb_mpk_), optional, intent(in) :: pad
integer(psb_mpk_),Intent(in), optional :: lb1,lb2
! ...Local Variables
integer(psb_mpk_),allocatable :: tmp(:,:)
integer(psb_ipk_) :: err_act,err
integer(psb_mpk_) :: dim,dim2,lb1_, lb2_, ub1_, ub2_,lbi1, lbi2, i
character(len=30) :: name
name='psb_r_m_m_rk2'
call psb_erractionsave(err_act)
info=psb_success_
if (present(lb1)) then
lb1_ = lb1
else
lb1_ = 1
endif
if (present(lb2)) then
lb2_ = lb2
else
lb2_ = 1
endif
ub1_ = lb1_ + len1 -1
ub2_ = lb2_ + len2 -1
if (len1 < 0) then
err=4025
call psb_errpush(err,name, l_err=(/len1*1_psb_lpk_/), &
& a_err='integer(psb_mpk_)')
goto 9999
end if
if (len2 < 0) then
err=4025
call psb_errpush(err,name, l_err=(/len2*1_psb_lpk_/), &
& a_err='integer(psb_mpk_)')
goto 9999
end if
if (allocated(rrax)) then
dim = size(rrax,1)
lbi1 = lbound(rrax,1)
dim2 = size(rrax,2)
lbi2 = lbound(rrax,2)
If ((dim /= len1).or.(dim2 /= len2).or.(lbi1 /= lb1_)&
& .or.(lbi2 /= lb2_)) Then
Allocate(tmp(lb1_:ub1_,lb2_:ub2_),stat=info)
if (info /= psb_success_) then
err=4025
call psb_errpush(err,name, l_err=(/len1*1_psb_lpk_*len2/), &
& a_err='integer(psb_mpk_)')
goto 9999
end if
tmp(lb1_:lb1_-1+min(len1,dim),lb2_:lb2_-1+min(len2,dim2)) = &
& rrax(lbi1:lbi1-1+min(len1,dim),lbi2:lbi2-1+min(len2,dim2))
call psb_move_alloc(tmp,rrax,info)
End If
else
dim = 0
dim2 = 0
Allocate(rrax(lb1_:ub1_,lb2_:ub2_),stat=info)
if (info /= psb_success_) then
err=4025
call psb_errpush(err,name, l_err=(/len1*1_psb_lpk_*len2/), &
& a_err='integer(psb_mpk_)')
goto 9999
end if
endif
if (present(pad)) then
!$omp parallel do private(i) shared(lb1_,dim,len1)
do i=lb1_-1+dim+1,lb1_-1+len1
rrax(i,:) = pad
end do
!$omp parallel do private(i) shared(lb1_,dim,len1,lb2_,dim2,len2)
do i=lb1_,lb1_-1+len1
rrax(i,lb2_-1+dim2+1:lb2_-1+len2) = pad
end do
endif
call psb_erractionrestore(err_act)
return
9999 continue
info = err
call psb_error_handler(err_act)
return
End Subroutine psb_r_m_m_rk2
Subroutine psb_r_e_m_rk1(len,rrax,info,pad,lb)
use psb_error_mod
! ...Subroutine Arguments
integer(psb_epk_),Intent(in) :: len
integer(psb_mpk_), allocatable, intent(inout) :: rrax(:)
integer(psb_ipk_) :: info
integer(psb_mpk_), optional, intent(in) :: pad
integer(psb_epk_), optional, intent(in) :: lb
! ...Local Variables
integer(psb_mpk_),allocatable :: tmp(:)
integer(psb_epk_) :: dim, lb_, lbi,ub_
integer(psb_ipk_) :: err_act,err
character(len=30) :: name
logical, parameter :: debug=.false.
name='psb_r_m_m_rk1'
call psb_erractionsave(err_act)
info=psb_success_
if (debug) write(psb_err_unit,*) 'reallocate D',len
if (present(lb)) then
lb_ = lb
else
lb_ = 1
endif
if ((len<0)) then
err=4025
call psb_errpush(err,name, e_err=(/len/), &
& a_err='integer(psb_mpk_)')
goto 9999
end if
ub_ = lb_ + len-1
if (allocated(rrax)) then
dim = size(rrax)
lbi = lbound(rrax,1)
If ((dim /= len).or.(lbi /= lb_)) Then
Allocate(tmp(lb_:ub_),stat=info)
if (info /= psb_success_) then
err=4025
call psb_errpush(err,name, e_err=(/len/), &
& a_err='integer(psb_mpk_)')
goto 9999
end if
tmp(lb_:lb_-1+min(len,dim))=rrax(lbi:lbi-1+min(len,dim))
call psb_move_alloc(tmp,rrax,info)
End If
else
dim = 0
Allocate(rrax(lb_:ub_),stat=info)
if (info /= psb_success_) then
err=4025
call psb_errpush(err,name, e_err=(/len/), &
& a_err='integer(psb_mpk_)')
goto 9999
end if
endif
if (present(pad)) then
rrax(lb_-1+dim+1:lb_-1+len) = pad
endif
call psb_erractionrestore(err_act)
return
9999 continue
info = err
call psb_error_handler(err_act)
return
End Subroutine psb_r_e_m_rk1
Subroutine psb_r_e_m_rk2(len1,len2,rrax,info,pad,lb1,lb2)
use psb_error_mod
! ...Subroutine Arguments
integer(psb_epk_),Intent(in) :: len1,len2
integer(psb_mpk_),allocatable :: rrax(:,:)
integer(psb_ipk_) :: info
integer(psb_mpk_), optional, intent(in) :: pad
integer(psb_epk_),Intent(in), optional :: lb1,lb2
! ...Local Variables
integer(psb_mpk_),allocatable :: tmp(:,:)
integer(psb_ipk_) :: err_act,err
integer(psb_epk_) :: dim,dim2,lb1_, lb2_, ub1_, ub2_,lbi1, lbi2
character(len=30) :: name
name='psb_r_e_m_rk2'
call psb_erractionsave(err_act)
info=psb_success_
if (present(lb1)) then
lb1_ = lb1
else
lb1_ = 1
endif
if (present(lb2)) then
lb2_ = lb2
else
lb2_ = 1
endif
ub1_ = lb1_ + len1 -1
ub2_ = lb2_ + len2 -1
if (len1 < 0) then
err=4025
call psb_errpush(err,name, e_err=(/len1/), &
& a_err='integer(psb_mpk_)')
goto 9999
end if
if (len2 < 0) then
err=4025
call psb_errpush(err,name, e_err=(/len2/), &
& a_err='integer(psb_mpk_)')
goto 9999
end if
if (allocated(rrax)) then
dim = size(rrax,1)
lbi1 = lbound(rrax,1)
dim2 = size(rrax,2)
lbi2 = lbound(rrax,2)
If ((dim /= len1).or.(dim2 /= len2).or.(lbi1 /= lb1_)&
& .or.(lbi2 /= lb2_)) Then
Allocate(tmp(lb1_:ub1_,lb2_:ub2_),stat=info)
if (info /= psb_success_) then
err=4025
call psb_errpush(err,name, e_err=(/(len1*len2)/), &
& a_err='integer(psb_mpk_)')
goto 9999
end if
tmp(lb1_:lb1_-1+min(len1,dim),lb2_:lb2_-1+min(len2,dim2)) = &
& rrax(lbi1:lbi1-1+min(len1,dim),lbi2:lbi2-1+min(len2,dim2))
call psb_move_alloc(tmp,rrax,info)
End If
else
dim = 0
dim2 = 0
Allocate(rrax(lb1_:ub1_,lb2_:ub2_),stat=info)
if (info /= psb_success_) then
err=4025
call psb_errpush(err,name, e_err=(/(len1*len2)/), &
& a_err='integer(psb_mpk_)')
goto 9999
end if
endif
if (present(pad)) then
rrax(lb1_-1+dim+1:lb1_-1+len1,:) = pad
rrax(lb1_:lb1_-1+dim,lb2_-1+dim2+1:lb2_-1+len2) = pad
endif
call psb_erractionrestore(err_act)
return
9999 continue
info = err
call psb_error_handler(err_act)
return
End Subroutine psb_r_e_m_rk2
Subroutine psb_r_me_m_rk2(len1,len2,rrax,info,pad,lb1,lb2)
use psb_error_mod
! ...Subroutine Arguments
integer(psb_mpk_),Intent(in) :: len1
integer(psb_epk_),Intent(in) :: len2
integer(psb_mpk_),allocatable :: rrax(:,:)
integer(psb_ipk_) :: info
integer(psb_mpk_), optional, intent(in) :: pad
integer(psb_mpk_),Intent(in), optional :: lb1,lb2
! ...Local Variables
integer(psb_mpk_),allocatable :: tmp(:,:)
integer(psb_ipk_) :: err_act,err
integer(psb_epk_) :: dim,lb1_, lb2_, ub1_, ub2_,lbi1, lbi2
integer(psb_epk_) :: dim2
character(len=30) :: name
name='psb_r_me_m_rk2'
call psb_erractionsave(err_act)
info=psb_success_
if (present(lb1)) then
lb1_ = lb1
else
lb1_ = 1
endif
if (present(lb2)) then
lb2_ = lb2
else
lb2_ = 1
endif
ub1_ = lb1_ + len1 -1
ub2_ = lb2_ + len2 -1
if (len1 < 0) then
err=4025
call psb_errpush(err,name, m_err=(/len1/), &
& a_err='integer(psb_mpk_)')
goto 9999
end if
if (len2 < 0) then
err=4025
call psb_errpush(err,name, e_err=(/len2/), &
& a_err='integer(psb_mpk_)')
goto 9999
end if
if (allocated(rrax)) then
dim = size(rrax,1)
lbi1 = lbound(rrax,1)
dim2 = size(rrax,2)
lbi2 = lbound(rrax,2)
If ((dim /= len1).or.(dim2 /= len2).or.(lbi1 /= lb1_)&
& .or.(lbi2 /= lb2_)) Then
Allocate(tmp(lb1_:ub1_,lb2_:ub2_),stat=info)
if (info /= psb_success_) then
err=4025
call psb_errpush(err,name, e_err=(/len1*len2/), &
& a_err='integer(psb_mpk_)')
goto 9999
end if
tmp(lb1_:lb1_-1+min(len1,dim),lb2_:lb2_-1+min(len2,dim2)) = &
& rrax(lbi1:lbi1-1+min(len1,dim),lbi2:lbi2-1+min(len2,dim2))
call psb_move_alloc(tmp,rrax,info)
End If
else
dim = 0
dim2 = 0
Allocate(rrax(lb1_:ub1_,lb2_:ub2_),stat=info)
if (info /= psb_success_) then
err=4025
call psb_errpush(err,name,e_err=(/len1*len2/),&
& a_err='integer(psb_mpk_)')
goto 9999
end if
endif
if (present(pad)) then
rrax(lb1_-1+dim+1:lb1_-1+len1,:) = pad
rrax(lb1_:lb1_-1+dim,lb2_-1+dim2+1:lb2_-1+len2) = pad
endif
call psb_erractionrestore(err_act)
return
9999 continue
info = err
call psb_error_handler(err_act)
return
End Subroutine psb_r_me_m_rk2
Subroutine psb_r_em_m_rk2(len1,len2,rrax,info,pad,lb1,lb2)
use psb_error_mod
! ...Subroutine Arguments
integer(psb_epk_),Intent(in) :: len1
integer(psb_mpk_),Intent(in) :: len2
integer(psb_mpk_),allocatable :: rrax(:,:)
integer(psb_ipk_) :: info
integer(psb_mpk_), optional, intent(in) :: pad
integer(psb_mpk_),Intent(in), optional :: lb1,lb2
! ...Local Variables
integer(psb_mpk_),allocatable :: tmp(:,:)
integer(psb_ipk_) :: err_act,err
integer(psb_epk_) :: dim2,lb1_, lb2_, ub1_, ub2_,lbi1, lbi2
integer(psb_epk_) :: dim
character(len=30) :: name
name='psb_r_me_m_rk2'
call psb_erractionsave(err_act)
info=psb_success_
if (present(lb1)) then
lb1_ = lb1
else
lb1_ = 1
endif
if (present(lb2)) then
lb2_ = lb2
else
lb2_ = 1
endif
ub1_ = lb1_ + len1 -1
ub2_ = lb2_ + len2 -1
if (len1 < 0) then
err=4025
call psb_errpush(err,name, e_err=(/len1/), &
& a_err='integer(psb_mpk_)')
goto 9999
end if
if (len2 < 0) then
err=4025
call psb_errpush(err,name, m_err=(/len2/), &
& a_err='integer(psb_mpk_)')
goto 9999
end if
if (allocated(rrax)) then
dim = size(rrax,1)
lbi1 = lbound(rrax,1)
dim2 = size(rrax,2)
lbi2 = lbound(rrax,2)
If ((dim /= len1).or.(dim2 /= len2).or.(lbi1 /= lb1_)&
& .or.(lbi2 /= lb2_)) Then
Allocate(tmp(lb1_:ub1_,lb2_:ub2_),stat=info)
if (info /= psb_success_) then
err=4025
call psb_errpush(err,name, e_err=(/len1*len2/), &
& a_err='integer(psb_mpk_)')
goto 9999
end if
tmp(lb1_:lb1_-1+min(len1,dim),lb2_:lb2_-1+min(len2,dim2)) = &
& rrax(lbi1:lbi1-1+min(len1,dim),lbi2:lbi2-1+min(len2,dim2))
call psb_move_alloc(tmp,rrax,info)
End If
else
dim = 0
dim2 = 0
Allocate(rrax(lb1_:ub1_,lb2_:ub2_),stat=info)
if (info /= psb_success_) then
err=4025
call psb_errpush(err,name, e_err=(/len1*len2/), &
& a_err='integer(psb_mpk_)')
goto 9999
end if
endif
if (present(pad)) then
rrax(lb1_-1+dim+1:lb1_-1+len1,:) = pad
rrax(lb1_:lb1_-1+dim,lb2_-1+dim2+1:lb2_-1+len2) = pad
endif
call psb_erractionrestore(err_act)
return
9999 continue
info = err
call psb_error_handler(err_act)
return
End Subroutine psb_r_em_m_rk2
Subroutine psb_r_m_2_m_rk1(len,rrax,y,info,pad)
use psb_error_mod
! ...Subroutine Arguments
integer(psb_mpk_),Intent(in) :: len
integer(psb_mpk_),allocatable, intent(inout) :: rrax(:),y(:)
integer(psb_ipk_) :: info
integer(psb_mpk_), optional, intent(in) :: pad
character(len=30) :: name
integer(psb_ipk_) :: err_act, err
name='psb_r_m_2_m_rk1'
call psb_erractionsave(err_act)
info=psb_success_
if(psb_errstatus_fatal()) then
info=psb_err_from_subroutine_
goto 9999
end if
call psb_realloc(len,rrax,info,pad=pad)
if (info /= psb_success_) then
err=4000
call psb_errpush(err,name)
goto 9999
end if
call psb_realloc(len,y,info,pad=pad)
if (info /= psb_success_) then
err=4000
call psb_errpush(err,name)
goto 9999
end if
call psb_erractionrestore(err_act)
return
9999 call psb_error_handler(err_act)
return
End Subroutine psb_r_m_2_m_rk1
Subroutine psb_r_e_2_m_rk1(len,rrax,y,info,pad)
use psb_error_mod
! ...Subroutine Arguments
integer(psb_epk_),Intent(in) :: len
integer(psb_mpk_),allocatable, intent(inout) :: rrax(:),y(:)
integer(psb_ipk_) :: info
integer(psb_mpk_), optional, intent(in) :: pad
character(len=30) :: name
integer(psb_ipk_) :: err_act, err
name='psb_r_m_2_m_rk1'
call psb_erractionsave(err_act)
info=psb_success_
if(psb_errstatus_fatal()) then
info=psb_err_from_subroutine_
goto 9999
end if
call psb_realloc(len,rrax,info,pad=pad)
if (info /= psb_success_) then
err=4000
call psb_errpush(err,name)
goto 9999
end if
call psb_realloc(len,y,info,pad=pad)
if (info /= psb_success_) then
err=4000
call psb_errpush(err,name)
goto 9999
end if
call psb_erractionrestore(err_act)
return
9999 call psb_error_handler(err_act)
return
End Subroutine psb_r_e_2_m_rk1
subroutine psb_ab_cpy_m_s(vin,vout,info)
use psb_error_mod
! ...Subroutine Arguments
integer(psb_mpk_), allocatable, intent(in) :: vin
integer(psb_mpk_), allocatable, intent(out) :: vout
integer(psb_ipk_) :: info
! ...Local Variables
integer(psb_ipk_) :: err_act
character(len=30) :: name, char_err
logical, parameter :: debug=.false.
name='psb_ab_cpy_m_s'
call psb_erractionsave(err_act)
info=psb_success_
if(psb_errstatus_fatal()) then
info=psb_err_from_subroutine_
goto 9999
end if
if (allocated(vin)) then
call psb_realloc(vout,info)
if (info /= psb_success_) then
info=psb_err_from_subroutine_
char_err='psb_realloc'
call psb_errpush(info,name,a_err=char_err)
goto 9999
else
vout = vin
endif
endif
call psb_erractionrestore(err_act)
return
9999 call psb_error_handler(err_act)
return
end subroutine psb_ab_cpy_m_s
subroutine psb_ab_cpy_m_rk1(vin,vout,info)
use psb_error_mod
! ...Subroutine Arguments
integer(psb_mpk_), allocatable, intent(in) :: vin(:)
integer(psb_mpk_), allocatable, intent(out) :: vout(:)
integer(psb_ipk_) :: info
! ...Local Variables
integer(psb_ipk_) :: isz,err_act,lb, i
character(len=30) :: name, char_err
logical, parameter :: debug=.false.
name='psb_ab_cpy_m_rk1'
call psb_erractionsave(err_act)
info=psb_success_
if(psb_errstatus_fatal()) then
info=psb_err_from_subroutine_
goto 9999
end if
if (allocated(vin)) then
isz = size(vin)
lb = lbound(vin,1)
call psb_realloc(isz,vout,info,lb=lb)
if (info /= psb_success_) then
info=psb_err_from_subroutine_
char_err='psb_realloc'
call psb_errpush(info,name,a_err=char_err)
goto 9999
else
!$omp parallel do private(i)
do i=lb,lb+isz-1
vout(i) = vin(i)
end do
!$omp end parallel do
endif
endif
call psb_erractionrestore(err_act)
return
9999 call psb_error_handler(err_act)
return
end subroutine psb_ab_cpy_m_rk1
subroutine psb_ab_cpy_m_rk2(vin,vout,info)
use psb_error_mod
! ...Subroutine Arguments
integer(psb_mpk_), allocatable, intent(in) :: vin(:,:)
integer(psb_mpk_), allocatable, intent(out) :: vout(:,:)
integer(psb_ipk_) :: info
! ...Local Variables
integer(psb_ipk_) :: isz1, isz2,err_act, lb1, lb2
character(len=30) :: name, char_err
logical, parameter :: debug=.false.
name='psb_ab_cpy_m_rk2'
call psb_erractionsave(err_act)
info=psb_success_
if(psb_errstatus_fatal()) then
info=psb_err_from_subroutine_
goto 9999
end if
if (allocated(vin)) then
isz1 = size(vin,1)
isz2 = size(vin,2)
lb1 = lbound(vin,1)
lb2 = lbound(vin,2)
call psb_realloc(isz1,isz2,vout,info,lb1=lb1,lb2=lb2)
if (info /= psb_success_) then
info=psb_err_from_subroutine_
char_err='psb_realloc'
call psb_errpush(info,name,a_err=char_err)
goto 9999
else
!$omp workshare
vout(:,:) = vin(:,:)
!$omp end workshare
endif
endif
call psb_erractionrestore(err_act)
return
9999 call psb_error_handler(err_act)
return
end subroutine psb_ab_cpy_m_rk2
subroutine psb_cpy_m_rk1(vin,vout,info)
use psb_error_mod
! ...Subroutine Arguments
integer(psb_mpk_), intent(in) :: vin(:)
integer(psb_mpk_), allocatable, intent(out) :: vout(:)
integer(psb_ipk_) :: info
! ...Local Variables
integer(psb_ipk_) :: isz,err_act,lb
character(len=30) :: name, char_err
logical, parameter :: debug=.false.
name='psb_cpy_m_rk1'
call psb_erractionsave(err_act)
info=psb_success_
if(psb_errstatus_fatal()) then
info=psb_err_from_subroutine_
goto 9999
end if
isz = size(vin)
lb = lbound(vin,1)
call psb_realloc(isz,vout,info,lb=lb)
if (info /= psb_success_) then
info=psb_err_from_subroutine_
char_err='psb_realloc'
call psb_errpush(info,name,a_err=char_err)
goto 9999
else
vout(:) = vin(:)
endif
call psb_erractionrestore(err_act)
return
9999 call psb_error_handler(err_act)
return
end subroutine psb_cpy_m_rk1
subroutine psb_cpy_m_rk2(vin,vout,info)
use psb_error_mod
! ...Subroutine Arguments
integer(psb_mpk_), intent(in) :: vin(:,:)
integer(psb_mpk_), allocatable, intent(out) :: vout(:,:)
integer(psb_ipk_) :: info
! ...Local Variables
integer(psb_ipk_) :: isz1, isz2,err_act, lb1, lb2
character(len=30) :: name, char_err
logical, parameter :: debug=.false.
name='psb_safe_cpy'
call psb_erractionsave(err_act)
info=psb_success_
if(psb_errstatus_fatal()) then
info=psb_err_from_subroutine_
goto 9999
end if
isz1 = size(vin,1)
isz2 = size(vin,2)
lb1 = lbound(vin,1)
lb2 = lbound(vin,2)
call psb_realloc(isz1,isz2,vout,info,lb1=lb1,lb2=lb2)
if (info /= psb_success_) then
info=psb_err_from_subroutine_
char_err='psb_realloc'
call psb_errpush(info,name,a_err=char_err)
goto 9999
else
vout(:,:) = vin(:,:)
endif
call psb_erractionrestore(err_act)
return
9999 call psb_error_handler(err_act)
return
end subroutine psb_cpy_m_rk2
function psb_size_m_rk1(vin) result(val)
integer(psb_epk_) :: val
integer(psb_mpk_), allocatable, intent(in) :: vin(:)
if (.not.allocated(vin)) then
val = 0
else
val = size(vin)
end if
end function psb_size_m_rk1
function psb_size_m_rk2(vin,dim) result(val)
integer(psb_epk_) :: val
integer(psb_mpk_), allocatable, intent(in) :: vin(:,:)
integer(psb_ipk_), optional :: dim
integer(psb_ipk_) :: dim_
if (.not.allocated(vin)) then
val = 0
else
if (present(dim)) then
dim_= dim
val = size(vin,dim=dim_)
else
val = size(vin)
end if
end if
end function psb_size_m_rk2
Subroutine psb_ensure_m_sz_m_rk1(len,v,info,pad,addsz,newsz)
use psb_error_mod
! ...Subroutine Arguments
integer(psb_mpk_),Intent(in) :: len
integer(psb_mpk_),allocatable, intent(inout) :: v(:)
integer(psb_ipk_) :: info
integer(psb_mpk_), optional, intent(in) :: addsz,newsz
integer(psb_mpk_), optional, intent(in) :: pad
! ...Local Variables
character(len=30) :: name
logical, parameter :: debug=.false.
integer(psb_ipk_) :: err_act
integer(psb_mpk_) :: isz
name='psb_ensure_m_sz_m_rk1'
call psb_erractionsave(err_act)
info = psb_success_
if (psb_errstatus_fatal()) then
info=psb_err_from_subroutine_
goto 9999
end if
isz = psb_size(v)
If (len > isz) Then
2 years ago
#if defined(OPENMP)
!$OMP CRITICAL
if (len > isz) then
2 years ago
if (present(newsz)) then
isz = max(len+1,1,newsz)
else if (present(addsz)) then
isz = max(len,1,isz+addsz)
2 years ago
else
isz = max(len,1,int(1.25*isz))
2 years ago
endif
call psb_realloc(isz,v,info,pad=pad)
end if
!$OMP END CRITICAL
if (info /= psb_success_) then
info=psb_err_from_subroutine_
call psb_errpush(info,name,a_err='psb_realloc')
goto 9999
end if
#else
if (len > isz) then
if (present(newsz)) then
isz = max(len+1,1,newsz)
else if (present(addsz)) then
isz = max(len,1,isz+addsz)
else
isz = max(len,1,int(1.25*isz))
endif
call psb_realloc(isz,v,info,pad=pad)
end if
if (info /= psb_success_) then
info=psb_err_from_subroutine_
call psb_errpush(info,name,a_err='psb_realloc')
goto 9999
End If
2 years ago
#endif
end If
call psb_erractionrestore(err_act)
return
9999 call psb_error_handler(err_act)
return
End Subroutine psb_ensure_m_sz_m_rk1
Subroutine psb_ensure_e_sz_m_rk1(len,v,info,pad,addsz,newsz)
use psb_error_mod
! ...Subroutine Arguments
integer(psb_epk_),Intent(in) :: len
integer(psb_mpk_),allocatable, intent(inout) :: v(:)
integer(psb_ipk_) :: info
integer(psb_epk_), optional, intent(in) :: addsz,newsz
integer(psb_mpk_), optional, intent(in) :: pad
! ...Local Variables
character(len=30) :: name
logical, parameter :: debug=.false.
integer(psb_ipk_) :: err_act
integer(psb_epk_) :: isz
name='psb_ensure_m_sz_m_rk1'
call psb_erractionsave(err_act)
info = psb_success_
if (psb_errstatus_fatal()) then
info=psb_err_from_subroutine_
goto 9999
end if
isz = psb_size(v)
If (len > isz) Then
if (present(newsz)) then
isz = max(len+1,1,newsz)
else if (present(addsz)) then
isz = max(len,1,isz+addsz)
else
isz = max(len,1,int(1.25*isz))
endif
call psb_realloc(isz,v,info,pad=pad)
if (info /= psb_success_) then
info=psb_err_from_subroutine_
call psb_errpush(info,name,a_err='psb_realloc')
goto 9999
End If
end If
call psb_erractionrestore(err_act)
return
9999 call psb_error_handler(err_act)
return
End Subroutine psb_ensure_e_sz_m_rk1
Subroutine psb_move_alloc_m_rk1(vin,vout,info)
use psb_error_mod
integer(psb_mpk_), allocatable, intent(inout) :: vin(:),vout(:)
integer(psb_ipk_), intent(out) :: info
!
!
info=psb_success_
call move_alloc(vin,vout)
end Subroutine psb_move_alloc_m_rk1
Subroutine psb_move_alloc_m_rk2(vin,vout,info)
use psb_error_mod
integer(psb_mpk_), allocatable, intent(inout) :: vin(:,:),vout(:,:)
integer(psb_ipk_), intent(out) :: info
!
!
info=psb_success_
call move_alloc(vin,vout)
end Subroutine psb_move_alloc_m_rk2
end module psb_m_realloc_mod