From 51084868185ec76d0b3bb0c3d3d7f607d9d335ff Mon Sep 17 00:00:00 2001 From: Salvatore Filippone Date: Mon, 7 Jan 2008 17:09:41 +0000 Subject: [PATCH] Split ldsc_pre_halo, preparing for restructuring of bldext. --- base/internals/psi_bld_hash.f90 | 144 ++++++++++++++++++++++++++++ base/internals/psi_bld_tmphalo.f90 | 148 +++++++++++++++++++++++++++++ 2 files changed, 292 insertions(+) create mode 100644 base/internals/psi_bld_hash.f90 create mode 100644 base/internals/psi_bld_tmphalo.f90 diff --git a/base/internals/psi_bld_hash.f90 b/base/internals/psi_bld_hash.f90 new file mode 100644 index 00000000..039c3b88 --- /dev/null +++ b/base/internals/psi_bld_hash.f90 @@ -0,0 +1,144 @@ +!!$ +!!$ 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: psi_bld_hash.f90 +! +! Subroutine: psi_bld_hash +! Convers the AVL tree data structure into the hashed list +! of ordered sublists. +! +! +! Arguments: +! desc - type(psb_desc_type). The communication descriptor. +! info - integer. return code. +! +subroutine psi_bld_hash(desc,info) + use psb_descriptor_type + use psb_serial_mod + use psb_const_mod + use psb_error_mod + use psb_penv_mod + use psb_realloc_mod + use psi_mod, psb_protect_name => psi_bld_hash + implicit none + type(psb_desc_type), intent(inout) :: desc + integer, intent(out) :: info + + integer :: i,j,np,me,lhalo,nhalo,& + & n_col, err_act, key, ih, nh, idx, nk,icomm + integer :: ictxt,n_row + character(len=20) :: name,ch_err + + info = 0 + name = 'psi_bld_hash' + call psb_erractionsave(err_act) + + ictxt = psb_cd_get_context(desc) + icomm = psb_cd_get_mpic(desc) + n_row = psb_cd_get_local_rows(desc) + n_col = psb_cd_get_local_cols(desc) + + ! check on blacs grid + call psb_info(ictxt, me, np) + if (np == -1) then + info = 2010 + call psb_errpush(info,name) + goto 9999 + endif + + + if (.not.(psb_is_bld_desc(desc).and.psb_is_large_desc(desc))) then + info = 1122 + call psb_errpush(info,name) + goto 9999 + end if + + + nk = n_col + call psb_realloc(nk,2,desc%glb_lc,info) + if (info ==0) call psb_realloc(psb_hash_size+1,desc%hashv,info,lb=0) + if (info /= 0) then + ch_err='psb_realloc' + call psb_errpush(info,name,a_err=ch_err) + goto 9999 + end if + + ! Build a hashed table of sorted lists to search for + ! indices. + desc%hashv(0:psb_hash_size) = 0 + do i=1, nk + key = desc%loc_to_glob(i) + ih = iand(key,psb_hash_mask) + desc%hashv(ih) = desc%hashv(ih) + 1 + end do + nh = desc%hashv(0) + idx = 1 + do i=1, psb_hash_size + desc%hashv(i-1) = idx + idx = idx + nh + nh = desc%hashv(i) + end do + do i=1, nk + key = desc%loc_to_glob(i) + ih = iand(key,psb_hash_mask) + idx = desc%hashv(ih) + desc%glb_lc(idx,1) = key + desc%glb_lc(idx,2) = i + desc%hashv(ih) = desc%hashv(ih) + 1 + end do + do i = psb_hash_size, 1, -1 + desc%hashv(i) = desc%hashv(i-1) + end do + desc%hashv(0) = 1 + do i=0, psb_hash_size-1 + idx = desc%hashv(i) + nh = desc%hashv(i+1) - desc%hashv(i) + if (nh > 1) then + call psb_msort(desc%glb_lc(idx:idx+nh-1,1),& + & ix=desc%glb_lc(idx:idx+nh-1,2),flag=psb_sort_keep_idx_) + end if + end do + + call psb_erractionrestore(err_act) + return + +9999 continue + call psb_erractionrestore(err_act) + + if (err_act.eq.psb_act_ret_) then + return + else + call psb_error(ictxt) + end if + return + + +end subroutine psi_bld_hash diff --git a/base/internals/psi_bld_tmphalo.f90 b/base/internals/psi_bld_tmphalo.f90 new file mode 100644 index 00000000..2d3b8732 --- /dev/null +++ b/base/internals/psi_bld_tmphalo.f90 @@ -0,0 +1,148 @@ +!!$ +!!$ 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: psi_bld_tmphalo.f90 +! +! Subroutine: psi_bld_tmphalo +! Build initial versions of data exchange lists. +! When the descriptor is for a large index space, we cannot build +! the data exchange lists "on-the-fly", but we also want to keep using the +! same format conversion routines we use in the small index space case, +! hence this adapter routine. +! +! +! Arguments: +! desc - type(psb_desc_type). The communication descriptor. +! info - integer. return code. +! +subroutine psi_bld_tmphalo(desc,info) + use psb_descriptor_type + use psb_serial_mod + use psb_const_mod + use psb_error_mod + use psb_penv_mod + use psb_realloc_mod + use psi_mod, psb_protect_name => psi_bld_tmphalo + implicit none + type(psb_desc_type), intent(inout) :: desc + integer, intent(out) :: info + + integer,allocatable :: helem(:),hproc(:) + integer,allocatable :: tmphl(:) + + integer :: i,j,np,me,lhalo,nhalo,& + & n_col, err_act, key, ih, nh, idx, nk,icomm + integer :: ictxt,n_row + character(len=20) :: name,ch_err + + info = 0 + name = 'psi_bld_tmphalo' + call psb_erractionsave(err_act) + + ictxt = psb_cd_get_context(desc) + icomm = psb_cd_get_mpic(desc) + n_row = psb_cd_get_local_rows(desc) + n_col = psb_cd_get_local_cols(desc) + + ! check on blacs grid + call psb_info(ictxt, me, np) + if (np == -1) then + info = 2010 + call psb_errpush(info,name) + goto 9999 + endif + + + if (.not.(psb_is_bld_desc(desc).and.psb_is_large_desc(desc))) then + info = 1122 + call psb_errpush(info,name) + goto 9999 + end if + + ! Here we do not know yet who owns what, so we have + ! to call fnd_owner. + nh = (n_col-n_row) + if (nh > 0) then + Allocate(helem(nh),stat=info) + if (info /= 0) then + call psb_errpush(4010,name,a_err='Allocate') + goto 9999 + end if + + do i=1, nh + helem(i) = desc%loc_to_glob(n_row+i) + end do + + call psi_fnd_owner(nh,helem,hproc,desc,info) + allocate(tmphl((3*((n_col-n_row)+1)+1)),stat=info) + if (info /= 0) then + call psb_errpush(4010,name,a_err='Allocate') + goto 9999 + end if + j = 1 + do i=1,nh + tmphl(j+0) = hproc(i) + if (tmphl(j+0)<0) then + write(0,*) 'Unrecoverable error: missing proc from asb' + end if + tmphl(j+1) = 1 + tmphl(j+2) = n_row+i + j = j + 3 + end do + tmphl(j) = -1 + lhalo = j + nhalo = (lhalo-1)/3 + else + allocate(tmphl(1),stat=info) + if (info /= 0) then + call psb_errpush(4010,name,a_err='Allocate') + goto 9999 + end if + tmphl=-1 + endif + + call psb_transfer(tmphl,desc%halo_index,info) + + call psb_erractionrestore(err_act) + return + +9999 continue + call psb_erractionrestore(err_act) + + if (err_act.eq.psb_act_ret_) then + return + else + call psb_error(ictxt) + end if + return + + +end subroutine psi_bld_tmphalo