!!$ !!$ Parallel Sparse BLAS v2.0 !!$ (C) Copyright 2006 Salvatore Filippone University of Rome Tor Vergata !!$ Alfredo Buttari University of Rome Tor Vergata !!$ !!$ 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. !!$ !!$ ! ! File: psb_zallc.f90 ! ! Function: psb_zalloc ! Allocates dense matrix for PSBLAS routines ! ! Parameters: ! x - the matrix to be allocated. ! desc_a - the communication descriptor. ! info - possibly returns an error code ! n - optional number of columns. subroutine psb_zalloc(x, desc_a, info, n) !....allocate dense matrix for psblas routines..... use psb_descriptor_type use psb_const_mod use psb_error_mod use psb_penv_mod implicit none !....parameters... complex(kind(1.d0)), allocatable, intent(out) :: x(:,:) type(psb_desc_type), intent(in) :: desc_a integer :: info integer, optional, intent(in) :: n !locals integer :: np,me,err,n_col,n_row,i,j,err_act integer :: ictxt,n_ integer :: int_err(5),exch(3) character(len=20) :: name, ch_err name='psb_zallc' if(psb_get_errstatus() /= 0) return info=0 err=0 int_err(1)=0 call psb_erractionsave(err_act) ictxt=psb_cd_get_context(desc_a) call psb_info(ictxt, me, np) if (np == -1) then info = 2010 call psb_errpush(info,name) goto 9999 endif !... check m and n parameters.... if (.not.psb_is_ok_desc(desc_a)) then info = 3110 call psb_errpush(info,name) goto 9999 endif if (present(n)) then n_ = n else n_ = 1 endif !global check on n parameters if (me == psb_root_) then exch(1)=n_ call psb_bcast(ictxt,exch(1),root=psb_root_) else call psb_bcast(ictxt,exch(1),root=psb_root_) if (exch(1) /= n_) then info=550 int_err(1)=1 call psb_errpush(info,name,int_err) goto 9999 endif endif !....allocate x ..... if (psb_is_asb_desc(desc_a).or.psb_is_upd_desc(desc_a)) then n_col = max(1,psb_cd_get_local_cols(desc_a)) allocate(x(n_col,n_),stat=info) if (info /= 0) then info=4010 ch_err='allocate' call psb_errpush(info,name,a_err=ch_err) goto 9999 endif do j=1,n_ do i=1,n_col x(i,j) = 0.0d0 end do end do else if (psb_is_bld_desc(desc_a)) then n_row = max(1,psb_cd_get_local_rows(desc_a)) allocate(x(n_row,n_),stat=info) if (info /= 0) then info=4010 ch_err='allocate' call psb_errpush(info,name,a_err=ch_err) goto 9999 endif do j = 1, n_ do i=1,n_row x(i,j) = 0.0d0 end do end do endif call psb_erractionrestore(err_act) return 9999 continue call psb_erractionrestore(err_act) if (err_act == act_abort) then call psb_error(ictxt) return end if return end subroutine psb_zalloc !!$ !!$ Parallel Sparse BLAS v2.0 !!$ (C) Copyright 2006 Salvatore Filippone University of Rome Tor Vergata !!$ Alfredo Buttari University of Rome Tor Vergata !!$ !!$ 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. !!$ !!$ ! ! Function: psb_zallocv ! Allocates dense matrix for PSBLAS routines ! ! Parameters: ! x - the matrix to be allocated. ! desc_a - the communication descriptor. ! info - possibly returns an error code subroutine psb_zallocv(x, desc_a,info,n) !....allocate sparse matrix structure for psblas routines..... use psb_descriptor_type use psb_const_mod use psb_realloc_mod use psb_error_mod use psb_penv_mod implicit none !....parameters... complex(kind(1.d0)), allocatable, intent(out) :: x(:) type(psb_desc_type), intent(in) :: desc_a integer :: info integer, optional, intent(in) :: n !locals integer :: np,me,n_col,n_row,i,err_act integer :: ictxt, n_ logical, parameter :: debug=.false. character(len=20) :: name, ch_err if(psb_get_errstatus() /= 0) return info=0 name='psb_zallcv' call psb_erractionsave(err_act) ictxt=psb_cd_get_context(desc_a) call psb_info(ictxt, me, np) ! ....verify blacs grid correctness.. if (np == -1) then info = 2010 call psb_errpush(info,name) goto 9999 endif if (debug) write(0,*) 'dall: is_ok?',psb_is_ok_desc(desc_a) !... check m and n parameters.... if (.not.psb_is_ok_desc(desc_a)) then info = 3110 call psb_errpush(info,name) goto 9999 endif ! As this is a rank-1 array, optional parameter N is actually ignored. !....allocate x ..... if (psb_is_asb_desc(desc_a).or.psb_is_upd_desc(desc_a)) then n_col = max(1,psb_cd_get_local_cols(desc_a)) call psb_realloc(n_col,x,info) if (info /= 0) then info=4010 ch_err='psb_realloc' call psb_errpush(info,name,a_err=ch_err) goto 9999 endif do i=1,n_col x(i) = 0.0d0 end do else if (psb_is_bld_desc(desc_a)) then n_row = max(1,psb_cd_get_local_rows(desc_a)) call psb_realloc(n_row,x,info) if (info /= 0) then info=4010 ch_err='psb_realloc' call psb_errpush(info,name,a_err=ch_err) goto 9999 endif do i=1,n_row x(i) = 0.0d0 end do endif call psb_erractionrestore(err_act) return 9999 continue call psb_erractionrestore(err_act) if (err_act == act_abort) then call psb_error(ictxt) return end if return end subroutine psb_zallocv