fdurastante
/
amg4psblas
mirror of https://github.com/sfilippone/amg4psblas.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

added use parmatch aggregator

Browse Source
mergeparmatch
Cirdans-Home 4 years ago
parent ee9ea93c2a
commit 816c59d994
6 changed files with 603 additions and 592 deletions
Show Stats Download Patch File Download Diff File

6
amgprec/amg_base_prec_type.F90
Unescape Escape View File

@ -509,7 +509,11 @@ contains
val = amg_soc2_ val = amg_soc2_
case('SOC1') case('SOC1')
val = amg_soc1_ val = amg_soc1_
case('DEC') case('MATCHBOXP','PARMATCH')
val = amg_matchboxp_
case('COUPLED','COUP')
val = amg_coupled_aggr_
case('DEC','DECOUPLED')
val = amg_dec_aggr_ val = amg_dec_aggr_
case('SYMDEC') case('SYMDEC')
val = amg_sym_dec_aggr_ val = amg_sym_dec_aggr_

295
amgprec/amg_c_onelev_mod.f90
Unescape Escape View File

@ -1,15 +1,15 @@
! !
! !
! AMG4PSBLAS version 1.0 ! AMG4PSBLAS version 1.0
! Algebraic Multigrid Package ! Algebraic Multigrid Package
! based on PSBLAS (Parallel Sparse BLAS version 3.7) ! based on PSBLAS (Parallel Sparse BLAS version 3.7)
! !
! (C) Copyright 2021 ! (C) Copyright 2021
! !
! Salvatore Filippone ! Salvatore Filippone
! Pasqua D'Ambra ! Pasqua D'Ambra
! Fabio Durastante ! Fabio Durastante
! !
! Redistribution and use in source and binary forms, with or without ! Redistribution and use in source and binary forms, with or without
! modification, are permitted provided that the following conditions ! modification, are permitted provided that the following conditions
! are met: ! are met:
@ -21,7 +21,7 @@
! 3. The name of the AMG4PSBLAS group or the names of its contributors may ! 3. The name of the AMG4PSBLAS group or the names of its contributors may
! not be used to endorse or promote products derived from this ! not be used to endorse or promote products derived from this
! software without specific written permission. ! software without specific written permission.
! !
! THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS ! THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
! ``AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED ! ``AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED
! TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR ! TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
@ -33,22 +33,22 @@
! CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ! 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 ! ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
! POSSIBILITY OF SUCH DAMAGE. ! POSSIBILITY OF SUCH DAMAGE.
! !
! !
! File: amg_c_onelev_mod.f90 ! File: amg_c_onelev_mod.f90
! !
! Module: amg_c_onelev_mod ! Module: amg_c_onelev_mod
! !
! This module defines: ! This module defines:
! - the amg_c_onelev_type data structure containing one level ! - the amg_c_onelev_type data structure containing one level
! of a multilevel preconditioner and related ! of a multilevel preconditioner and related
! data structures; ! data structures;
! !
! It contains routines for ! It contains routines for
! - Building and applying; ! - Building and applying;
! - checking if the preconditioner is correctly defined; ! - checking if the preconditioner is correctly defined;
! - printing a description of the preconditioner; ! - printing a description of the preconditioner;
! - deallocating the preconditioner data structure. ! - deallocating the preconditioner data structure.
! !
module amg_c_onelev_mod module amg_c_onelev_mod
@ -56,6 +56,7 @@ module amg_c_onelev_mod
use amg_base_prec_type use amg_base_prec_type
use amg_c_base_smoother_mod use amg_c_base_smoother_mod
use amg_c_dec_aggregator_mod use amg_c_dec_aggregator_mod
use psb_base_mod, only : psb_cspmat_type, psb_c_vect_type, & use psb_base_mod, only : psb_cspmat_type, psb_c_vect_type, &
& psb_c_base_vect_type, psb_lcspmat_type, psb_clinmap_type, psb_spk_, & & psb_c_base_vect_type, psb_lcspmat_type, psb_clinmap_type, psb_spk_, &
& psb_ipk_, psb_epk_, psb_lpk_, psb_desc_type, psb_i_base_vect_type, & & psb_ipk_, psb_epk_, psb_lpk_, psb_desc_type, psb_i_base_vect_type, &
@ -73,11 +74,11 @@ module amg_c_onelev_mod
! class(amg_c_base_smoother_type), pointer :: sm2 => null() ! class(amg_c_base_smoother_type), pointer :: sm2 => null()
! class(amg_cmlprec_wrk_type), allocatable :: wrk ! class(amg_cmlprec_wrk_type), allocatable :: wrk
! class(amg_c_base_aggregator_type), allocatable :: aggr ! class(amg_c_base_aggregator_type), allocatable :: aggr
! type(amg_sml_parms) :: parms ! type(amg_sml_parms) :: parms
! type(psb_cspmat_type) :: ac ! type(psb_cspmat_type) :: ac
! type(psb_cesc_type) :: desc_ac ! type(psb_cesc_type) :: desc_ac
! type(psb_cspmat_type), pointer :: base_a => null() ! type(psb_cspmat_type), pointer :: base_a => null()
! type(psb_desc_type), pointer :: base_desc => null() ! type(psb_desc_type), pointer :: base_desc => null()
! type(psb_clinmap_type) :: map ! type(psb_clinmap_type) :: map
! end type amg_conelev_type ! end type amg_conelev_type
! !
@ -93,7 +94,7 @@ module amg_c_onelev_mod
! Workspace for application of preconditioner; may be ! Workspace for application of preconditioner; may be
! pre-allocated to save time in the application within a ! pre-allocated to save time in the application within a
! Krylov solver. ! Krylov solver.
! aggr - class(amg_c_base_aggregator_type), allocatable ! aggr - class(amg_c_base_aggregator_type), allocatable
! The aggregator object: holds the algorithmic choices and ! The aggregator object: holds the algorithmic choices and
! (possibly) additional data for building the aggregation. ! (possibly) additional data for building the aggregation.
! parms - type(amg_sml_parms) ! parms - type(amg_sml_parms)
@ -104,7 +105,7 @@ module amg_c_onelev_mod
! The communication descriptor associated to the matrix ! The communication descriptor associated to the matrix
! stored in ac. ! stored in ac.
! base_a - type(psb_cspmat_type), pointer. ! base_a - type(psb_cspmat_type), pointer.
! Pointer (really a pointer!) to the local part of the current ! Pointer (really a pointer!) to the local part of the current
! matrix (so we have a unified treatment of residuals). ! matrix (so we have a unified treatment of residuals).
! We need this to avoid passing explicitly the current matrix ! We need this to avoid passing explicitly the current matrix
! to the routine which applies the preconditioner. ! to the routine which applies the preconditioner.
@ -115,13 +116,13 @@ module amg_c_onelev_mod
! vector spaces associated to the index spaces of the previous ! vector spaces associated to the index spaces of the previous
! and current levels. ! and current levels.
! !
! Methods: ! Methods:
! Most methods follow the encapsulation hierarchy: they take whatever action ! Most methods follow the encapsulation hierarchy: they take whatever action
! is appropriate for the current object, then call the corresponding method for ! is appropriate for the current object, then call the corresponding method for
! the contained object. ! the contained object.
! As an example: the descr() method prints out a description of the ! As an example: the descr() method prints out a description of the
! level. It starts by invoking the descr() method of the parms object, ! level. It starts by invoking the descr() method of the parms object,
! then calls the descr() method of the smoother object. ! then calls the descr() method of the smoother object.
! !
! descr - Prints a description of the object. ! descr - Prints a description of the object.
! default - Set default values ! default - Set default values
@ -130,14 +131,14 @@ module amg_c_onelev_mod
! it is passed to the smoother object for further processing. ! it is passed to the smoother object for further processing.
! check - Sanity checks. ! check - Sanity checks.
! sizeof - Total memory occupation in bytes ! sizeof - Total memory occupation in bytes
! get_nzeros - Number of nonzeros ! get_nzeros - Number of nonzeros
! get_wrksz - How many workspace vector does apply_vect need ! get_wrksz - How many workspace vector does apply_vect need
! allocate_wrk - Allocate auxiliary workspace ! allocate_wrk - Allocate auxiliary workspace
! free_wrk - Free auxiliary workspace ! free_wrk - Free auxiliary workspace
! bld_tprol - Invoke the aggr method to build the tentative prolongator ! bld_tprol - Invoke the aggr method to build the tentative prolongator
! mat_asb - Build the final (possibly smoothed) prolongator and coarse matrix. ! mat_asb - Build the final (possibly smoothed) prolongator and coarse matrix.
!
! !
!
type amg_cmlprec_wrk_type type amg_cmlprec_wrk_type
complex(psb_spk_), allocatable :: tx(:), ty(:), x2l(:), y2l(:) complex(psb_spk_), allocatable :: tx(:), ty(:), x2l(:), y2l(:)
type(psb_c_vect_type) :: vtx, vty, vx2l, vy2l type(psb_c_vect_type) :: vtx, vty, vx2l, vy2l
@ -148,10 +149,10 @@ module amg_c_onelev_mod
procedure, pass(wk) :: clone => c_wrk_clone procedure, pass(wk) :: clone => c_wrk_clone
procedure, pass(wk) :: move_alloc => c_wrk_move_alloc procedure, pass(wk) :: move_alloc => c_wrk_move_alloc
procedure, pass(wk) :: cnv => c_wrk_cnv procedure, pass(wk) :: cnv => c_wrk_cnv
procedure, pass(wk) :: sizeof => c_wrk_sizeof procedure, pass(wk) :: sizeof => c_wrk_sizeof
end type amg_cmlprec_wrk_type end type amg_cmlprec_wrk_type
private :: c_wrk_alloc, c_wrk_free, & private :: c_wrk_alloc, c_wrk_free, &
& c_wrk_clone, c_wrk_move_alloc, c_wrk_cnv, c_wrk_sizeof & c_wrk_clone, c_wrk_move_alloc, c_wrk_cnv, c_wrk_sizeof
type amg_c_remap_data_type type amg_c_remap_data_type
type(psb_cspmat_type) :: ac_pre_remap type(psb_cspmat_type) :: ac_pre_remap
@ -161,19 +162,19 @@ module amg_c_onelev_mod
contains contains
procedure, pass(rmp) :: clone => c_remap_data_clone procedure, pass(rmp) :: clone => c_remap_data_clone
end type amg_c_remap_data_type end type amg_c_remap_data_type
type amg_c_onelev_type type amg_c_onelev_type
class(amg_c_base_smoother_type), allocatable :: sm, sm2a class(amg_c_base_smoother_type), allocatable :: sm, sm2a
class(amg_c_base_smoother_type), pointer :: sm2 => null() class(amg_c_base_smoother_type), pointer :: sm2 => null()
class(amg_cmlprec_wrk_type), allocatable :: wrk class(amg_cmlprec_wrk_type), allocatable :: wrk
class(amg_c_base_aggregator_type), allocatable :: aggr class(amg_c_base_aggregator_type), allocatable :: aggr
type(amg_sml_parms) :: parms type(amg_sml_parms) :: parms
type(psb_cspmat_type) :: ac type(psb_cspmat_type) :: ac
integer(psb_ipk_) :: ac_nz_loc integer(psb_ipk_) :: ac_nz_loc
integer(psb_lpk_) :: ac_nz_tot integer(psb_lpk_) :: ac_nz_tot
type(psb_desc_type) :: desc_ac type(psb_desc_type) :: desc_ac
type(psb_cspmat_type), pointer :: base_a => null() type(psb_cspmat_type), pointer :: base_a => null()
type(psb_desc_type), pointer :: base_desc => null() type(psb_desc_type), pointer :: base_desc => null()
type(psb_lcspmat_type) :: tprol type(psb_lcspmat_type) :: tprol
type(psb_clinmap_type) :: linmap type(psb_clinmap_type) :: linmap
type(amg_c_remap_data_type) :: remap_data type(amg_c_remap_data_type) :: remap_data
@ -197,7 +198,7 @@ module amg_c_onelev_mod
procedure, pass(lv) :: setsm => amg_c_base_onelev_setsm procedure, pass(lv) :: setsm => amg_c_base_onelev_setsm
procedure, pass(lv) :: setsv => amg_c_base_onelev_setsv procedure, pass(lv) :: setsv => amg_c_base_onelev_setsv
procedure, pass(lv) :: setag => amg_c_base_onelev_setag procedure, pass(lv) :: setag => amg_c_base_onelev_setag
generic, public :: set => cseti, csetr, csetc, setsm, setsv, setag generic, public :: set => cseti, csetr, csetc, setsm, setsv, setag
procedure, pass(lv) :: sizeof => c_base_onelev_sizeof procedure, pass(lv) :: sizeof => c_base_onelev_sizeof
procedure, pass(lv) :: get_nzeros => c_base_onelev_get_nzeros procedure, pass(lv) :: get_nzeros => c_base_onelev_get_nzeros
procedure, pass(lv) :: get_wrksz => c_base_onelev_get_wrksize procedure, pass(lv) :: get_wrksz => c_base_onelev_get_wrksize
@ -205,8 +206,8 @@ module amg_c_onelev_mod
procedure, pass(lv) :: free_wrk => c_base_onelev_free_wrk procedure, pass(lv) :: free_wrk => c_base_onelev_free_wrk
procedure, nopass :: stringval => amg_stringval procedure, nopass :: stringval => amg_stringval
procedure, pass(lv) :: move_alloc => c_base_onelev_move_alloc procedure, pass(lv) :: move_alloc => c_base_onelev_move_alloc
procedure, pass(lv) :: map_rstr_a => amg_c_base_onelev_map_rstr_a procedure, pass(lv) :: map_rstr_a => amg_c_base_onelev_map_rstr_a
procedure, pass(lv) :: map_prol_a => amg_c_base_onelev_map_prol_a procedure, pass(lv) :: map_prol_a => amg_c_base_onelev_map_prol_a
procedure, pass(lv) :: map_rstr_v => amg_c_base_onelev_map_rstr_v procedure, pass(lv) :: map_rstr_v => amg_c_base_onelev_map_rstr_v
@ -226,11 +227,11 @@ module amg_c_onelev_mod
& c_base_onelev_get_wrksize, c_base_onelev_allocate_wrk, & & c_base_onelev_get_wrksize, c_base_onelev_allocate_wrk, &
& c_base_onelev_free_wrk & c_base_onelev_free_wrk
interface interface
subroutine amg_c_base_onelev_mat_asb(lv,a,desc_a,ilaggr,nlaggr,t_prol,info) subroutine amg_c_base_onelev_mat_asb(lv,a,desc_a,ilaggr,nlaggr,t_prol,info)
import :: psb_cspmat_type, psb_desc_type, psb_spk_, psb_ipk_, psb_lcspmat_type, psb_lpk_ import :: psb_cspmat_type, psb_desc_type, psb_spk_, psb_ipk_, psb_lcspmat_type, psb_lpk_
import :: amg_c_onelev_type import :: amg_c_onelev_type
implicit none implicit none
class(amg_c_onelev_type), intent(inout), target :: lv class(amg_c_onelev_type), intent(inout), target :: lv
type(psb_cspmat_type), intent(in) :: a type(psb_cspmat_type), intent(in) :: a
type(psb_desc_type), intent(inout) :: desc_a type(psb_desc_type), intent(inout) :: desc_a
@ -255,7 +256,7 @@ module amg_c_onelev_mod
end subroutine amg_c_base_onelev_build end subroutine amg_c_base_onelev_build
end interface end interface
interface interface
subroutine amg_c_base_onelev_descr(lv,il,nl,ilmin,info,iout, verbosity) subroutine amg_c_base_onelev_descr(lv,il,nl,ilmin,info,iout, verbosity)
import :: psb_cspmat_type, psb_c_vect_type, psb_c_base_vect_type, & import :: psb_cspmat_type, psb_c_vect_type, psb_c_base_vect_type, &
& psb_clinmap_type, psb_spk_, amg_c_onelev_type, & & psb_clinmap_type, psb_spk_, amg_c_onelev_type, &
@ -270,127 +271,127 @@ module amg_c_onelev_mod
end subroutine amg_c_base_onelev_descr end subroutine amg_c_base_onelev_descr
end interface end interface
interface interface
subroutine amg_c_base_onelev_cnv(lv,info,amold,vmold,imold) subroutine amg_c_base_onelev_cnv(lv,info,amold,vmold,imold)
import :: amg_c_onelev_type, psb_c_base_vect_type, psb_spk_, & import :: amg_c_onelev_type, psb_c_base_vect_type, psb_spk_, &
& psb_c_base_sparse_mat, psb_ipk_, psb_i_base_vect_type & psb_c_base_sparse_mat, psb_ipk_, psb_i_base_vect_type
! Arguments ! Arguments
class(amg_c_onelev_type), intent(inout) :: lv class(amg_c_onelev_type), intent(inout) :: lv
integer(psb_ipk_), intent(out) :: info integer(psb_ipk_), intent(out) :: info
class(psb_c_base_sparse_mat), intent(in), optional :: amold class(psb_c_base_sparse_mat), intent(in), optional :: amold
class(psb_c_base_vect_type), intent(in), optional :: vmold class(psb_c_base_vect_type), intent(in), optional :: vmold
class(psb_i_base_vect_type), intent(in), optional :: imold class(psb_i_base_vect_type), intent(in), optional :: imold
end subroutine amg_c_base_onelev_cnv end subroutine amg_c_base_onelev_cnv
end interface end interface
interface interface
subroutine amg_c_base_onelev_free(lv,info) subroutine amg_c_base_onelev_free(lv,info)
import :: psb_cspmat_type, psb_c_vect_type, psb_c_base_vect_type, & import :: psb_cspmat_type, psb_c_vect_type, psb_c_base_vect_type, &
& psb_clinmap_type, psb_spk_, amg_c_onelev_type, & & psb_clinmap_type, psb_spk_, amg_c_onelev_type, &
& psb_ipk_, psb_epk_, psb_desc_type & psb_ipk_, psb_epk_, psb_desc_type
implicit none implicit none
class(amg_c_onelev_type), intent(inout) :: lv class(amg_c_onelev_type), intent(inout) :: lv
integer(psb_ipk_), intent(out) :: info integer(psb_ipk_), intent(out) :: info
end subroutine amg_c_base_onelev_free end subroutine amg_c_base_onelev_free
end interface end interface
interface interface
subroutine amg_c_base_onelev_check(lv,info) subroutine amg_c_base_onelev_check(lv,info)
import :: psb_cspmat_type, psb_c_vect_type, psb_c_base_vect_type, & import :: psb_cspmat_type, psb_c_vect_type, psb_c_base_vect_type, &
& psb_clinmap_type, psb_spk_, amg_c_onelev_type, & & psb_clinmap_type, psb_spk_, amg_c_onelev_type, &
& psb_ipk_, psb_epk_, psb_desc_type & psb_ipk_, psb_epk_, psb_desc_type
Implicit None Implicit None
! Arguments ! Arguments
class(amg_c_onelev_type), intent(inout) :: lv class(amg_c_onelev_type), intent(inout) :: lv
integer(psb_ipk_), intent(out) :: info integer(psb_ipk_), intent(out) :: info
end subroutine amg_c_base_onelev_check end subroutine amg_c_base_onelev_check
end interface end interface
interface interface
subroutine amg_c_base_onelev_setsm(lv,val,info,pos) subroutine amg_c_base_onelev_setsm(lv,val,info,pos)
import :: psb_spk_, amg_c_onelev_type, amg_c_base_smoother_type, & import :: psb_spk_, amg_c_onelev_type, amg_c_base_smoother_type, &
& psb_ipk_, psb_epk_, psb_desc_type & psb_ipk_, psb_epk_, psb_desc_type
Implicit None Implicit None
! Arguments ! Arguments
class(amg_c_onelev_type), target, intent(inout) :: lv class(amg_c_onelev_type), target, intent(inout) :: lv
class(amg_c_base_smoother_type), intent(in) :: val class(amg_c_base_smoother_type), intent(in) :: val
integer(psb_ipk_), intent(out) :: info integer(psb_ipk_), intent(out) :: info
character(len=*), optional, intent(in) :: pos character(len=*), optional, intent(in) :: pos
end subroutine amg_c_base_onelev_setsm end subroutine amg_c_base_onelev_setsm
end interface end interface
interface interface
subroutine amg_c_base_onelev_setsv(lv,val,info,pos) subroutine amg_c_base_onelev_setsv(lv,val,info,pos)
import :: psb_spk_, amg_c_onelev_type, amg_c_base_solver_type, & import :: psb_spk_, amg_c_onelev_type, amg_c_base_solver_type, &
& psb_ipk_, psb_epk_, psb_desc_type & psb_ipk_, psb_epk_, psb_desc_type
Implicit None Implicit None
! Arguments ! Arguments
class(amg_c_onelev_type), target, intent(inout) :: lv class(amg_c_onelev_type), target, intent(inout) :: lv
class(amg_c_base_solver_type), intent(in) :: val class(amg_c_base_solver_type), intent(in) :: val
integer(psb_ipk_), intent(out) :: info integer(psb_ipk_), intent(out) :: info
character(len=*), optional, intent(in) :: pos character(len=*), optional, intent(in) :: pos
end subroutine amg_c_base_onelev_setsv end subroutine amg_c_base_onelev_setsv
end interface end interface
interface interface
subroutine amg_c_base_onelev_setag(lv,val,info,pos) subroutine amg_c_base_onelev_setag(lv,val,info,pos)
import :: psb_spk_, amg_c_onelev_type, amg_c_base_aggregator_type, & import :: psb_spk_, amg_c_onelev_type, amg_c_base_aggregator_type, &
& psb_ipk_, psb_epk_, psb_desc_type & psb_ipk_, psb_epk_, psb_desc_type
Implicit None Implicit None
! Arguments ! Arguments
class(amg_c_onelev_type), target, intent(inout) :: lv class(amg_c_onelev_type), target, intent(inout) :: lv
class(amg_c_base_aggregator_type), intent(in) :: val class(amg_c_base_aggregator_type), intent(in) :: val
integer(psb_ipk_), intent(out) :: info integer(psb_ipk_), intent(out) :: info
character(len=*), optional, intent(in) :: pos character(len=*), optional, intent(in) :: pos
end subroutine amg_c_base_onelev_setag end subroutine amg_c_base_onelev_setag
end interface end interface
interface interface
subroutine amg_c_base_onelev_cseti(lv,what,val,info,pos,idx) subroutine amg_c_base_onelev_cseti(lv,what,val,info,pos,idx)
import :: psb_cspmat_type, psb_c_vect_type, psb_c_base_vect_type, & import :: psb_cspmat_type, psb_c_vect_type, psb_c_base_vect_type, &
& psb_clinmap_type, psb_spk_, amg_c_onelev_type, & & psb_clinmap_type, psb_spk_, amg_c_onelev_type, &
& psb_ipk_, psb_epk_, psb_desc_type & psb_ipk_, psb_epk_, psb_desc_type
Implicit None Implicit None
! Arguments ! Arguments
class(amg_c_onelev_type), intent(inout) :: lv class(amg_c_onelev_type), intent(inout) :: lv
character(len=*), intent(in) :: what character(len=*), intent(in) :: what
integer(psb_ipk_), intent(in) :: val integer(psb_ipk_), intent(in) :: val
integer(psb_ipk_), intent(out) :: info integer(psb_ipk_), intent(out) :: info
character(len=*), optional, intent(in) :: pos character(len=*), optional, intent(in) :: pos
integer(psb_ipk_), intent(in), optional :: idx integer(psb_ipk_), intent(in), optional :: idx
end subroutine amg_c_base_onelev_cseti end subroutine amg_c_base_onelev_cseti
end interface end interface
interface interface
subroutine amg_c_base_onelev_csetc(lv,what,val,info,pos,idx) subroutine amg_c_base_onelev_csetc(lv,what,val,info,pos,idx)
import :: psb_cspmat_type, psb_c_vect_type, psb_c_base_vect_type, & import :: psb_cspmat_type, psb_c_vect_type, psb_c_base_vect_type, &
& psb_clinmap_type, psb_spk_, amg_c_onelev_type, & & psb_clinmap_type, psb_spk_, amg_c_onelev_type, &
& psb_ipk_, psb_epk_, psb_desc_type & psb_ipk_, psb_epk_, psb_desc_type
Implicit None Implicit None
! Arguments ! Arguments
class(amg_c_onelev_type), intent(inout) :: lv class(amg_c_onelev_type), intent(inout) :: lv
character(len=*), intent(in) :: what character(len=*), intent(in) :: what
character(len=*), intent(in) :: val character(len=*), intent(in) :: val
integer(psb_ipk_), intent(out) :: info integer(psb_ipk_), intent(out) :: info
character(len=*), optional, intent(in) :: pos character(len=*), optional, intent(in) :: pos
integer(psb_ipk_), intent(in), optional :: idx integer(psb_ipk_), intent(in), optional :: idx
end subroutine amg_c_base_onelev_csetc end subroutine amg_c_base_onelev_csetc
end interface end interface
interface interface
subroutine amg_c_base_onelev_csetr(lv,what,val,info,pos,idx) subroutine amg_c_base_onelev_csetr(lv,what,val,info,pos,idx)
import :: psb_cspmat_type, psb_c_vect_type, psb_c_base_vect_type, & import :: psb_cspmat_type, psb_c_vect_type, psb_c_base_vect_type, &
& psb_clinmap_type, psb_spk_, amg_c_onelev_type, & & psb_clinmap_type, psb_spk_, amg_c_onelev_type, &
& psb_ipk_, psb_epk_, psb_desc_type & psb_ipk_, psb_epk_, psb_desc_type
Implicit None Implicit None
class(amg_c_onelev_type), intent(inout) :: lv class(amg_c_onelev_type), intent(inout) :: lv
character(len=*), intent(in) :: what character(len=*), intent(in) :: what
real(psb_spk_), intent(in) :: val real(psb_spk_), intent(in) :: val
integer(psb_ipk_), intent(out) :: info integer(psb_ipk_), intent(out) :: info
character(len=*), optional, intent(in) :: pos character(len=*), optional, intent(in) :: pos
@ -398,13 +399,13 @@ interface
end subroutine amg_c_base_onelev_csetr end subroutine amg_c_base_onelev_csetr
end interface end interface
interface interface
subroutine amg_c_base_onelev_dump(lv,level,info,prefix,head,ac,rp,smoother,& subroutine amg_c_base_onelev_dump(lv,level,info,prefix,head,ac,rp,smoother,&
& solver,tprol,global_num) & solver,tprol,global_num)
import :: psb_cspmat_type, psb_c_vect_type, psb_c_base_vect_type, & import :: psb_cspmat_type, psb_c_vect_type, psb_c_base_vect_type, &
& psb_clinmap_type, psb_spk_, amg_c_onelev_type, & & psb_clinmap_type, psb_spk_, amg_c_onelev_type, &
& psb_ipk_, psb_epk_, psb_desc_type & psb_ipk_, psb_epk_, psb_desc_type
implicit none implicit none
class(amg_c_onelev_type), intent(in) :: lv class(amg_c_onelev_type), intent(in) :: lv
integer(psb_ipk_), intent(in) :: level integer(psb_ipk_), intent(in) :: level
integer(psb_ipk_), intent(out) :: info integer(psb_ipk_), intent(out) :: info
@ -436,7 +437,7 @@ interface
end subroutine amg_c_base_onelev_map_rstr_v end subroutine amg_c_base_onelev_map_rstr_v
end interface end interface
interface interface
subroutine amg_c_base_onelev_map_prol_a(lv,alpha,v,beta,u,info,work) subroutine amg_c_base_onelev_map_prol_a(lv,alpha,v,beta,u,info,work)
import import
implicit none implicit none
@ -459,15 +460,15 @@ interface
type(psb_c_vect_type), optional, target, intent(inout) :: vtx,vty type(psb_c_vect_type), optional, target, intent(inout) :: vtx,vty
end subroutine amg_c_base_onelev_map_prol_v end subroutine amg_c_base_onelev_map_prol_v
end interface end interface
contains contains
! !
! Function returning the size of the amg_prec_type data structure ! Function returning the size of the amg_prec_type data structure
! in bytes or in number of nonzeros of the operator(s) involved. ! in bytes or in number of nonzeros of the operator(s) involved.
! !
function c_base_onelev_get_nzeros(lv) result(val) function c_base_onelev_get_nzeros(lv) result(val)
implicit none implicit none
class(amg_c_onelev_type), intent(in) :: lv class(amg_c_onelev_type), intent(in) :: lv
integer(psb_epk_) :: val integer(psb_epk_) :: val
integer(psb_ipk_) :: i integer(psb_ipk_) :: i
@ -479,16 +480,16 @@ contains
end function c_base_onelev_get_nzeros end function c_base_onelev_get_nzeros
function c_base_onelev_sizeof(lv) result(val) function c_base_onelev_sizeof(lv) result(val)
implicit none implicit none
class(amg_c_onelev_type), intent(in) :: lv class(amg_c_onelev_type), intent(in) :: lv
integer(psb_epk_) :: val integer(psb_epk_) :: val
integer(psb_ipk_) :: i integer(psb_ipk_) :: i
val = psb_sizeof_ip+psb_sizeof_lp val = psb_sizeof_ip+psb_sizeof_lp
val = val + lv%desc_ac%sizeof() val = val + lv%desc_ac%sizeof()
val = val + lv%ac%sizeof() val = val + lv%ac%sizeof()
val = val + lv%tprol%sizeof() val = val + lv%tprol%sizeof()
val = val + lv%linmap%sizeof() val = val + lv%linmap%sizeof()
if (allocated(lv%sm)) val = val + lv%sm%sizeof() if (allocated(lv%sm)) val = val + lv%sm%sizeof()
if (allocated(lv%sm2a)) val = val + lv%sm2a%sizeof() if (allocated(lv%sm2a)) val = val + lv%sm2a%sizeof()
if (allocated(lv%aggr)) val = val + lv%aggr%sizeof() if (allocated(lv%aggr)) val = val + lv%aggr%sizeof()
@ -497,19 +498,19 @@ contains
subroutine c_base_onelev_nullify(lv) subroutine c_base_onelev_nullify(lv)
implicit none implicit none
class(amg_c_onelev_type), intent(inout) :: lv class(amg_c_onelev_type), intent(inout) :: lv
nullify(lv%base_a) nullify(lv%base_a)
nullify(lv%base_desc) nullify(lv%base_desc)
nullify(lv%sm2) nullify(lv%sm2)
end subroutine c_base_onelev_nullify end subroutine c_base_onelev_nullify
! !
! Multilevel defaults: ! Multilevel defaults:
! multiplicative vs. additive ML framework; ! multiplicative vs. additive ML framework;
! Smoothed decoupled aggregation with zero threshold; ! Smoothed decoupled aggregation with zero threshold;
! distributed coarse matrix; ! distributed coarse matrix;
! damping omega computed with the max-norm estimate of the ! damping omega computed with the max-norm estimate of the
! dominant eigenvalue; ! dominant eigenvalue;
@ -519,10 +520,10 @@ contains
subroutine c_base_onelev_default(lv) subroutine c_base_onelev_default(lv)
Implicit None Implicit None
! Arguments ! Arguments
class(amg_c_onelev_type), target, intent(inout) :: lv class(amg_c_onelev_type), target, intent(inout) :: lv
integer(psb_ipk_) :: info integer(psb_ipk_) :: info
lv%parms%sweeps_pre = 1 lv%parms%sweeps_pre = 1
lv%parms%sweeps_post = 1 lv%parms%sweeps_post = 1
@ -537,7 +538,7 @@ contains
lv%parms%aggr_filter = amg_no_filter_mat_ lv%parms%aggr_filter = amg_no_filter_mat_
lv%parms%aggr_omega_val = szero lv%parms%aggr_omega_val = szero
lv%parms%aggr_thresh = 0.01_psb_spk_ lv%parms%aggr_thresh = 0.01_psb_spk_
if (allocated(lv%sm)) call lv%sm%default() if (allocated(lv%sm)) call lv%sm%default()
if (allocated(lv%sm2a)) then if (allocated(lv%sm2a)) then
call lv%sm2a%default() call lv%sm2a%default()
@ -547,7 +548,7 @@ contains
end if end if
if (.not.allocated(lv%aggr)) allocate(amg_c_dec_aggregator_type :: lv%aggr,stat=info) if (.not.allocated(lv%aggr)) allocate(amg_c_dec_aggregator_type :: lv%aggr,stat=info)
if (allocated(lv%aggr)) call lv%aggr%default() if (allocated(lv%aggr)) call lv%aggr%default()
return return
end subroutine c_base_onelev_default end subroutine c_base_onelev_default
@ -562,9 +563,9 @@ contains
type(psb_lcspmat_type), intent(out) :: t_prol type(psb_lcspmat_type), intent(out) :: t_prol
type(amg_saggr_data), intent(in) :: ag_data type(amg_saggr_data), intent(in) :: ag_data
integer(psb_ipk_), intent(out) :: info integer(psb_ipk_), intent(out) :: info
call lv%aggr%bld_tprol(lv%parms,ag_data,a,desc_a,ilaggr,nlaggr,t_prol,info) call lv%aggr%bld_tprol(lv%parms,ag_data,a,desc_a,ilaggr,nlaggr,t_prol,info)
end subroutine c_base_onelev_bld_tprol end subroutine c_base_onelev_bld_tprol
@ -574,7 +575,7 @@ contains
integer(psb_ipk_), intent(out) :: info integer(psb_ipk_), intent(out) :: info
call lv%aggr%update_next(lvnext%aggr,info) call lv%aggr%update_next(lvnext%aggr,info)
end subroutine c_base_onelev_update_aggr end subroutine c_base_onelev_update_aggr
@ -583,33 +584,33 @@ contains
Implicit None Implicit None
! Arguments ! Arguments
class(amg_c_onelev_type), target, intent(inout) :: lv class(amg_c_onelev_type), target, intent(inout) :: lv
class(amg_c_onelev_type), target, intent(inout) :: lvout class(amg_c_onelev_type), target, intent(inout) :: lvout
integer(psb_ipk_), intent(out) :: info integer(psb_ipk_), intent(out) :: info
info = psb_success_ info = psb_success_
if (allocated(lv%sm)) then if (allocated(lv%sm)) then
call lv%sm%clone(lvout%sm,info) call lv%sm%clone(lvout%sm,info)
else else
if (allocated(lvout%sm)) then if (allocated(lvout%sm)) then
call lvout%sm%free(info) call lvout%sm%free(info)
if (info==psb_success_) deallocate(lvout%sm,stat=info) if (info==psb_success_) deallocate(lvout%sm,stat=info)
end if end if
end if end if
if (allocated(lv%sm2a)) then if (allocated(lv%sm2a)) then
call lv%sm%clone(lvout%sm2a,info) call lv%sm%clone(lvout%sm2a,info)
lvout%sm2 => lvout%sm2a lvout%sm2 => lvout%sm2a
else else
if (allocated(lvout%sm2a)) then if (allocated(lvout%sm2a)) then
call lvout%sm2a%free(info) call lvout%sm2a%free(info)
if (info==psb_success_) deallocate(lvout%sm2a,stat=info) if (info==psb_success_) deallocate(lvout%sm2a,stat=info)
end if end if
lvout%sm2 => lvout%sm lvout%sm2 => lvout%sm
end if end if
if (allocated(lv%aggr)) then if (allocated(lv%aggr)) then
call lv%aggr%clone(lvout%aggr,info) call lv%aggr%clone(lvout%aggr,info)
else else
if (allocated(lvout%aggr)) then if (allocated(lvout%aggr)) then
call lvout%aggr%free(info) call lvout%aggr%free(info)
if (info==psb_success_) deallocate(lvout%aggr,stat=info) if (info==psb_success_) deallocate(lvout%aggr,stat=info)
end if end if
@ -622,7 +623,7 @@ contains
if (info == psb_success_) call lv%remap_data%clone(lvout%remap_data,info) if (info == psb_success_) call lv%remap_data%clone(lvout%remap_data,info)
lvout%base_a => lv%base_a lvout%base_a => lv%base_a
lvout%base_desc => lv%base_desc lvout%base_desc => lv%base_desc
return return
end subroutine c_base_onelev_clone end subroutine c_base_onelev_clone
@ -631,12 +632,12 @@ contains
use psb_base_mod use psb_base_mod
implicit none implicit none
class(amg_c_onelev_type), target, intent(inout) :: lv, b class(amg_c_onelev_type), target, intent(inout) :: lv, b
integer(psb_ipk_), intent(out) :: info integer(psb_ipk_), intent(out) :: info
call b%free(info) call b%free(info)
b%parms = lv%parms b%parms = lv%parms
b%szratio = lv%szratio b%szratio = lv%szratio
if (associated(lv%sm2,lv%sm2a)) then if (associated(lv%sm2,lv%sm2a)) then
call move_alloc(lv%sm,b%sm) call move_alloc(lv%sm,b%sm)
call move_alloc(lv%sm2a,b%sm2a) call move_alloc(lv%sm2a,b%sm2a)
b%sm2 =>b%sm2a b%sm2 =>b%sm2a
@ -647,18 +648,18 @@ contains
end if end if
call move_alloc(lv%aggr,b%aggr) call move_alloc(lv%aggr,b%aggr)
if (info == psb_success_) call psb_move_alloc(lv%ac,b%ac,info) if (info == psb_success_) call psb_move_alloc(lv%ac,b%ac,info)
if (info == psb_success_) call psb_move_alloc(lv%tprol,b%tprol,info) if (info == psb_success_) call psb_move_alloc(lv%tprol,b%tprol,info)
if (info == psb_success_) call psb_move_alloc(lv%desc_ac,b%desc_ac,info) if (info == psb_success_) call psb_move_alloc(lv%desc_ac,b%desc_ac,info)
if (info == psb_success_) call psb_move_alloc(lv%linmap,b%linmap,info) if (info == psb_success_) call psb_move_alloc(lv%linmap,b%linmap,info)
b%base_a => lv%base_a b%base_a => lv%base_a
b%base_desc => lv%base_desc b%base_desc => lv%base_desc
end subroutine c_base_onelev_move_alloc end subroutine c_base_onelev_move_alloc
function c_base_onelev_get_wrksize(lv) result(val) function c_base_onelev_get_wrksize(lv) result(val)
implicit none implicit none
class(amg_c_onelev_type), intent(inout) :: lv class(amg_c_onelev_type), intent(inout) :: lv
integer(psb_ipk_) :: val integer(psb_ipk_) :: val
@ -679,26 +680,26 @@ contains
select case(lv%parms%ml_cycle) select case(lv%parms%ml_cycle)
case(amg_add_ml_,amg_mult_ml_,amg_vcycle_ml_, amg_wcycle_ml_) case(amg_add_ml_,amg_mult_ml_,amg_vcycle_ml_, amg_wcycle_ml_)
! We're good ! We're good
case(amg_kcycle_ml_, amg_kcyclesym_ml_) case(amg_kcycle_ml_, amg_kcyclesym_ml_)
! !
! We need 7 in inneritkcycle. ! We need 7 in inneritkcycle.
! Can we reuse vtx? ! Can we reuse vtx?
! !
val = val + 7 val = val + 7
case default case default
! Need a better error signaling ? ! Need a better error signaling ?
val = -1 val = -1
end select end select
end function c_base_onelev_get_wrksize end function c_base_onelev_get_wrksize
subroutine c_base_onelev_allocate_wrk(lv,info,vmold) subroutine c_base_onelev_allocate_wrk(lv,info,vmold)
use psb_base_mod use psb_base_mod
implicit none implicit none
class(amg_c_onelev_type), target, intent(inout) :: lv class(amg_c_onelev_type), target, intent(inout) :: lv
integer(psb_ipk_), intent(out) :: info integer(psb_ipk_), intent(out) :: info
class(psb_c_base_vect_type), intent(in), optional :: vmold class(psb_c_base_vect_type), intent(in), optional :: vmold
! !
integer(psb_ipk_) :: nwv, i integer(psb_ipk_) :: nwv, i
@ -711,22 +712,22 @@ contains
! Need to fix this, we need two different allocations ! Need to fix this, we need two different allocations
! !
call lv%wrk%alloc(nwv,lv%base_desc,info,vmold=vmold,& call lv%wrk%alloc(nwv,lv%base_desc,info,vmold=vmold,&
& desc2=lv%remap_data%desc_ac_pre_remap) & desc2=lv%remap_data%desc_ac_pre_remap)
else else
call lv%wrk%alloc(nwv,lv%base_desc,info,vmold=vmold) call lv%wrk%alloc(nwv,lv%base_desc,info,vmold=vmold)
end if end if
end if end if
end subroutine c_base_onelev_allocate_wrk end subroutine c_base_onelev_allocate_wrk
subroutine c_base_onelev_free_wrk(lv,info) subroutine c_base_onelev_free_wrk(lv,info)
use psb_base_mod use psb_base_mod
implicit none implicit none
class(amg_c_onelev_type), target, intent(inout) :: lv class(amg_c_onelev_type), target, intent(inout) :: lv
integer(psb_ipk_), intent(out) :: info integer(psb_ipk_), intent(out) :: info
! !
integer(psb_ipk_) :: nwv,i integer(psb_ipk_) :: nwv,i
info = psb_success_ info = psb_success_
if (allocated(lv%wrk)) then if (allocated(lv%wrk)) then
@ -734,17 +735,17 @@ contains
if (info == 0) deallocate(lv%wrk,stat=info) if (info == 0) deallocate(lv%wrk,stat=info)
end if end if
end subroutine c_base_onelev_free_wrk end subroutine c_base_onelev_free_wrk
subroutine c_wrk_alloc(wk,nwv,desc,info,vmold, desc2) subroutine c_wrk_alloc(wk,nwv,desc,info,vmold, desc2)
use psb_base_mod use psb_base_mod
Implicit None Implicit None
! Arguments ! Arguments
class(amg_cmlprec_wrk_type), target, intent(inout) :: wk class(amg_cmlprec_wrk_type), target, intent(inout) :: wk
integer(psb_ipk_), intent(in) :: nwv integer(psb_ipk_), intent(in) :: nwv
type(psb_desc_type), intent(in) :: desc type(psb_desc_type), intent(in) :: desc
integer(psb_ipk_), intent(out) :: info integer(psb_ipk_), intent(out) :: info
class(psb_c_base_vect_type), intent(in), optional :: vmold class(psb_c_base_vect_type), intent(in), optional :: vmold
type(psb_desc_type), intent(in), optional :: desc2 type(psb_desc_type), intent(in), optional :: desc2
! !
@ -808,14 +809,14 @@ contains
end do end do
end if end if
end subroutine c_wrk_alloc end subroutine c_wrk_alloc
subroutine c_wrk_free(wk,info) subroutine c_wrk_free(wk,info)
Implicit None Implicit None
! Arguments ! Arguments
class(amg_cmlprec_wrk_type), target, intent(inout) :: wk class(amg_cmlprec_wrk_type), target, intent(inout) :: wk
integer(psb_ipk_), intent(out) :: info integer(psb_ipk_), intent(out) :: info
! !
integer(psb_ipk_) :: i integer(psb_ipk_) :: i
info = psb_success_ info = psb_success_
@ -836,7 +837,7 @@ contains
end if end if
end subroutine c_wrk_free end subroutine c_wrk_free
subroutine c_wrk_clone(wk,wkout,info) subroutine c_wrk_clone(wk,wkout,info)
use psb_base_mod use psb_base_mod
Implicit None Implicit None
@ -844,11 +845,11 @@ contains
! Arguments ! Arguments
class(amg_cmlprec_wrk_type), target, intent(inout) :: wk class(amg_cmlprec_wrk_type), target, intent(inout) :: wk
class(amg_cmlprec_wrk_type), target, intent(inout) :: wkout class(amg_cmlprec_wrk_type), target, intent(inout) :: wkout
integer(psb_ipk_), intent(out) :: info integer(psb_ipk_), intent(out) :: info
! !
integer(psb_ipk_) :: i integer(psb_ipk_) :: i
info = psb_success_ info = psb_success_
call psb_safe_ab_cpy(wk%tx,wkout%tx,info) call psb_safe_ab_cpy(wk%tx,wkout%tx,info)
call psb_safe_ab_cpy(wk%ty,wkout%ty,info) call psb_safe_ab_cpy(wk%ty,wkout%ty,info)
call psb_safe_ab_cpy(wk%x2l,wkout%x2l,info) call psb_safe_ab_cpy(wk%x2l,wkout%x2l,info)
@ -870,12 +871,12 @@ contains
return return
end subroutine c_wrk_clone end subroutine c_wrk_clone
subroutine c_wrk_move_alloc(wk, b,info) subroutine c_wrk_move_alloc(wk, b,info)
implicit none implicit none
class(amg_cmlprec_wrk_type), target, intent(inout) :: wk, b class(amg_cmlprec_wrk_type), target, intent(inout) :: wk, b
integer(psb_ipk_), intent(out) :: info integer(psb_ipk_), intent(out) :: info
call b%free(info) call b%free(info)
call move_alloc(wk%tx,b%tx) call move_alloc(wk%tx,b%tx)
call move_alloc(wk%ty,b%ty) call move_alloc(wk%ty,b%ty)
@ -888,17 +889,17 @@ contains
call move_alloc(wk%vx2l%v,b%vx2l%v) call move_alloc(wk%vx2l%v,b%vx2l%v)
call move_alloc(wk%vy2l%v,b%vy2l%v) call move_alloc(wk%vy2l%v,b%vy2l%v)
call move_alloc(wk%wv,b%wv) call move_alloc(wk%wv,b%wv)
end subroutine c_wrk_move_alloc end subroutine c_wrk_move_alloc
subroutine c_wrk_cnv(wk,info,vmold) subroutine c_wrk_cnv(wk,info,vmold)
use psb_base_mod use psb_base_mod
Implicit None Implicit None
! Arguments ! Arguments
class(amg_cmlprec_wrk_type), target, intent(inout) :: wk class(amg_cmlprec_wrk_type), target, intent(inout) :: wk
integer(psb_ipk_), intent(out) :: info integer(psb_ipk_), intent(out) :: info
class(psb_c_base_vect_type), intent(in), optional :: vmold class(psb_c_base_vect_type), intent(in), optional :: vmold
! !
integer(psb_ipk_) :: i integer(psb_ipk_) :: i
@ -919,7 +920,7 @@ contains
function c_wrk_sizeof(wk) result(val) function c_wrk_sizeof(wk) result(val)
use psb_realloc_mod use psb_realloc_mod
implicit none implicit none
class(amg_cmlprec_wrk_type), intent(in) :: wk class(amg_cmlprec_wrk_type), intent(in) :: wk
integer(psb_epk_) :: val integer(psb_epk_) :: val
integer :: i integer :: i
@ -938,14 +939,14 @@ contains
end do end do
end if end if
end function c_wrk_sizeof end function c_wrk_sizeof
subroutine c_remap_data_clone(rmp, remap_out, info) subroutine c_remap_data_clone(rmp, remap_out, info)
use psb_base_mod use psb_base_mod
implicit none implicit none
! Arguments ! Arguments
class(amg_c_remap_data_type), target, intent(inout) :: rmp class(amg_c_remap_data_type), target, intent(inout) :: rmp
class(amg_c_remap_data_type), target, intent(inout) :: remap_out class(amg_c_remap_data_type), target, intent(inout) :: remap_out
integer(psb_ipk_), intent(out) :: info integer(psb_ipk_), intent(out) :: info
! !
integer(psb_ipk_) :: i integer(psb_ipk_) :: i
@ -956,7 +957,7 @@ contains
& call rmp%desc_ac_pre_remap%clone(remap_out%desc_ac_pre_remap,info) & call rmp%desc_ac_pre_remap%clone(remap_out%desc_ac_pre_remap,info)
remap_out%idest = rmp%idest remap_out%idest = rmp%idest
call psb_safe_ab_cpy(rmp%isrc,remap_out%isrc,info) call psb_safe_ab_cpy(rmp%isrc,remap_out%isrc,info)
call psb_safe_ab_cpy(rmp%nrsrc,remap_out%nrsrc,info) call psb_safe_ab_cpy(rmp%nrsrc,remap_out%nrsrc,info)
end subroutine c_remap_data_clone end subroutine c_remap_data_clone
end module amg_c_onelev_mod end module amg_c_onelev_mod

296
amgprec/amg_d_onelev_mod.f90
Unescape Escape View File

@ -1,15 +1,15 @@
! !
! !
! AMG4PSBLAS version 1.0 ! AMG4PSBLAS version 1.0
! Algebraic Multigrid Package ! Algebraic Multigrid Package
! based on PSBLAS (Parallel Sparse BLAS version 3.7) ! based on PSBLAS (Parallel Sparse BLAS version 3.7)
! !
! (C) Copyright 2021 ! (C) Copyright 2021
! !
! Salvatore Filippone ! Salvatore Filippone
! Pasqua D'Ambra ! Pasqua D'Ambra
! Fabio Durastante ! Fabio Durastante
! !
! Redistribution and use in source and binary forms, with or without ! Redistribution and use in source and binary forms, with or without
! modification, are permitted provided that the following conditions ! modification, are permitted provided that the following conditions
! are met: ! are met:
@ -21,7 +21,7 @@
! 3. The name of the AMG4PSBLAS group or the names of its contributors may ! 3. The name of the AMG4PSBLAS group or the names of its contributors may
! not be used to endorse or promote products derived from this ! not be used to endorse or promote products derived from this
! software without specific written permission. ! software without specific written permission.
! !
! THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS ! THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
! ``AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED ! ``AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED
! TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR ! TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
@ -33,22 +33,22 @@
! CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ! 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 ! ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
! POSSIBILITY OF SUCH DAMAGE. ! POSSIBILITY OF SUCH DAMAGE.
! !
! !
! File: amg_d_onelev_mod.f90 ! File: amg_d_onelev_mod.f90
! !
! Module: amg_d_onelev_mod ! Module: amg_d_onelev_mod
! !
! This module defines: ! This module defines:
! - the amg_d_onelev_type data structure containing one level ! - the amg_d_onelev_type data structure containing one level
! of a multilevel preconditioner and related ! of a multilevel preconditioner and related
! data structures; ! data structures;
! !
! It contains routines for ! It contains routines for
! - Building and applying; ! - Building and applying;
! - checking if the preconditioner is correctly defined; ! - checking if the preconditioner is correctly defined;
! - printing a description of the preconditioner; ! - printing a description of the preconditioner;
! - deallocating the preconditioner data structure. ! - deallocating the preconditioner data structure.
! !
module amg_d_onelev_mod module amg_d_onelev_mod
@ -56,6 +56,8 @@ module amg_d_onelev_mod
use amg_base_prec_type use amg_base_prec_type
use amg_d_base_smoother_mod use amg_d_base_smoother_mod
use amg_d_dec_aggregator_mod use amg_d_dec_aggregator_mod
use amg_d_parmatch_aggregator_mod
use psb_base_mod, only : psb_dspmat_type, psb_d_vect_type, & use psb_base_mod, only : psb_dspmat_type, psb_d_vect_type, &
& psb_d_base_vect_type, psb_ldspmat_type, psb_dlinmap_type, psb_dpk_, & & psb_d_base_vect_type, psb_ldspmat_type, psb_dlinmap_type, psb_dpk_, &
& psb_ipk_, psb_epk_, psb_lpk_, psb_desc_type, psb_i_base_vect_type, & & psb_ipk_, psb_epk_, psb_lpk_, psb_desc_type, psb_i_base_vect_type, &
@ -73,11 +75,11 @@ module amg_d_onelev_mod
! class(amg_d_base_smoother_type), pointer :: sm2 => null() ! class(amg_d_base_smoother_type), pointer :: sm2 => null()
! class(amg_dmlprec_wrk_type), allocatable :: wrk ! class(amg_dmlprec_wrk_type), allocatable :: wrk
! class(amg_d_base_aggregator_type), allocatable :: aggr ! class(amg_d_base_aggregator_type), allocatable :: aggr
! type(amg_dml_parms) :: parms ! type(amg_dml_parms) :: parms
! type(psb_dspmat_type) :: ac ! type(psb_dspmat_type) :: ac
! type(psb_desc_type) :: desc_ac ! type(psb_desc_type) :: desc_ac
! type(psb_dspmat_type), pointer :: base_a => null() ! type(psb_dspmat_type), pointer :: base_a => null()
! type(psb_desc_type), pointer :: base_desc => null() ! type(psb_desc_type), pointer :: base_desc => null()
! type(psb_dlinmap_type) :: map ! type(psb_dlinmap_type) :: map
! end type amg_donelev_type ! end type amg_donelev_type
! !
@ -93,7 +95,7 @@ module amg_d_onelev_mod
! Workspace for application of preconditioner; may be ! Workspace for application of preconditioner; may be
! pre-allocated to save time in the application within a ! pre-allocated to save time in the application within a
! Krylov solver. ! Krylov solver.
! aggr - class(amg_d_base_aggregator_type), allocatable ! aggr - class(amg_d_base_aggregator_type), allocatable
! The aggregator object: holds the algorithmic choices and ! The aggregator object: holds the algorithmic choices and
! (possibly) additional data for building the aggregation. ! (possibly) additional data for building the aggregation.
! parms - type(amg_dml_parms) ! parms - type(amg_dml_parms)
@ -104,7 +106,7 @@ module amg_d_onelev_mod
! The communication descriptor associated to the matrix ! The communication descriptor associated to the matrix
! stored in ac. ! stored in ac.
! base_a - type(psb_dspmat_type), pointer. ! base_a - type(psb_dspmat_type), pointer.
! Pointer (really a pointer!) to the local part of the current ! Pointer (really a pointer!) to the local part of the current
! matrix (so we have a unified treatment of residuals). ! matrix (so we have a unified treatment of residuals).
! We need this to avoid passing explicitly the current matrix ! We need this to avoid passing explicitly the current matrix
! to the routine which applies the preconditioner. ! to the routine which applies the preconditioner.
@ -115,13 +117,13 @@ module amg_d_onelev_mod
! vector spaces associated to the index spaces of the previous ! vector spaces associated to the index spaces of the previous
! and current levels. ! and current levels.
! !
! Methods: ! Methods:
! Most methods follow the encapsulation hierarchy: they take whatever action ! Most methods follow the encapsulation hierarchy: they take whatever action
! is appropriate for the current object, then call the corresponding method for ! is appropriate for the current object, then call the corresponding method for
! the contained object. ! the contained object.
! As an example: the descr() method prints out a description of the ! As an example: the descr() method prints out a description of the
! level. It starts by invoking the descr() method of the parms object, ! level. It starts by invoking the descr() method of the parms object,
! then calls the descr() method of the smoother object. ! then calls the descr() method of the smoother object.
! !
! descr - Prints a description of the object. ! descr - Prints a description of the object.
! default - Set default values ! default - Set default values
@ -130,14 +132,14 @@ module amg_d_onelev_mod
! it is passed to the smoother object for further processing. ! it is passed to the smoother object for further processing.
! check - Sanity checks. ! check - Sanity checks.
! sizeof - Total memory occupation in bytes ! sizeof - Total memory occupation in bytes
! get_nzeros - Number of nonzeros ! get_nzeros - Number of nonzeros
! get_wrksz - How many workspace vector does apply_vect need ! get_wrksz - How many workspace vector does apply_vect need
! allocate_wrk - Allocate auxiliary workspace ! allocate_wrk - Allocate auxiliary workspace
! free_wrk - Free auxiliary workspace ! free_wrk - Free auxiliary workspace
! bld_tprol - Invoke the aggr method to build the tentative prolongator ! bld_tprol - Invoke the aggr method to build the tentative prolongator
! mat_asb - Build the final (possibly smoothed) prolongator and coarse matrix. ! mat_asb - Build the final (possibly smoothed) prolongator and coarse matrix.
!
! !
!
type amg_dmlprec_wrk_type type amg_dmlprec_wrk_type
real(psb_dpk_), allocatable :: tx(:), ty(:), x2l(:), y2l(:) real(psb_dpk_), allocatable :: tx(:), ty(:), x2l(:), y2l(:)
type(psb_d_vect_type) :: vtx, vty, vx2l, vy2l type(psb_d_vect_type) :: vtx, vty, vx2l, vy2l
@ -148,10 +150,10 @@ module amg_d_onelev_mod
procedure, pass(wk) :: clone => d_wrk_clone procedure, pass(wk) :: clone => d_wrk_clone
procedure, pass(wk) :: move_alloc => d_wrk_move_alloc procedure, pass(wk) :: move_alloc => d_wrk_move_alloc
procedure, pass(wk) :: cnv => d_wrk_cnv procedure, pass(wk) :: cnv => d_wrk_cnv
procedure, pass(wk) :: sizeof => d_wrk_sizeof procedure, pass(wk) :: sizeof => d_wrk_sizeof
end type amg_dmlprec_wrk_type end type amg_dmlprec_wrk_type
private :: d_wrk_alloc, d_wrk_free, & private :: d_wrk_alloc, d_wrk_free, &
& d_wrk_clone, d_wrk_move_alloc, d_wrk_cnv, d_wrk_sizeof & d_wrk_clone, d_wrk_move_alloc, d_wrk_cnv, d_wrk_sizeof
type amg_d_remap_data_type type amg_d_remap_data_type
type(psb_dspmat_type) :: ac_pre_remap type(psb_dspmat_type) :: ac_pre_remap
@ -161,19 +163,19 @@ module amg_d_onelev_mod
contains contains
procedure, pass(rmp) :: clone => d_remap_data_clone procedure, pass(rmp) :: clone => d_remap_data_clone
end type amg_d_remap_data_type end type amg_d_remap_data_type
type amg_d_onelev_type type amg_d_onelev_type
class(amg_d_base_smoother_type), allocatable :: sm, sm2a class(amg_d_base_smoother_type), allocatable :: sm, sm2a
class(amg_d_base_smoother_type), pointer :: sm2 => null() class(amg_d_base_smoother_type), pointer :: sm2 => null()
class(amg_dmlprec_wrk_type), allocatable :: wrk class(amg_dmlprec_wrk_type), allocatable :: wrk
class(amg_d_base_aggregator_type), allocatable :: aggr class(amg_d_base_aggregator_type), allocatable :: aggr
type(amg_dml_parms) :: parms type(amg_dml_parms) :: parms
type(psb_dspmat_type) :: ac type(psb_dspmat_type) :: ac
integer(psb_ipk_) :: ac_nz_loc integer(psb_ipk_) :: ac_nz_loc
integer(psb_lpk_) :: ac_nz_tot integer(psb_lpk_) :: ac_nz_tot
type(psb_desc_type) :: desc_ac type(psb_desc_type) :: desc_ac
type(psb_dspmat_type), pointer :: base_a => null() type(psb_dspmat_type), pointer :: base_a => null()
type(psb_desc_type), pointer :: base_desc => null() type(psb_desc_type), pointer :: base_desc => null()
type(psb_ldspmat_type) :: tprol type(psb_ldspmat_type) :: tprol
type(psb_dlinmap_type) :: linmap type(psb_dlinmap_type) :: linmap
type(amg_d_remap_data_type) :: remap_data type(amg_d_remap_data_type) :: remap_data
@ -197,7 +199,7 @@ module amg_d_onelev_mod
procedure, pass(lv) :: setsm => amg_d_base_onelev_setsm procedure, pass(lv) :: setsm => amg_d_base_onelev_setsm
procedure, pass(lv) :: setsv => amg_d_base_onelev_setsv procedure, pass(lv) :: setsv => amg_d_base_onelev_setsv
procedure, pass(lv) :: setag => amg_d_base_onelev_setag procedure, pass(lv) :: setag => amg_d_base_onelev_setag
generic, public :: set => cseti, csetr, csetc, setsm, setsv, setag generic, public :: set => cseti, csetr, csetc, setsm, setsv, setag
procedure, pass(lv) :: sizeof => d_base_onelev_sizeof procedure, pass(lv) :: sizeof => d_base_onelev_sizeof
procedure, pass(lv) :: get_nzeros => d_base_onelev_get_nzeros procedure, pass(lv) :: get_nzeros => d_base_onelev_get_nzeros
procedure, pass(lv) :: get_wrksz => d_base_onelev_get_wrksize procedure, pass(lv) :: get_wrksz => d_base_onelev_get_wrksize
@ -205,8 +207,8 @@ module amg_d_onelev_mod
procedure, pass(lv) :: free_wrk => d_base_onelev_free_wrk procedure, pass(lv) :: free_wrk => d_base_onelev_free_wrk
procedure, nopass :: stringval => amg_stringval procedure, nopass :: stringval => amg_stringval
procedure, pass(lv) :: move_alloc => d_base_onelev_move_alloc procedure, pass(lv) :: move_alloc => d_base_onelev_move_alloc
procedure, pass(lv) :: map_rstr_a => amg_d_base_onelev_map_rstr_a procedure, pass(lv) :: map_rstr_a => amg_d_base_onelev_map_rstr_a
procedure, pass(lv) :: map_prol_a => amg_d_base_onelev_map_prol_a procedure, pass(lv) :: map_prol_a => amg_d_base_onelev_map_prol_a
procedure, pass(lv) :: map_rstr_v => amg_d_base_onelev_map_rstr_v procedure, pass(lv) :: map_rstr_v => amg_d_base_onelev_map_rstr_v
@ -226,11 +228,11 @@ module amg_d_onelev_mod
& d_base_onelev_get_wrksize, d_base_onelev_allocate_wrk, & & d_base_onelev_get_wrksize, d_base_onelev_allocate_wrk, &
& d_base_onelev_free_wrk & d_base_onelev_free_wrk
interface interface
subroutine amg_d_base_onelev_mat_asb(lv,a,desc_a,ilaggr,nlaggr,t_prol,info) subroutine amg_d_base_onelev_mat_asb(lv,a,desc_a,ilaggr,nlaggr,t_prol,info)
import :: psb_dspmat_type, psb_desc_type, psb_dpk_, psb_ipk_, psb_ldspmat_type, psb_lpk_ import :: psb_dspmat_type, psb_desc_type, psb_dpk_, psb_ipk_, psb_ldspmat_type, psb_lpk_
import :: amg_d_onelev_type import :: amg_d_onelev_type
implicit none implicit none
class(amg_d_onelev_type), intent(inout), target :: lv class(amg_d_onelev_type), intent(inout), target :: lv
type(psb_dspmat_type), intent(in) :: a type(psb_dspmat_type), intent(in) :: a
type(psb_desc_type), intent(inout) :: desc_a type(psb_desc_type), intent(inout) :: desc_a
@ -255,7 +257,7 @@ module amg_d_onelev_mod
end subroutine amg_d_base_onelev_build end subroutine amg_d_base_onelev_build
end interface end interface
interface interface
subroutine amg_d_base_onelev_descr(lv,il,nl,ilmin,info,iout, verbosity) subroutine amg_d_base_onelev_descr(lv,il,nl,ilmin,info,iout, verbosity)
import :: psb_dspmat_type, psb_d_vect_type, psb_d_base_vect_type, & import :: psb_dspmat_type, psb_d_vect_type, psb_d_base_vect_type, &
& psb_dlinmap_type, psb_dpk_, amg_d_onelev_type, & & psb_dlinmap_type, psb_dpk_, amg_d_onelev_type, &
@ -270,127 +272,127 @@ module amg_d_onelev_mod
end subroutine amg_d_base_onelev_descr end subroutine amg_d_base_onelev_descr
end interface end interface
interface interface
subroutine amg_d_base_onelev_cnv(lv,info,amold,vmold,imold) subroutine amg_d_base_onelev_cnv(lv,info,amold,vmold,imold)
import :: amg_d_onelev_type, psb_d_base_vect_type, psb_dpk_, & import :: amg_d_onelev_type, psb_d_base_vect_type, psb_dpk_, &
& psb_d_base_sparse_mat, psb_ipk_, psb_i_base_vect_type & psb_d_base_sparse_mat, psb_ipk_, psb_i_base_vect_type
! Arguments ! Arguments
class(amg_d_onelev_type), intent(inout) :: lv class(amg_d_onelev_type), intent(inout) :: lv
integer(psb_ipk_), intent(out) :: info integer(psb_ipk_), intent(out) :: info
class(psb_d_base_sparse_mat), intent(in), optional :: amold class(psb_d_base_sparse_mat), intent(in), optional :: amold
class(psb_d_base_vect_type), intent(in), optional :: vmold class(psb_d_base_vect_type), intent(in), optional :: vmold
class(psb_i_base_vect_type), intent(in), optional :: imold class(psb_i_base_vect_type), intent(in), optional :: imold
end subroutine amg_d_base_onelev_cnv end subroutine amg_d_base_onelev_cnv
end interface end interface
interface interface
subroutine amg_d_base_onelev_free(lv,info) subroutine amg_d_base_onelev_free(lv,info)
import :: psb_dspmat_type, psb_d_vect_type, psb_d_base_vect_type, & import :: psb_dspmat_type, psb_d_vect_type, psb_d_base_vect_type, &
& psb_dlinmap_type, psb_dpk_, amg_d_onelev_type, & & psb_dlinmap_type, psb_dpk_, amg_d_onelev_type, &
& psb_ipk_, psb_epk_, psb_desc_type & psb_ipk_, psb_epk_, psb_desc_type
implicit none implicit none
class(amg_d_onelev_type), intent(inout) :: lv class(amg_d_onelev_type), intent(inout) :: lv
integer(psb_ipk_), intent(out) :: info integer(psb_ipk_), intent(out) :: info
end subroutine amg_d_base_onelev_free end subroutine amg_d_base_onelev_free
end interface end interface
interface interface
subroutine amg_d_base_onelev_check(lv,info) subroutine amg_d_base_onelev_check(lv,info)
import :: psb_dspmat_type, psb_d_vect_type, psb_d_base_vect_type, & import :: psb_dspmat_type, psb_d_vect_type, psb_d_base_vect_type, &
& psb_dlinmap_type, psb_dpk_, amg_d_onelev_type, & & psb_dlinmap_type, psb_dpk_, amg_d_onelev_type, &
& psb_ipk_, psb_epk_, psb_desc_type & psb_ipk_, psb_epk_, psb_desc_type
Implicit None Implicit None
! Arguments ! Arguments
class(amg_d_onelev_type), intent(inout) :: lv class(amg_d_onelev_type), intent(inout) :: lv
integer(psb_ipk_), intent(out) :: info integer(psb_ipk_), intent(out) :: info
end subroutine amg_d_base_onelev_check end subroutine amg_d_base_onelev_check
end interface end interface
interface interface
subroutine amg_d_base_onelev_setsm(lv,val,info,pos) subroutine amg_d_base_onelev_setsm(lv,val,info,pos)
import :: psb_dpk_, amg_d_onelev_type, amg_d_base_smoother_type, & import :: psb_dpk_, amg_d_onelev_type, amg_d_base_smoother_type, &
& psb_ipk_, psb_epk_, psb_desc_type & psb_ipk_, psb_epk_, psb_desc_type
Implicit None Implicit None
! Arguments ! Arguments
class(amg_d_onelev_type), target, intent(inout) :: lv class(amg_d_onelev_type), target, intent(inout) :: lv
class(amg_d_base_smoother_type), intent(in) :: val class(amg_d_base_smoother_type), intent(in) :: val
integer(psb_ipk_), intent(out) :: info integer(psb_ipk_), intent(out) :: info
character(len=*), optional, intent(in) :: pos character(len=*), optional, intent(in) :: pos
end subroutine amg_d_base_onelev_setsm end subroutine amg_d_base_onelev_setsm
end interface end interface
interface interface
subroutine amg_d_base_onelev_setsv(lv,val,info,pos) subroutine amg_d_base_onelev_setsv(lv,val,info,pos)
import :: psb_dpk_, amg_d_onelev_type, amg_d_base_solver_type, & import :: psb_dpk_, amg_d_onelev_type, amg_d_base_solver_type, &
& psb_ipk_, psb_epk_, psb_desc_type & psb_ipk_, psb_epk_, psb_desc_type
Implicit None Implicit None
! Arguments ! Arguments
class(amg_d_onelev_type), target, intent(inout) :: lv class(amg_d_onelev_type), target, intent(inout) :: lv
class(amg_d_base_solver_type), intent(in) :: val class(amg_d_base_solver_type), intent(in) :: val
integer(psb_ipk_), intent(out) :: info integer(psb_ipk_), intent(out) :: info
character(len=*), optional, intent(in) :: pos character(len=*), optional, intent(in) :: pos
end subroutine amg_d_base_onelev_setsv end subroutine amg_d_base_onelev_setsv
end interface end interface
interface interface
subroutine amg_d_base_onelev_setag(lv,val,info,pos) subroutine amg_d_base_onelev_setag(lv,val,info,pos)
import :: psb_dpk_, amg_d_onelev_type, amg_d_base_aggregator_type, & import :: psb_dpk_, amg_d_onelev_type, amg_d_base_aggregator_type, &
& psb_ipk_, psb_epk_, psb_desc_type & psb_ipk_, psb_epk_, psb_desc_type
Implicit None Implicit None
! Arguments ! Arguments
class(amg_d_onelev_type), target, intent(inout) :: lv class(amg_d_onelev_type), target, intent(inout) :: lv
class(amg_d_base_aggregator_type), intent(in) :: val class(amg_d_base_aggregator_type), intent(in) :: val
integer(psb_ipk_), intent(out) :: info integer(psb_ipk_), intent(out) :: info
character(len=*), optional, intent(in) :: pos character(len=*), optional, intent(in) :: pos
end subroutine amg_d_base_onelev_setag end subroutine amg_d_base_onelev_setag
end interface end interface
interface interface
subroutine amg_d_base_onelev_cseti(lv,what,val,info,pos,idx) subroutine amg_d_base_onelev_cseti(lv,what,val,info,pos,idx)
import :: psb_dspmat_type, psb_d_vect_type, psb_d_base_vect_type, & import :: psb_dspmat_type, psb_d_vect_type, psb_d_base_vect_type, &
& psb_dlinmap_type, psb_dpk_, amg_d_onelev_type, & & psb_dlinmap_type, psb_dpk_, amg_d_onelev_type, &
& psb_ipk_, psb_epk_, psb_desc_type & psb_ipk_, psb_epk_, psb_desc_type
Implicit None Implicit None
! Arguments ! Arguments
class(amg_d_onelev_type), intent(inout) :: lv class(amg_d_onelev_type), intent(inout) :: lv
character(len=*), intent(in) :: what character(len=*), intent(in) :: what
integer(psb_ipk_), intent(in) :: val integer(psb_ipk_), intent(in) :: val
integer(psb_ipk_), intent(out) :: info integer(psb_ipk_), intent(out) :: info
character(len=*), optional, intent(in) :: pos character(len=*), optional, intent(in) :: pos
integer(psb_ipk_), intent(in), optional :: idx integer(psb_ipk_), intent(in), optional :: idx
end subroutine amg_d_base_onelev_cseti end subroutine amg_d_base_onelev_cseti
end interface end interface
interface interface
subroutine amg_d_base_onelev_csetc(lv,what,val,info,pos,idx) subroutine amg_d_base_onelev_csetc(lv,what,val,info,pos,idx)
import :: psb_dspmat_type, psb_d_vect_type, psb_d_base_vect_type, & import :: psb_dspmat_type, psb_d_vect_type, psb_d_base_vect_type, &
& psb_dlinmap_type, psb_dpk_, amg_d_onelev_type, & & psb_dlinmap_type, psb_dpk_, amg_d_onelev_type, &
& psb_ipk_, psb_epk_, psb_desc_type & psb_ipk_, psb_epk_, psb_desc_type
Implicit None Implicit None
! Arguments ! Arguments
class(amg_d_onelev_type), intent(inout) :: lv class(amg_d_onelev_type), intent(inout) :: lv
character(len=*), intent(in) :: what character(len=*), intent(in) :: what
character(len=*), intent(in) :: val character(len=*), intent(in) :: val
integer(psb_ipk_), intent(out) :: info integer(psb_ipk_), intent(out) :: info
character(len=*), optional, intent(in) :: pos character(len=*), optional, intent(in) :: pos
integer(psb_ipk_), intent(in), optional :: idx integer(psb_ipk_), intent(in), optional :: idx
end subroutine amg_d_base_onelev_csetc end subroutine amg_d_base_onelev_csetc
end interface end interface
interface interface
subroutine amg_d_base_onelev_csetr(lv,what,val,info,pos,idx) subroutine amg_d_base_onelev_csetr(lv,what,val,info,pos,idx)
import :: psb_dspmat_type, psb_d_vect_type, psb_d_base_vect_type, & import :: psb_dspmat_type, psb_d_vect_type, psb_d_base_vect_type, &
& psb_dlinmap_type, psb_dpk_, amg_d_onelev_type, & & psb_dlinmap_type, psb_dpk_, amg_d_onelev_type, &
& psb_ipk_, psb_epk_, psb_desc_type & psb_ipk_, psb_epk_, psb_desc_type
Implicit None Implicit None
class(amg_d_onelev_type), intent(inout) :: lv class(amg_d_onelev_type), intent(inout) :: lv
character(len=*), intent(in) :: what character(len=*), intent(in) :: what
real(psb_dpk_), intent(in) :: val real(psb_dpk_), intent(in) :: val
integer(psb_ipk_), intent(out) :: info integer(psb_ipk_), intent(out) :: info
character(len=*), optional, intent(in) :: pos character(len=*), optional, intent(in) :: pos
@ -398,13 +400,13 @@ interface
end subroutine amg_d_base_onelev_csetr end subroutine amg_d_base_onelev_csetr
end interface end interface
interface interface
subroutine amg_d_base_onelev_dump(lv,level,info,prefix,head,ac,rp,smoother,& subroutine amg_d_base_onelev_dump(lv,level,info,prefix,head,ac,rp,smoother,&
& solver,tprol,global_num) & solver,tprol,global_num)
import :: psb_dspmat_type, psb_d_vect_type, psb_d_base_vect_type, & import :: psb_dspmat_type, psb_d_vect_type, psb_d_base_vect_type, &
& psb_dlinmap_type, psb_dpk_, amg_d_onelev_type, & & psb_dlinmap_type, psb_dpk_, amg_d_onelev_type, &
& psb_ipk_, psb_epk_, psb_desc_type & psb_ipk_, psb_epk_, psb_desc_type
implicit none implicit none
class(amg_d_onelev_type), intent(in) :: lv class(amg_d_onelev_type), intent(in) :: lv
integer(psb_ipk_), intent(in) :: level integer(psb_ipk_), intent(in) :: level
integer(psb_ipk_), intent(out) :: info integer(psb_ipk_), intent(out) :: info
@ -436,7 +438,7 @@ interface
end subroutine amg_d_base_onelev_map_rstr_v end subroutine amg_d_base_onelev_map_rstr_v
end interface end interface
interface interface
subroutine amg_d_base_onelev_map_prol_a(lv,alpha,v,beta,u,info,work) subroutine amg_d_base_onelev_map_prol_a(lv,alpha,v,beta,u,info,work)
import import
implicit none implicit none
@ -459,15 +461,15 @@ interface
type(psb_d_vect_type), optional, target, intent(inout) :: vtx,vty type(psb_d_vect_type), optional, target, intent(inout) :: vtx,vty
end subroutine amg_d_base_onelev_map_prol_v end subroutine amg_d_base_onelev_map_prol_v
end interface end interface
contains contains
! !
! Function returning the size of the amg_prec_type data structure ! Function returning the size of the amg_prec_type data structure
! in bytes or in number of nonzeros of the operator(s) involved. ! in bytes or in number of nonzeros of the operator(s) involved.
! !
function d_base_onelev_get_nzeros(lv) result(val) function d_base_onelev_get_nzeros(lv) result(val)
implicit none implicit none
class(amg_d_onelev_type), intent(in) :: lv class(amg_d_onelev_type), intent(in) :: lv
integer(psb_epk_) :: val integer(psb_epk_) :: val
integer(psb_ipk_) :: i integer(psb_ipk_) :: i
@ -479,16 +481,16 @@ contains
end function d_base_onelev_get_nzeros end function d_base_onelev_get_nzeros
function d_base_onelev_sizeof(lv) result(val) function d_base_onelev_sizeof(lv) result(val)
implicit none implicit none
class(amg_d_onelev_type), intent(in) :: lv class(amg_d_onelev_type), intent(in) :: lv
integer(psb_epk_) :: val integer(psb_epk_) :: val
integer(psb_ipk_) :: i integer(psb_ipk_) :: i
val = psb_sizeof_ip+psb_sizeof_lp val = psb_sizeof_ip+psb_sizeof_lp
val = val + lv%desc_ac%sizeof() val = val + lv%desc_ac%sizeof()
val = val + lv%ac%sizeof() val = val + lv%ac%sizeof()
val = val + lv%tprol%sizeof() val = val + lv%tprol%sizeof()
val = val + lv%linmap%sizeof() val = val + lv%linmap%sizeof()
if (allocated(lv%sm)) val = val + lv%sm%sizeof() if (allocated(lv%sm)) val = val + lv%sm%sizeof()
if (allocated(lv%sm2a)) val = val + lv%sm2a%sizeof() if (allocated(lv%sm2a)) val = val + lv%sm2a%sizeof()
if (allocated(lv%aggr)) val = val + lv%aggr%sizeof() if (allocated(lv%aggr)) val = val + lv%aggr%sizeof()
@ -497,19 +499,19 @@ contains
subroutine d_base_onelev_nullify(lv) subroutine d_base_onelev_nullify(lv)
implicit none implicit none
class(amg_d_onelev_type), intent(inout) :: lv class(amg_d_onelev_type), intent(inout) :: lv
nullify(lv%base_a) nullify(lv%base_a)
nullify(lv%base_desc) nullify(lv%base_desc)
nullify(lv%sm2) nullify(lv%sm2)
end subroutine d_base_onelev_nullify end subroutine d_base_onelev_nullify
! !
! Multilevel defaults: ! Multilevel defaults:
! multiplicative vs. additive ML framework; ! multiplicative vs. additive ML framework;
! Smoothed decoupled aggregation with zero threshold; ! Smoothed decoupled aggregation with zero threshold;
! distributed coarse matrix; ! distributed coarse matrix;
! damping omega computed with the max-norm estimate of the ! damping omega computed with the max-norm estimate of the
! dominant eigenvalue; ! dominant eigenvalue;
@ -519,10 +521,10 @@ contains
subroutine d_base_onelev_default(lv) subroutine d_base_onelev_default(lv)
Implicit None Implicit None
! Arguments ! Arguments
class(amg_d_onelev_type), target, intent(inout) :: lv class(amg_d_onelev_type), target, intent(inout) :: lv
integer(psb_ipk_) :: info integer(psb_ipk_) :: info
lv%parms%sweeps_pre = 1 lv%parms%sweeps_pre = 1
lv%parms%sweeps_post = 1 lv%parms%sweeps_post = 1
@ -537,7 +539,7 @@ contains
lv%parms%aggr_filter = amg_no_filter_mat_ lv%parms%aggr_filter = amg_no_filter_mat_
lv%parms%aggr_omega_val = dzero lv%parms%aggr_omega_val = dzero
lv%parms%aggr_thresh = 0.01_psb_dpk_ lv%parms%aggr_thresh = 0.01_psb_dpk_
if (allocated(lv%sm)) call lv%sm%default() if (allocated(lv%sm)) call lv%sm%default()
if (allocated(lv%sm2a)) then if (allocated(lv%sm2a)) then
call lv%sm2a%default() call lv%sm2a%default()
@ -547,7 +549,7 @@ contains
end if end if
if (.not.allocated(lv%aggr)) allocate(amg_d_dec_aggregator_type :: lv%aggr,stat=info) if (.not.allocated(lv%aggr)) allocate(amg_d_dec_aggregator_type :: lv%aggr,stat=info)
if (allocated(lv%aggr)) call lv%aggr%default() if (allocated(lv%aggr)) call lv%aggr%default()
return return
end subroutine d_base_onelev_default end subroutine d_base_onelev_default
@ -562,9 +564,9 @@ contains
type(psb_ldspmat_type), intent(out) :: t_prol type(psb_ldspmat_type), intent(out) :: t_prol
type(amg_daggr_data), intent(in) :: ag_data type(amg_daggr_data), intent(in) :: ag_data
integer(psb_ipk_), intent(out) :: info integer(psb_ipk_), intent(out) :: info
call lv%aggr%bld_tprol(lv%parms,ag_data,a,desc_a,ilaggr,nlaggr,t_prol,info) call lv%aggr%bld_tprol(lv%parms,ag_data,a,desc_a,ilaggr,nlaggr,t_prol,info)
end subroutine d_base_onelev_bld_tprol end subroutine d_base_onelev_bld_tprol
@ -574,7 +576,7 @@ contains
integer(psb_ipk_), intent(out) :: info integer(psb_ipk_), intent(out) :: info
call lv%aggr%update_next(lvnext%aggr,info) call lv%aggr%update_next(lvnext%aggr,info)
end subroutine d_base_onelev_update_aggr end subroutine d_base_onelev_update_aggr
@ -583,33 +585,33 @@ contains
Implicit None Implicit None
! Arguments ! Arguments
class(amg_d_onelev_type), target, intent(inout) :: lv class(amg_d_onelev_type), target, intent(inout) :: lv
class(amg_d_onelev_type), target, intent(inout) :: lvout class(amg_d_onelev_type), target, intent(inout) :: lvout
integer(psb_ipk_), intent(out) :: info integer(psb_ipk_), intent(out) :: info
info = psb_success_ info = psb_success_
if (allocated(lv%sm)) then if (allocated(lv%sm)) then
call lv%sm%clone(lvout%sm,info) call lv%sm%clone(lvout%sm,info)
else else
if (allocated(lvout%sm)) then if (allocated(lvout%sm)) then
call lvout%sm%free(info) call lvout%sm%free(info)
if (info==psb_success_) deallocate(lvout%sm,stat=info) if (info==psb_success_) deallocate(lvout%sm,stat=info)
end if end if
end if end if
if (allocated(lv%sm2a)) then if (allocated(lv%sm2a)) then
call lv%sm%clone(lvout%sm2a,info) call lv%sm%clone(lvout%sm2a,info)
lvout%sm2 => lvout%sm2a lvout%sm2 => lvout%sm2a
else else
if (allocated(lvout%sm2a)) then if (allocated(lvout%sm2a)) then
call lvout%sm2a%free(info) call lvout%sm2a%free(info)
if (info==psb_success_) deallocate(lvout%sm2a,stat=info) if (info==psb_success_) deallocate(lvout%sm2a,stat=info)
end if end if
lvout%sm2 => lvout%sm lvout%sm2 => lvout%sm
end if end if
if (allocated(lv%aggr)) then if (allocated(lv%aggr)) then
call lv%aggr%clone(lvout%aggr,info) call lv%aggr%clone(lvout%aggr,info)
else else
if (allocated(lvout%aggr)) then if (allocated(lvout%aggr)) then
call lvout%aggr%free(info) call lvout%aggr%free(info)
if (info==psb_success_) deallocate(lvout%aggr,stat=info) if (info==psb_success_) deallocate(lvout%aggr,stat=info)
end if end if
@ -622,7 +624,7 @@ contains
if (info == psb_success_) call lv%remap_data%clone(lvout%remap_data,info) if (info == psb_success_) call lv%remap_data%clone(lvout%remap_data,info)
lvout%base_a => lv%base_a lvout%base_a => lv%base_a
lvout%base_desc => lv%base_desc lvout%base_desc => lv%base_desc
return return
end subroutine d_base_onelev_clone end subroutine d_base_onelev_clone
@ -631,12 +633,12 @@ contains
use psb_base_mod use psb_base_mod
implicit none implicit none
class(amg_d_onelev_type), target, intent(inout) :: lv, b class(amg_d_onelev_type), target, intent(inout) :: lv, b
integer(psb_ipk_), intent(out) :: info integer(psb_ipk_), intent(out) :: info
call b%free(info) call b%free(info)
b%parms = lv%parms b%parms = lv%parms
b%szratio = lv%szratio b%szratio = lv%szratio
if (associated(lv%sm2,lv%sm2a)) then if (associated(lv%sm2,lv%sm2a)) then
call move_alloc(lv%sm,b%sm) call move_alloc(lv%sm,b%sm)
call move_alloc(lv%sm2a,b%sm2a) call move_alloc(lv%sm2a,b%sm2a)
b%sm2 =>b%sm2a b%sm2 =>b%sm2a
@ -647,18 +649,18 @@ contains
end if end if
call move_alloc(lv%aggr,b%aggr) call move_alloc(lv%aggr,b%aggr)
if (info == psb_success_) call psb_move_alloc(lv%ac,b%ac,info) if (info == psb_success_) call psb_move_alloc(lv%ac,b%ac,info)
if (info == psb_success_) call psb_move_alloc(lv%tprol,b%tprol,info) if (info == psb_success_) call psb_move_alloc(lv%tprol,b%tprol,info)
if (info == psb_success_) call psb_move_alloc(lv%desc_ac,b%desc_ac,info) if (info == psb_success_) call psb_move_alloc(lv%desc_ac,b%desc_ac,info)
if (info == psb_success_) call psb_move_alloc(lv%linmap,b%linmap,info) if (info == psb_success_) call psb_move_alloc(lv%linmap,b%linmap,info)
b%base_a => lv%base_a b%base_a => lv%base_a
b%base_desc => lv%base_desc b%base_desc => lv%base_desc
end subroutine d_base_onelev_move_alloc end subroutine d_base_onelev_move_alloc
function d_base_onelev_get_wrksize(lv) result(val) function d_base_onelev_get_wrksize(lv) result(val)
implicit none implicit none
class(amg_d_onelev_type), intent(inout) :: lv class(amg_d_onelev_type), intent(inout) :: lv
integer(psb_ipk_) :: val integer(psb_ipk_) :: val
@ -679,26 +681,26 @@ contains
select case(lv%parms%ml_cycle) select case(lv%parms%ml_cycle)
case(amg_add_ml_,amg_mult_ml_,amg_vcycle_ml_, amg_wcycle_ml_) case(amg_add_ml_,amg_mult_ml_,amg_vcycle_ml_, amg_wcycle_ml_)
! We're good ! We're good
case(amg_kcycle_ml_, amg_kcyclesym_ml_) case(amg_kcycle_ml_, amg_kcyclesym_ml_)
! !
! We need 7 in inneritkcycle. ! We need 7 in inneritkcycle.
! Can we reuse vtx? ! Can we reuse vtx?
! !
val = val + 7 val = val + 7
case default case default
! Need a better error signaling ? ! Need a better error signaling ?
val = -1 val = -1
end select end select
end function d_base_onelev_get_wrksize end function d_base_onelev_get_wrksize
subroutine d_base_onelev_allocate_wrk(lv,info,vmold) subroutine d_base_onelev_allocate_wrk(lv,info,vmold)
use psb_base_mod use psb_base_mod
implicit none implicit none
class(amg_d_onelev_type), target, intent(inout) :: lv class(amg_d_onelev_type), target, intent(inout) :: lv
integer(psb_ipk_), intent(out) :: info integer(psb_ipk_), intent(out) :: info
class(psb_d_base_vect_type), intent(in), optional :: vmold class(psb_d_base_vect_type), intent(in), optional :: vmold
! !
integer(psb_ipk_) :: nwv, i integer(psb_ipk_) :: nwv, i
@ -711,22 +713,22 @@ contains
! Need to fix this, we need two different allocations ! Need to fix this, we need two different allocations
! !
call lv%wrk%alloc(nwv,lv%base_desc,info,vmold=vmold,& call lv%wrk%alloc(nwv,lv%base_desc,info,vmold=vmold,&
& desc2=lv%remap_data%desc_ac_pre_remap) & desc2=lv%remap_data%desc_ac_pre_remap)
else else
call lv%wrk%alloc(nwv,lv%base_desc,info,vmold=vmold) call lv%wrk%alloc(nwv,lv%base_desc,info,vmold=vmold)
end if end if
end if end if
end subroutine d_base_onelev_allocate_wrk end subroutine d_base_onelev_allocate_wrk
subroutine d_base_onelev_free_wrk(lv,info) subroutine d_base_onelev_free_wrk(lv,info)
use psb_base_mod use psb_base_mod
implicit none implicit none
class(amg_d_onelev_type), target, intent(inout) :: lv class(amg_d_onelev_type), target, intent(inout) :: lv
integer(psb_ipk_), intent(out) :: info integer(psb_ipk_), intent(out) :: info
! !
integer(psb_ipk_) :: nwv,i integer(psb_ipk_) :: nwv,i
info = psb_success_ info = psb_success_
if (allocated(lv%wrk)) then if (allocated(lv%wrk)) then
@ -734,17 +736,17 @@ contains
if (info == 0) deallocate(lv%wrk,stat=info) if (info == 0) deallocate(lv%wrk,stat=info)
end if end if
end subroutine d_base_onelev_free_wrk end subroutine d_base_onelev_free_wrk
subroutine d_wrk_alloc(wk,nwv,desc,info,vmold, desc2) subroutine d_wrk_alloc(wk,nwv,desc,info,vmold, desc2)
use psb_base_mod use psb_base_mod
Implicit None Implicit None
! Arguments ! Arguments
class(amg_dmlprec_wrk_type), target, intent(inout) :: wk class(amg_dmlprec_wrk_type), target, intent(inout) :: wk
integer(psb_ipk_), intent(in) :: nwv integer(psb_ipk_), intent(in) :: nwv
type(psb_desc_type), intent(in) :: desc type(psb_desc_type), intent(in) :: desc
integer(psb_ipk_), intent(out) :: info integer(psb_ipk_), intent(out) :: info
class(psb_d_base_vect_type), intent(in), optional :: vmold class(psb_d_base_vect_type), intent(in), optional :: vmold
type(psb_desc_type), intent(in), optional :: desc2 type(psb_desc_type), intent(in), optional :: desc2
! !
@ -808,14 +810,14 @@ contains
end do end do
end if end if
end subroutine d_wrk_alloc end subroutine d_wrk_alloc
subroutine d_wrk_free(wk,info) subroutine d_wrk_free(wk,info)
Implicit None Implicit None
! Arguments ! Arguments
class(amg_dmlprec_wrk_type), target, intent(inout) :: wk class(amg_dmlprec_wrk_type), target, intent(inout) :: wk
integer(psb_ipk_), intent(out) :: info integer(psb_ipk_), intent(out) :: info
! !
integer(psb_ipk_) :: i integer(psb_ipk_) :: i
info = psb_success_ info = psb_success_
@ -836,7 +838,7 @@ contains
end if end if
end subroutine d_wrk_free end subroutine d_wrk_free
subroutine d_wrk_clone(wk,wkout,info) subroutine d_wrk_clone(wk,wkout,info)
use psb_base_mod use psb_base_mod
Implicit None Implicit None
@ -844,11 +846,11 @@ contains
! Arguments ! Arguments
class(amg_dmlprec_wrk_type), target, intent(inout) :: wk class(amg_dmlprec_wrk_type), target, intent(inout) :: wk
class(amg_dmlprec_wrk_type), target, intent(inout) :: wkout class(amg_dmlprec_wrk_type), target, intent(inout) :: wkout
integer(psb_ipk_), intent(out) :: info integer(psb_ipk_), intent(out) :: info
! !
integer(psb_ipk_) :: i integer(psb_ipk_) :: i
info = psb_success_ info = psb_success_
call psb_safe_ab_cpy(wk%tx,wkout%tx,info) call psb_safe_ab_cpy(wk%tx,wkout%tx,info)
call psb_safe_ab_cpy(wk%ty,wkout%ty,info) call psb_safe_ab_cpy(wk%ty,wkout%ty,info)
call psb_safe_ab_cpy(wk%x2l,wkout%x2l,info) call psb_safe_ab_cpy(wk%x2l,wkout%x2l,info)
@ -870,12 +872,12 @@ contains
return return
end subroutine d_wrk_clone end subroutine d_wrk_clone
subroutine d_wrk_move_alloc(wk, b,info) subroutine d_wrk_move_alloc(wk, b,info)
implicit none implicit none
class(amg_dmlprec_wrk_type), target, intent(inout) :: wk, b class(amg_dmlprec_wrk_type), target, intent(inout) :: wk, b
integer(psb_ipk_), intent(out) :: info integer(psb_ipk_), intent(out) :: info
call b%free(info) call b%free(info)
call move_alloc(wk%tx,b%tx) call move_alloc(wk%tx,b%tx)
call move_alloc(wk%ty,b%ty) call move_alloc(wk%ty,b%ty)
@ -888,17 +890,17 @@ contains
call move_alloc(wk%vx2l%v,b%vx2l%v) call move_alloc(wk%vx2l%v,b%vx2l%v)
call move_alloc(wk%vy2l%v,b%vy2l%v) call move_alloc(wk%vy2l%v,b%vy2l%v)
call move_alloc(wk%wv,b%wv) call move_alloc(wk%wv,b%wv)
end subroutine d_wrk_move_alloc end subroutine d_wrk_move_alloc
subroutine d_wrk_cnv(wk,info,vmold) subroutine d_wrk_cnv(wk,info,vmold)
use psb_base_mod use psb_base_mod
Implicit None Implicit None
! Arguments ! Arguments
class(amg_dmlprec_wrk_type), target, intent(inout) :: wk class(amg_dmlprec_wrk_type), target, intent(inout) :: wk
integer(psb_ipk_), intent(out) :: info integer(psb_ipk_), intent(out) :: info
class(psb_d_base_vect_type), intent(in), optional :: vmold class(psb_d_base_vect_type), intent(in), optional :: vmold
! !
integer(psb_ipk_) :: i integer(psb_ipk_) :: i
@ -919,7 +921,7 @@ contains
function d_wrk_sizeof(wk) result(val) function d_wrk_sizeof(wk) result(val)
use psb_realloc_mod use psb_realloc_mod
implicit none implicit none
class(amg_dmlprec_wrk_type), intent(in) :: wk class(amg_dmlprec_wrk_type), intent(in) :: wk
integer(psb_epk_) :: val integer(psb_epk_) :: val
integer :: i integer :: i
@ -938,14 +940,14 @@ contains
end do end do
end if end if
end function d_wrk_sizeof end function d_wrk_sizeof
subroutine d_remap_data_clone(rmp, remap_out, info) subroutine d_remap_data_clone(rmp, remap_out, info)
use psb_base_mod use psb_base_mod
implicit none implicit none
! Arguments ! Arguments
class(amg_d_remap_data_type), target, intent(inout) :: rmp class(amg_d_remap_data_type), target, intent(inout) :: rmp
class(amg_d_remap_data_type), target, intent(inout) :: remap_out class(amg_d_remap_data_type), target, intent(inout) :: remap_out
integer(psb_ipk_), intent(out) :: info integer(psb_ipk_), intent(out) :: info
! !
integer(psb_ipk_) :: i integer(psb_ipk_) :: i
@ -956,7 +958,7 @@ contains
& call rmp%desc_ac_pre_remap%clone(remap_out%desc_ac_pre_remap,info) & call rmp%desc_ac_pre_remap%clone(remap_out%desc_ac_pre_remap,info)
remap_out%idest = rmp%idest remap_out%idest = rmp%idest
call psb_safe_ab_cpy(rmp%isrc,remap_out%isrc,info) call psb_safe_ab_cpy(rmp%isrc,remap_out%isrc,info)
call psb_safe_ab_cpy(rmp%nrsrc,remap_out%nrsrc,info) call psb_safe_ab_cpy(rmp%nrsrc,remap_out%nrsrc,info)
end subroutine d_remap_data_clone end subroutine d_remap_data_clone
end module amg_d_onelev_mod end module amg_d_onelev_mod

296
amgprec/amg_s_onelev_mod.f90
Unescape Escape View File

@ -1,15 +1,15 @@
! !
! !
! AMG4PSBLAS version 1.0 ! AMG4PSBLAS version 1.0
! Algebraic Multigrid Package ! Algebraic Multigrid Package
! based on PSBLAS (Parallel Sparse BLAS version 3.7) ! based on PSBLAS (Parallel Sparse BLAS version 3.7)
! !
! (C) Copyright 2021 ! (C) Copyright 2021
! !
! Salvatore Filippone ! Salvatore Filippone
! Pasqua D'Ambra ! Pasqua D'Ambra
! Fabio Durastante ! Fabio Durastante
! !
! Redistribution and use in source and binary forms, with or without ! Redistribution and use in source and binary forms, with or without
! modification, are permitted provided that the following conditions ! modification, are permitted provided that the following conditions
! are met: ! are met:
@ -21,7 +21,7 @@
! 3. The name of the AMG4PSBLAS group or the names of its contributors may ! 3. The name of the AMG4PSBLAS group or the names of its contributors may
! not be used to endorse or promote products derived from this ! not be used to endorse or promote products derived from this
! software without specific written permission. ! software without specific written permission.
! !
! THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS ! THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
! ``AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED ! ``AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED
! TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR ! TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
@ -33,22 +33,22 @@
! CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ! 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 ! ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
! POSSIBILITY OF SUCH DAMAGE. ! POSSIBILITY OF SUCH DAMAGE.
! !
! !
! File: amg_s_onelev_mod.f90 ! File: amg_s_onelev_mod.f90
! !
! Module: amg_s_onelev_mod ! Module: amg_s_onelev_mod
! !
! This module defines: ! This module defines:
! - the amg_s_onelev_type data structure containing one level ! - the amg_s_onelev_type data structure containing one level
! of a multilevel preconditioner and related ! of a multilevel preconditioner and related
! data structures; ! data structures;
! !
! It contains routines for ! It contains routines for
! - Building and applying; ! - Building and applying;
! - checking if the preconditioner is correctly defined; ! - checking if the preconditioner is correctly defined;
! - printing a description of the preconditioner; ! - printing a description of the preconditioner;
! - deallocating the preconditioner data structure. ! - deallocating the preconditioner data structure.
! !
module amg_s_onelev_mod module amg_s_onelev_mod
@ -56,6 +56,8 @@ module amg_s_onelev_mod
use amg_base_prec_type use amg_base_prec_type
use amg_s_base_smoother_mod use amg_s_base_smoother_mod
use amg_s_dec_aggregator_mod use amg_s_dec_aggregator_mod
use amg_s_parmatch_aggregator_mod
use psb_base_mod, only : psb_sspmat_type, psb_s_vect_type, & use psb_base_mod, only : psb_sspmat_type, psb_s_vect_type, &
& psb_s_base_vect_type, psb_lsspmat_type, psb_slinmap_type, psb_spk_, & & psb_s_base_vect_type, psb_lsspmat_type, psb_slinmap_type, psb_spk_, &
& psb_ipk_, psb_epk_, psb_lpk_, psb_desc_type, psb_i_base_vect_type, & & psb_ipk_, psb_epk_, psb_lpk_, psb_desc_type, psb_i_base_vect_type, &
@ -73,11 +75,11 @@ module amg_s_onelev_mod
! class(amg_s_base_smoother_type), pointer :: sm2 => null() ! class(amg_s_base_smoother_type), pointer :: sm2 => null()
! class(amg_smlprec_wrk_type), allocatable :: wrk ! class(amg_smlprec_wrk_type), allocatable :: wrk
! class(amg_s_base_aggregator_type), allocatable :: aggr ! class(amg_s_base_aggregator_type), allocatable :: aggr
! type(amg_sml_parms) :: parms ! type(amg_sml_parms) :: parms
! type(psb_sspmat_type) :: ac ! type(psb_sspmat_type) :: ac
! type(psb_sesc_type) :: desc_ac ! type(psb_sesc_type) :: desc_ac
! type(psb_sspmat_type), pointer :: base_a => null() ! type(psb_sspmat_type), pointer :: base_a => null()
! type(psb_desc_type), pointer :: base_desc => null() ! type(psb_desc_type), pointer :: base_desc => null()
! type(psb_slinmap_type) :: map ! type(psb_slinmap_type) :: map
! end type amg_sonelev_type ! end type amg_sonelev_type
! !
@ -93,7 +95,7 @@ module amg_s_onelev_mod
! Workspace for application of preconditioner; may be ! Workspace for application of preconditioner; may be
! pre-allocated to save time in the application within a ! pre-allocated to save time in the application within a
! Krylov solver. ! Krylov solver.
! aggr - class(amg_s_base_aggregator_type), allocatable ! aggr - class(amg_s_base_aggregator_type), allocatable
! The aggregator object: holds the algorithmic choices and ! The aggregator object: holds the algorithmic choices and
! (possibly) additional data for building the aggregation. ! (possibly) additional data for building the aggregation.
! parms - type(amg_sml_parms) ! parms - type(amg_sml_parms)
@ -104,7 +106,7 @@ module amg_s_onelev_mod
! The communication descriptor associated to the matrix ! The communication descriptor associated to the matrix
! stored in ac. ! stored in ac.
! base_a - type(psb_sspmat_type), pointer. ! base_a - type(psb_sspmat_type), pointer.
! Pointer (really a pointer!) to the local part of the current ! Pointer (really a pointer!) to the local part of the current
! matrix (so we have a unified treatment of residuals). ! matrix (so we have a unified treatment of residuals).
! We need this to avoid passing explicitly the current matrix ! We need this to avoid passing explicitly the current matrix
! to the routine which applies the preconditioner. ! to the routine which applies the preconditioner.
@ -115,13 +117,13 @@ module amg_s_onelev_mod
! vector spaces associated to the index spaces of the previous ! vector spaces associated to the index spaces of the previous
! and current levels. ! and current levels.
! !
! Methods: ! Methods:
! Most methods follow the encapsulation hierarchy: they take whatever action ! Most methods follow the encapsulation hierarchy: they take whatever action
! is appropriate for the current object, then call the corresponding method for ! is appropriate for the current object, then call the corresponding method for
! the contained object. ! the contained object.
! As an example: the descr() method prints out a description of the ! As an example: the descr() method prints out a description of the
! level. It starts by invoking the descr() method of the parms object, ! level. It starts by invoking the descr() method of the parms object,
! then calls the descr() method of the smoother object. ! then calls the descr() method of the smoother object.
! !
! descr - Prints a description of the object. ! descr - Prints a description of the object.
! default - Set default values ! default - Set default values
@ -130,14 +132,14 @@ module amg_s_onelev_mod
! it is passed to the smoother object for further processing. ! it is passed to the smoother object for further processing.
! check - Sanity checks. ! check - Sanity checks.
! sizeof - Total memory occupation in bytes ! sizeof - Total memory occupation in bytes
! get_nzeros - Number of nonzeros ! get_nzeros - Number of nonzeros
! get_wrksz - How many workspace vector does apply_vect need ! get_wrksz - How many workspace vector does apply_vect need
! allocate_wrk - Allocate auxiliary workspace ! allocate_wrk - Allocate auxiliary workspace
! free_wrk - Free auxiliary workspace ! free_wrk - Free auxiliary workspace
! bld_tprol - Invoke the aggr method to build the tentative prolongator ! bld_tprol - Invoke the aggr method to build the tentative prolongator
! mat_asb - Build the final (possibly smoothed) prolongator and coarse matrix. ! mat_asb - Build the final (possibly smoothed) prolongator and coarse matrix.
!
! !
!
type amg_smlprec_wrk_type type amg_smlprec_wrk_type
real(psb_spk_), allocatable :: tx(:), ty(:), x2l(:), y2l(:) real(psb_spk_), allocatable :: tx(:), ty(:), x2l(:), y2l(:)
type(psb_s_vect_type) :: vtx, vty, vx2l, vy2l type(psb_s_vect_type) :: vtx, vty, vx2l, vy2l
@ -148,10 +150,10 @@ module amg_s_onelev_mod
procedure, pass(wk) :: clone => s_wrk_clone procedure, pass(wk) :: clone => s_wrk_clone
procedure, pass(wk) :: move_alloc => s_wrk_move_alloc procedure, pass(wk) :: move_alloc => s_wrk_move_alloc
procedure, pass(wk) :: cnv => s_wrk_cnv procedure, pass(wk) :: cnv => s_wrk_cnv
procedure, pass(wk) :: sizeof => s_wrk_sizeof procedure, pass(wk) :: sizeof => s_wrk_sizeof
end type amg_smlprec_wrk_type end type amg_smlprec_wrk_type
private :: s_wrk_alloc, s_wrk_free, & private :: s_wrk_alloc, s_wrk_free, &
& s_wrk_clone, s_wrk_move_alloc, s_wrk_cnv, s_wrk_sizeof & s_wrk_clone, s_wrk_move_alloc, s_wrk_cnv, s_wrk_sizeof
type amg_s_remap_data_type type amg_s_remap_data_type
type(psb_sspmat_type) :: ac_pre_remap type(psb_sspmat_type) :: ac_pre_remap
@ -161,19 +163,19 @@ module amg_s_onelev_mod
contains contains
procedure, pass(rmp) :: clone => s_remap_data_clone procedure, pass(rmp) :: clone => s_remap_data_clone
end type amg_s_remap_data_type end type amg_s_remap_data_type
type amg_s_onelev_type type amg_s_onelev_type
class(amg_s_base_smoother_type), allocatable :: sm, sm2a class(amg_s_base_smoother_type), allocatable :: sm, sm2a
class(amg_s_base_smoother_type), pointer :: sm2 => null() class(amg_s_base_smoother_type), pointer :: sm2 => null()
class(amg_smlprec_wrk_type), allocatable :: wrk class(amg_smlprec_wrk_type), allocatable :: wrk
class(amg_s_base_aggregator_type), allocatable :: aggr class(amg_s_base_aggregator_type), allocatable :: aggr
type(amg_sml_parms) :: parms type(amg_sml_parms) :: parms
type(psb_sspmat_type) :: ac type(psb_sspmat_type) :: ac
integer(psb_ipk_) :: ac_nz_loc integer(psb_ipk_) :: ac_nz_loc
integer(psb_lpk_) :: ac_nz_tot integer(psb_lpk_) :: ac_nz_tot
type(psb_desc_type) :: desc_ac type(psb_desc_type) :: desc_ac
type(psb_sspmat_type), pointer :: base_a => null() type(psb_sspmat_type), pointer :: base_a => null()
type(psb_desc_type), pointer :: base_desc => null() type(psb_desc_type), pointer :: base_desc => null()
type(psb_lsspmat_type) :: tprol type(psb_lsspmat_type) :: tprol
type(psb_slinmap_type) :: linmap type(psb_slinmap_type) :: linmap
type(amg_s_remap_data_type) :: remap_data type(amg_s_remap_data_type) :: remap_data
@ -197,7 +199,7 @@ module amg_s_onelev_mod
procedure, pass(lv) :: setsm => amg_s_base_onelev_setsm procedure, pass(lv) :: setsm => amg_s_base_onelev_setsm
procedure, pass(lv) :: setsv => amg_s_base_onelev_setsv procedure, pass(lv) :: setsv => amg_s_base_onelev_setsv
procedure, pass(lv) :: setag => amg_s_base_onelev_setag procedure, pass(lv) :: setag => amg_s_base_onelev_setag
generic, public :: set => cseti, csetr, csetc, setsm, setsv, setag generic, public :: set => cseti, csetr, csetc, setsm, setsv, setag
procedure, pass(lv) :: sizeof => s_base_onelev_sizeof procedure, pass(lv) :: sizeof => s_base_onelev_sizeof
procedure, pass(lv) :: get_nzeros => s_base_onelev_get_nzeros procedure, pass(lv) :: get_nzeros => s_base_onelev_get_nzeros
procedure, pass(lv) :: get_wrksz => s_base_onelev_get_wrksize procedure, pass(lv) :: get_wrksz => s_base_onelev_get_wrksize
@ -205,8 +207,8 @@ module amg_s_onelev_mod
procedure, pass(lv) :: free_wrk => s_base_onelev_free_wrk procedure, pass(lv) :: free_wrk => s_base_onelev_free_wrk
procedure, nopass :: stringval => amg_stringval procedure, nopass :: stringval => amg_stringval
procedure, pass(lv) :: move_alloc => s_base_onelev_move_alloc procedure, pass(lv) :: move_alloc => s_base_onelev_move_alloc
procedure, pass(lv) :: map_rstr_a => amg_s_base_onelev_map_rstr_a procedure, pass(lv) :: map_rstr_a => amg_s_base_onelev_map_rstr_a
procedure, pass(lv) :: map_prol_a => amg_s_base_onelev_map_prol_a procedure, pass(lv) :: map_prol_a => amg_s_base_onelev_map_prol_a
procedure, pass(lv) :: map_rstr_v => amg_s_base_onelev_map_rstr_v procedure, pass(lv) :: map_rstr_v => amg_s_base_onelev_map_rstr_v
@ -226,11 +228,11 @@ module amg_s_onelev_mod
& s_base_onelev_get_wrksize, s_base_onelev_allocate_wrk, & & s_base_onelev_get_wrksize, s_base_onelev_allocate_wrk, &
& s_base_onelev_free_wrk & s_base_onelev_free_wrk
interface interface
subroutine amg_s_base_onelev_mat_asb(lv,a,desc_a,ilaggr,nlaggr,t_prol,info) subroutine amg_s_base_onelev_mat_asb(lv,a,desc_a,ilaggr,nlaggr,t_prol,info)
import :: psb_sspmat_type, psb_desc_type, psb_spk_, psb_ipk_, psb_lsspmat_type, psb_lpk_ import :: psb_sspmat_type, psb_desc_type, psb_spk_, psb_ipk_, psb_lsspmat_type, psb_lpk_
import :: amg_s_onelev_type import :: amg_s_onelev_type
implicit none implicit none
class(amg_s_onelev_type), intent(inout), target :: lv class(amg_s_onelev_type), intent(inout), target :: lv
type(psb_sspmat_type), intent(in) :: a type(psb_sspmat_type), intent(in) :: a
type(psb_desc_type), intent(inout) :: desc_a type(psb_desc_type), intent(inout) :: desc_a
@ -255,7 +257,7 @@ module amg_s_onelev_mod
end subroutine amg_s_base_onelev_build end subroutine amg_s_base_onelev_build
end interface end interface
interface interface
subroutine amg_s_base_onelev_descr(lv,il,nl,ilmin,info,iout, verbosity) subroutine amg_s_base_onelev_descr(lv,il,nl,ilmin,info,iout, verbosity)
import :: psb_sspmat_type, psb_s_vect_type, psb_s_base_vect_type, & import :: psb_sspmat_type, psb_s_vect_type, psb_s_base_vect_type, &
& psb_slinmap_type, psb_spk_, amg_s_onelev_type, & & psb_slinmap_type, psb_spk_, amg_s_onelev_type, &
@ -270,127 +272,127 @@ module amg_s_onelev_mod
end subroutine amg_s_base_onelev_descr end subroutine amg_s_base_onelev_descr
end interface end interface
interface interface
subroutine amg_s_base_onelev_cnv(lv,info,amold,vmold,imold) subroutine amg_s_base_onelev_cnv(lv,info,amold,vmold,imold)
import :: amg_s_onelev_type, psb_s_base_vect_type, psb_spk_, & import :: amg_s_onelev_type, psb_s_base_vect_type, psb_spk_, &
& psb_s_base_sparse_mat, psb_ipk_, psb_i_base_vect_type & psb_s_base_sparse_mat, psb_ipk_, psb_i_base_vect_type
! Arguments ! Arguments
class(amg_s_onelev_type), intent(inout) :: lv class(amg_s_onelev_type), intent(inout) :: lv
integer(psb_ipk_), intent(out) :: info integer(psb_ipk_), intent(out) :: info
class(psb_s_base_sparse_mat), intent(in), optional :: amold class(psb_s_base_sparse_mat), intent(in), optional :: amold
class(psb_s_base_vect_type), intent(in), optional :: vmold class(psb_s_base_vect_type), intent(in), optional :: vmold
class(psb_i_base_vect_type), intent(in), optional :: imold class(psb_i_base_vect_type), intent(in), optional :: imold
end subroutine amg_s_base_onelev_cnv end subroutine amg_s_base_onelev_cnv
end interface end interface
interface interface
subroutine amg_s_base_onelev_free(lv,info) subroutine amg_s_base_onelev_free(lv,info)
import :: psb_sspmat_type, psb_s_vect_type, psb_s_base_vect_type, & import :: psb_sspmat_type, psb_s_vect_type, psb_s_base_vect_type, &
& psb_slinmap_type, psb_spk_, amg_s_onelev_type, & & psb_slinmap_type, psb_spk_, amg_s_onelev_type, &
& psb_ipk_, psb_epk_, psb_desc_type & psb_ipk_, psb_epk_, psb_desc_type
implicit none implicit none
class(amg_s_onelev_type), intent(inout) :: lv class(amg_s_onelev_type), intent(inout) :: lv
integer(psb_ipk_), intent(out) :: info integer(psb_ipk_), intent(out) :: info
end subroutine amg_s_base_onelev_free end subroutine amg_s_base_onelev_free
end interface end interface
interface interface
subroutine amg_s_base_onelev_check(lv,info) subroutine amg_s_base_onelev_check(lv,info)
import :: psb_sspmat_type, psb_s_vect_type, psb_s_base_vect_type, & import :: psb_sspmat_type, psb_s_vect_type, psb_s_base_vect_type, &
& psb_slinmap_type, psb_spk_, amg_s_onelev_type, & & psb_slinmap_type, psb_spk_, amg_s_onelev_type, &
& psb_ipk_, psb_epk_, psb_desc_type & psb_ipk_, psb_epk_, psb_desc_type
Implicit None Implicit None
! Arguments ! Arguments
class(amg_s_onelev_type), intent(inout) :: lv class(amg_s_onelev_type), intent(inout) :: lv
integer(psb_ipk_), intent(out) :: info integer(psb_ipk_), intent(out) :: info
end subroutine amg_s_base_onelev_check end subroutine amg_s_base_onelev_check
end interface end interface
interface interface
subroutine amg_s_base_onelev_setsm(lv,val,info,pos) subroutine amg_s_base_onelev_setsm(lv,val,info,pos)
import :: psb_spk_, amg_s_onelev_type, amg_s_base_smoother_type, & import :: psb_spk_, amg_s_onelev_type, amg_s_base_smoother_type, &
& psb_ipk_, psb_epk_, psb_desc_type & psb_ipk_, psb_epk_, psb_desc_type
Implicit None Implicit None
! Arguments ! Arguments
class(amg_s_onelev_type), target, intent(inout) :: lv class(amg_s_onelev_type), target, intent(inout) :: lv
class(amg_s_base_smoother_type), intent(in) :: val class(amg_s_base_smoother_type), intent(in) :: val
integer(psb_ipk_), intent(out) :: info integer(psb_ipk_), intent(out) :: info
character(len=*), optional, intent(in) :: pos character(len=*), optional, intent(in) :: pos
end subroutine amg_s_base_onelev_setsm end subroutine amg_s_base_onelev_setsm
end interface end interface
interface interface
subroutine amg_s_base_onelev_setsv(lv,val,info,pos) subroutine amg_s_base_onelev_setsv(lv,val,info,pos)
import :: psb_spk_, amg_s_onelev_type, amg_s_base_solver_type, & import :: psb_spk_, amg_s_onelev_type, amg_s_base_solver_type, &
& psb_ipk_, psb_epk_, psb_desc_type & psb_ipk_, psb_epk_, psb_desc_type
Implicit None Implicit None
! Arguments ! Arguments
class(amg_s_onelev_type), target, intent(inout) :: lv class(amg_s_onelev_type), target, intent(inout) :: lv
class(amg_s_base_solver_type), intent(in) :: val class(amg_s_base_solver_type), intent(in) :: val
integer(psb_ipk_), intent(out) :: info integer(psb_ipk_), intent(out) :: info
character(len=*), optional, intent(in) :: pos character(len=*), optional, intent(in) :: pos
end subroutine amg_s_base_onelev_setsv end subroutine amg_s_base_onelev_setsv
end interface end interface
interface interface
subroutine amg_s_base_onelev_setag(lv,val,info,pos) subroutine amg_s_base_onelev_setag(lv,val,info,pos)
import :: psb_spk_, amg_s_onelev_type, amg_s_base_aggregator_type, & import :: psb_spk_, amg_s_onelev_type, amg_s_base_aggregator_type, &
& psb_ipk_, psb_epk_, psb_desc_type & psb_ipk_, psb_epk_, psb_desc_type
Implicit None Implicit None
! Arguments ! Arguments
class(amg_s_onelev_type), target, intent(inout) :: lv class(amg_s_onelev_type), target, intent(inout) :: lv
class(amg_s_base_aggregator_type), intent(in) :: val class(amg_s_base_aggregator_type), intent(in) :: val
integer(psb_ipk_), intent(out) :: info integer(psb_ipk_), intent(out) :: info
character(len=*), optional, intent(in) :: pos character(len=*), optional, intent(in) :: pos
end subroutine amg_s_base_onelev_setag end subroutine amg_s_base_onelev_setag
end interface end interface
interface interface
subroutine amg_s_base_onelev_cseti(lv,what,val,info,pos,idx) subroutine amg_s_base_onelev_cseti(lv,what,val,info,pos,idx)
import :: psb_sspmat_type, psb_s_vect_type, psb_s_base_vect_type, & import :: psb_sspmat_type, psb_s_vect_type, psb_s_base_vect_type, &
& psb_slinmap_type, psb_spk_, amg_s_onelev_type, & & psb_slinmap_type, psb_spk_, amg_s_onelev_type, &
& psb_ipk_, psb_epk_, psb_desc_type & psb_ipk_, psb_epk_, psb_desc_type
Implicit None Implicit None
! Arguments ! Arguments
class(amg_s_onelev_type), intent(inout) :: lv class(amg_s_onelev_type), intent(inout) :: lv
character(len=*), intent(in) :: what character(len=*), intent(in) :: what
integer(psb_ipk_), intent(in) :: val integer(psb_ipk_), intent(in) :: val
integer(psb_ipk_), intent(out) :: info integer(psb_ipk_), intent(out) :: info
character(len=*), optional, intent(in) :: pos character(len=*), optional, intent(in) :: pos
integer(psb_ipk_), intent(in), optional :: idx integer(psb_ipk_), intent(in), optional :: idx
end subroutine amg_s_base_onelev_cseti end subroutine amg_s_base_onelev_cseti
end interface end interface
interface interface
subroutine amg_s_base_onelev_csetc(lv,what,val,info,pos,idx) subroutine amg_s_base_onelev_csetc(lv,what,val,info,pos,idx)
import :: psb_sspmat_type, psb_s_vect_type, psb_s_base_vect_type, & import :: psb_sspmat_type, psb_s_vect_type, psb_s_base_vect_type, &
& psb_slinmap_type, psb_spk_, amg_s_onelev_type, & & psb_slinmap_type, psb_spk_, amg_s_onelev_type, &
& psb_ipk_, psb_epk_, psb_desc_type & psb_ipk_, psb_epk_, psb_desc_type
Implicit None Implicit None
! Arguments ! Arguments
class(amg_s_onelev_type), intent(inout) :: lv class(amg_s_onelev_type), intent(inout) :: lv
character(len=*), intent(in) :: what character(len=*), intent(in) :: what
character(len=*), intent(in) :: val character(len=*), intent(in) :: val
integer(psb_ipk_), intent(out) :: info integer(psb_ipk_), intent(out) :: info
character(len=*), optional, intent(in) :: pos character(len=*), optional, intent(in) :: pos
integer(psb_ipk_), intent(in), optional :: idx integer(psb_ipk_), intent(in), optional :: idx
end subroutine amg_s_base_onelev_csetc end subroutine amg_s_base_onelev_csetc
end interface end interface
interface interface
subroutine amg_s_base_onelev_csetr(lv,what,val,info,pos,idx) subroutine amg_s_base_onelev_csetr(lv,what,val,info,pos,idx)
import :: psb_sspmat_type, psb_s_vect_type, psb_s_base_vect_type, & import :: psb_sspmat_type, psb_s_vect_type, psb_s_base_vect_type, &
& psb_slinmap_type, psb_spk_, amg_s_onelev_type, & & psb_slinmap_type, psb_spk_, amg_s_onelev_type, &
& psb_ipk_, psb_epk_, psb_desc_type & psb_ipk_, psb_epk_, psb_desc_type
Implicit None Implicit None
class(amg_s_onelev_type), intent(inout) :: lv class(amg_s_onelev_type), intent(inout) :: lv
character(len=*), intent(in) :: what character(len=*), intent(in) :: what
real(psb_spk_), intent(in) :: val real(psb_spk_), intent(in) :: val
integer(psb_ipk_), intent(out) :: info integer(psb_ipk_), intent(out) :: info
character(len=*), optional, intent(in) :: pos character(len=*), optional, intent(in) :: pos
@ -398,13 +400,13 @@ interface
end subroutine amg_s_base_onelev_csetr end subroutine amg_s_base_onelev_csetr
end interface end interface
interface interface
subroutine amg_s_base_onelev_dump(lv,level,info,prefix,head,ac,rp,smoother,& subroutine amg_s_base_onelev_dump(lv,level,info,prefix,head,ac,rp,smoother,&
& solver,tprol,global_num) & solver,tprol,global_num)
import :: psb_sspmat_type, psb_s_vect_type, psb_s_base_vect_type, & import :: psb_sspmat_type, psb_s_vect_type, psb_s_base_vect_type, &
& psb_slinmap_type, psb_spk_, amg_s_onelev_type, & & psb_slinmap_type, psb_spk_, amg_s_onelev_type, &
& psb_ipk_, psb_epk_, psb_desc_type & psb_ipk_, psb_epk_, psb_desc_type
implicit none implicit none
class(amg_s_onelev_type), intent(in) :: lv class(amg_s_onelev_type), intent(in) :: lv
integer(psb_ipk_), intent(in) :: level integer(psb_ipk_), intent(in) :: level
integer(psb_ipk_), intent(out) :: info integer(psb_ipk_), intent(out) :: info
@ -436,7 +438,7 @@ interface
end subroutine amg_s_base_onelev_map_rstr_v end subroutine amg_s_base_onelev_map_rstr_v
end interface end interface
interface interface
subroutine amg_s_base_onelev_map_prol_a(lv,alpha,v,beta,u,info,work) subroutine amg_s_base_onelev_map_prol_a(lv,alpha,v,beta,u,info,work)
import import
implicit none implicit none
@ -459,15 +461,15 @@ interface
type(psb_s_vect_type), optional, target, intent(inout) :: vtx,vty type(psb_s_vect_type), optional, target, intent(inout) :: vtx,vty
end subroutine amg_s_base_onelev_map_prol_v end subroutine amg_s_base_onelev_map_prol_v
end interface end interface
contains contains
! !
! Function returning the size of the amg_prec_type data structure ! Function returning the size of the amg_prec_type data structure
! in bytes or in number of nonzeros of the operator(s) involved. ! in bytes or in number of nonzeros of the operator(s) involved.
! !
function s_base_onelev_get_nzeros(lv) result(val) function s_base_onelev_get_nzeros(lv) result(val)
implicit none implicit none
class(amg_s_onelev_type), intent(in) :: lv class(amg_s_onelev_type), intent(in) :: lv
integer(psb_epk_) :: val integer(psb_epk_) :: val
integer(psb_ipk_) :: i integer(psb_ipk_) :: i
@ -479,16 +481,16 @@ contains
end function s_base_onelev_get_nzeros end function s_base_onelev_get_nzeros
function s_base_onelev_sizeof(lv) result(val) function s_base_onelev_sizeof(lv) result(val)
implicit none implicit none
class(amg_s_onelev_type), intent(in) :: lv class(amg_s_onelev_type), intent(in) :: lv
integer(psb_epk_) :: val integer(psb_epk_) :: val
integer(psb_ipk_) :: i integer(psb_ipk_) :: i
val = psb_sizeof_ip+psb_sizeof_lp val = psb_sizeof_ip+psb_sizeof_lp
val = val + lv%desc_ac%sizeof() val = val + lv%desc_ac%sizeof()
val = val + lv%ac%sizeof() val = val + lv%ac%sizeof()
val = val + lv%tprol%sizeof() val = val + lv%tprol%sizeof()
val = val + lv%linmap%sizeof() val = val + lv%linmap%sizeof()
if (allocated(lv%sm)) val = val + lv%sm%sizeof() if (allocated(lv%sm)) val = val + lv%sm%sizeof()
if (allocated(lv%sm2a)) val = val + lv%sm2a%sizeof() if (allocated(lv%sm2a)) val = val + lv%sm2a%sizeof()
if (allocated(lv%aggr)) val = val + lv%aggr%sizeof() if (allocated(lv%aggr)) val = val + lv%aggr%sizeof()
@ -497,19 +499,19 @@ contains
subroutine s_base_onelev_nullify(lv) subroutine s_base_onelev_nullify(lv)
implicit none implicit none
class(amg_s_onelev_type), intent(inout) :: lv class(amg_s_onelev_type), intent(inout) :: lv
nullify(lv%base_a) nullify(lv%base_a)
nullify(lv%base_desc) nullify(lv%base_desc)
nullify(lv%sm2) nullify(lv%sm2)
end subroutine s_base_onelev_nullify end subroutine s_base_onelev_nullify
! !
! Multilevel defaults: ! Multilevel defaults:
! multiplicative vs. additive ML framework; ! multiplicative vs. additive ML framework;
! Smoothed decoupled aggregation with zero threshold; ! Smoothed decoupled aggregation with zero threshold;
! distributed coarse matrix; ! distributed coarse matrix;
! damping omega computed with the max-norm estimate of the ! damping omega computed with the max-norm estimate of the
! dominant eigenvalue; ! dominant eigenvalue;
@ -519,10 +521,10 @@ contains
subroutine s_base_onelev_default(lv) subroutine s_base_onelev_default(lv)
Implicit None Implicit None
! Arguments ! Arguments
class(amg_s_onelev_type), target, intent(inout) :: lv class(amg_s_onelev_type), target, intent(inout) :: lv
integer(psb_ipk_) :: info integer(psb_ipk_) :: info
lv%parms%sweeps_pre = 1 lv%parms%sweeps_pre = 1
lv%parms%sweeps_post = 1 lv%parms%sweeps_post = 1
@ -537,7 +539,7 @@ contains
lv%parms%aggr_filter = amg_no_filter_mat_ lv%parms%aggr_filter = amg_no_filter_mat_
lv%parms%aggr_omega_val = szero lv%parms%aggr_omega_val = szero
lv%parms%aggr_thresh = 0.01_psb_spk_ lv%parms%aggr_thresh = 0.01_psb_spk_
if (allocated(lv%sm)) call lv%sm%default() if (allocated(lv%sm)) call lv%sm%default()
if (allocated(lv%sm2a)) then if (allocated(lv%sm2a)) then
call lv%sm2a%default() call lv%sm2a%default()
@ -547,7 +549,7 @@ contains
end if end if
if (.not.allocated(lv%aggr)) allocate(amg_s_dec_aggregator_type :: lv%aggr,stat=info) if (.not.allocated(lv%aggr)) allocate(amg_s_dec_aggregator_type :: lv%aggr,stat=info)
if (allocated(lv%aggr)) call lv%aggr%default() if (allocated(lv%aggr)) call lv%aggr%default()
return return
end subroutine s_base_onelev_default end subroutine s_base_onelev_default
@ -562,9 +564,9 @@ contains
type(psb_lsspmat_type), intent(out) :: t_prol type(psb_lsspmat_type), intent(out) :: t_prol
type(amg_saggr_data), intent(in) :: ag_data type(amg_saggr_data), intent(in) :: ag_data
integer(psb_ipk_), intent(out) :: info integer(psb_ipk_), intent(out) :: info
call lv%aggr%bld_tprol(lv%parms,ag_data,a,desc_a,ilaggr,nlaggr,t_prol,info) call lv%aggr%bld_tprol(lv%parms,ag_data,a,desc_a,ilaggr,nlaggr,t_prol,info)
end subroutine s_base_onelev_bld_tprol end subroutine s_base_onelev_bld_tprol
@ -574,7 +576,7 @@ contains
integer(psb_ipk_), intent(out) :: info integer(psb_ipk_), intent(out) :: info
call lv%aggr%update_next(lvnext%aggr,info) call lv%aggr%update_next(lvnext%aggr,info)
end subroutine s_base_onelev_update_aggr end subroutine s_base_onelev_update_aggr
@ -583,33 +585,33 @@ contains
Implicit None Implicit None
! Arguments ! Arguments
class(amg_s_onelev_type), target, intent(inout) :: lv class(amg_s_onelev_type), target, intent(inout) :: lv
class(amg_s_onelev_type), target, intent(inout) :: lvout class(amg_s_onelev_type), target, intent(inout) :: lvout
integer(psb_ipk_), intent(out) :: info integer(psb_ipk_), intent(out) :: info
info = psb_success_ info = psb_success_
if (allocated(lv%sm)) then if (allocated(lv%sm)) then
call lv%sm%clone(lvout%sm,info) call lv%sm%clone(lvout%sm,info)
else else
if (allocated(lvout%sm)) then if (allocated(lvout%sm)) then
call lvout%sm%free(info) call lvout%sm%free(info)
if (info==psb_success_) deallocate(lvout%sm,stat=info) if (info==psb_success_) deallocate(lvout%sm,stat=info)
end if end if
end if end if
if (allocated(lv%sm2a)) then if (allocated(lv%sm2a)) then
call lv%sm%clone(lvout%sm2a,info) call lv%sm%clone(lvout%sm2a,info)
lvout%sm2 => lvout%sm2a lvout%sm2 => lvout%sm2a
else else
if (allocated(lvout%sm2a)) then if (allocated(lvout%sm2a)) then
call lvout%sm2a%free(info) call lvout%sm2a%free(info)
if (info==psb_success_) deallocate(lvout%sm2a,stat=info) if (info==psb_success_) deallocate(lvout%sm2a,stat=info)
end if end if
lvout%sm2 => lvout%sm lvout%sm2 => lvout%sm
end if end if
if (allocated(lv%aggr)) then if (allocated(lv%aggr)) then
call lv%aggr%clone(lvout%aggr,info) call lv%aggr%clone(lvout%aggr,info)
else else
if (allocated(lvout%aggr)) then if (allocated(lvout%aggr)) then
call lvout%aggr%free(info) call lvout%aggr%free(info)
if (info==psb_success_) deallocate(lvout%aggr,stat=info) if (info==psb_success_) deallocate(lvout%aggr,stat=info)
end if end if
@ -622,7 +624,7 @@ contains
if (info == psb_success_) call lv%remap_data%clone(lvout%remap_data,info) if (info == psb_success_) call lv%remap_data%clone(lvout%remap_data,info)
lvout%base_a => lv%base_a lvout%base_a => lv%base_a
lvout%base_desc => lv%base_desc lvout%base_desc => lv%base_desc
return return
end subroutine s_base_onelev_clone end subroutine s_base_onelev_clone
@ -631,12 +633,12 @@ contains
use psb_base_mod use psb_base_mod
implicit none implicit none
class(amg_s_onelev_type), target, intent(inout) :: lv, b class(amg_s_onelev_type), target, intent(inout) :: lv, b
integer(psb_ipk_), intent(out) :: info integer(psb_ipk_), intent(out) :: info
call b%free(info) call b%free(info)
b%parms = lv%parms b%parms = lv%parms
b%szratio = lv%szratio b%szratio = lv%szratio
if (associated(lv%sm2,lv%sm2a)) then if (associated(lv%sm2,lv%sm2a)) then
call move_alloc(lv%sm,b%sm) call move_alloc(lv%sm,b%sm)
call move_alloc(lv%sm2a,b%sm2a) call move_alloc(lv%sm2a,b%sm2a)
b%sm2 =>b%sm2a b%sm2 =>b%sm2a
@ -647,18 +649,18 @@ contains
end if end if
call move_alloc(lv%aggr,b%aggr) call move_alloc(lv%aggr,b%aggr)
if (info == psb_success_) call psb_move_alloc(lv%ac,b%ac,info) if (info == psb_success_) call psb_move_alloc(lv%ac,b%ac,info)
if (info == psb_success_) call psb_move_alloc(lv%tprol,b%tprol,info) if (info == psb_success_) call psb_move_alloc(lv%tprol,b%tprol,info)
if (info == psb_success_) call psb_move_alloc(lv%desc_ac,b%desc_ac,info) if (info == psb_success_) call psb_move_alloc(lv%desc_ac,b%desc_ac,info)
if (info == psb_success_) call psb_move_alloc(lv%linmap,b%linmap,info) if (info == psb_success_) call psb_move_alloc(lv%linmap,b%linmap,info)
b%base_a => lv%base_a b%base_a => lv%base_a
b%base_desc => lv%base_desc b%base_desc => lv%base_desc
end subroutine s_base_onelev_move_alloc end subroutine s_base_onelev_move_alloc
function s_base_onelev_get_wrksize(lv) result(val) function s_base_onelev_get_wrksize(lv) result(val)
implicit none implicit none
class(amg_s_onelev_type), intent(inout) :: lv class(amg_s_onelev_type), intent(inout) :: lv
integer(psb_ipk_) :: val integer(psb_ipk_) :: val
@ -679,26 +681,26 @@ contains
select case(lv%parms%ml_cycle) select case(lv%parms%ml_cycle)
case(amg_add_ml_,amg_mult_ml_,amg_vcycle_ml_, amg_wcycle_ml_) case(amg_add_ml_,amg_mult_ml_,amg_vcycle_ml_, amg_wcycle_ml_)
! We're good ! We're good
case(amg_kcycle_ml_, amg_kcyclesym_ml_) case(amg_kcycle_ml_, amg_kcyclesym_ml_)
! !
! We need 7 in inneritkcycle. ! We need 7 in inneritkcycle.
! Can we reuse vtx? ! Can we reuse vtx?
! !
val = val + 7 val = val + 7
case default case default
! Need a better error signaling ? ! Need a better error signaling ?
val = -1 val = -1
end select end select
end function s_base_onelev_get_wrksize end function s_base_onelev_get_wrksize
subroutine s_base_onelev_allocate_wrk(lv,info,vmold) subroutine s_base_onelev_allocate_wrk(lv,info,vmold)
use psb_base_mod use psb_base_mod
implicit none implicit none
class(amg_s_onelev_type), target, intent(inout) :: lv class(amg_s_onelev_type), target, intent(inout) :: lv
integer(psb_ipk_), intent(out) :: info integer(psb_ipk_), intent(out) :: info
class(psb_s_base_vect_type), intent(in), optional :: vmold class(psb_s_base_vect_type), intent(in), optional :: vmold
! !
integer(psb_ipk_) :: nwv, i integer(psb_ipk_) :: nwv, i
@ -711,22 +713,22 @@ contains
! Need to fix this, we need two different allocations ! Need to fix this, we need two different allocations
! !
call lv%wrk%alloc(nwv,lv%base_desc,info,vmold=vmold,& call lv%wrk%alloc(nwv,lv%base_desc,info,vmold=vmold,&
& desc2=lv%remap_data%desc_ac_pre_remap) & desc2=lv%remap_data%desc_ac_pre_remap)
else else
call lv%wrk%alloc(nwv,lv%base_desc,info,vmold=vmold) call lv%wrk%alloc(nwv,lv%base_desc,info,vmold=vmold)
end if end if
end if end if
end subroutine s_base_onelev_allocate_wrk end subroutine s_base_onelev_allocate_wrk
subroutine s_base_onelev_free_wrk(lv,info) subroutine s_base_onelev_free_wrk(lv,info)
use psb_base_mod use psb_base_mod
implicit none implicit none
class(amg_s_onelev_type), target, intent(inout) :: lv class(amg_s_onelev_type), target, intent(inout) :: lv
integer(psb_ipk_), intent(out) :: info integer(psb_ipk_), intent(out) :: info
! !
integer(psb_ipk_) :: nwv,i integer(psb_ipk_) :: nwv,i
info = psb_success_ info = psb_success_
if (allocated(lv%wrk)) then if (allocated(lv%wrk)) then
@ -734,17 +736,17 @@ contains
if (info == 0) deallocate(lv%wrk,stat=info) if (info == 0) deallocate(lv%wrk,stat=info)
end if end if
end subroutine s_base_onelev_free_wrk end subroutine s_base_onelev_free_wrk
subroutine s_wrk_alloc(wk,nwv,desc,info,vmold, desc2) subroutine s_wrk_alloc(wk,nwv,desc,info,vmold, desc2)
use psb_base_mod use psb_base_mod
Implicit None Implicit None
! Arguments ! Arguments
class(amg_smlprec_wrk_type), target, intent(inout) :: wk class(amg_smlprec_wrk_type), target, intent(inout) :: wk
integer(psb_ipk_), intent(in) :: nwv integer(psb_ipk_), intent(in) :: nwv
type(psb_desc_type), intent(in) :: desc type(psb_desc_type), intent(in) :: desc
integer(psb_ipk_), intent(out) :: info integer(psb_ipk_), intent(out) :: info
class(psb_s_base_vect_type), intent(in), optional :: vmold class(psb_s_base_vect_type), intent(in), optional :: vmold
type(psb_desc_type), intent(in), optional :: desc2 type(psb_desc_type), intent(in), optional :: desc2
! !
@ -808,14 +810,14 @@ contains
end do end do
end if end if
end subroutine s_wrk_alloc end subroutine s_wrk_alloc
subroutine s_wrk_free(wk,info) subroutine s_wrk_free(wk,info)
Implicit None Implicit None
! Arguments ! Arguments
class(amg_smlprec_wrk_type), target, intent(inout) :: wk class(amg_smlprec_wrk_type), target, intent(inout) :: wk
integer(psb_ipk_), intent(out) :: info integer(psb_ipk_), intent(out) :: info
! !
integer(psb_ipk_) :: i integer(psb_ipk_) :: i
info = psb_success_ info = psb_success_
@ -836,7 +838,7 @@ contains
end if end if
end subroutine s_wrk_free end subroutine s_wrk_free
subroutine s_wrk_clone(wk,wkout,info) subroutine s_wrk_clone(wk,wkout,info)
use psb_base_mod use psb_base_mod
Implicit None Implicit None
@ -844,11 +846,11 @@ contains
! Arguments ! Arguments
class(amg_smlprec_wrk_type), target, intent(inout) :: wk class(amg_smlprec_wrk_type), target, intent(inout) :: wk
class(amg_smlprec_wrk_type), target, intent(inout) :: wkout class(amg_smlprec_wrk_type), target, intent(inout) :: wkout
integer(psb_ipk_), intent(out) :: info integer(psb_ipk_), intent(out) :: info
! !
integer(psb_ipk_) :: i integer(psb_ipk_) :: i
info = psb_success_ info = psb_success_
call psb_safe_ab_cpy(wk%tx,wkout%tx,info) call psb_safe_ab_cpy(wk%tx,wkout%tx,info)
call psb_safe_ab_cpy(wk%ty,wkout%ty,info) call psb_safe_ab_cpy(wk%ty,wkout%ty,info)
call psb_safe_ab_cpy(wk%x2l,wkout%x2l,info) call psb_safe_ab_cpy(wk%x2l,wkout%x2l,info)
@ -870,12 +872,12 @@ contains
return return
end subroutine s_wrk_clone end subroutine s_wrk_clone
subroutine s_wrk_move_alloc(wk, b,info) subroutine s_wrk_move_alloc(wk, b,info)
implicit none implicit none
class(amg_smlprec_wrk_type), target, intent(inout) :: wk, b class(amg_smlprec_wrk_type), target, intent(inout) :: wk, b
integer(psb_ipk_), intent(out) :: info integer(psb_ipk_), intent(out) :: info
call b%free(info) call b%free(info)
call move_alloc(wk%tx,b%tx) call move_alloc(wk%tx,b%tx)
call move_alloc(wk%ty,b%ty) call move_alloc(wk%ty,b%ty)
@ -888,17 +890,17 @@ contains
call move_alloc(wk%vx2l%v,b%vx2l%v) call move_alloc(wk%vx2l%v,b%vx2l%v)
call move_alloc(wk%vy2l%v,b%vy2l%v) call move_alloc(wk%vy2l%v,b%vy2l%v)
call move_alloc(wk%wv,b%wv) call move_alloc(wk%wv,b%wv)
end subroutine s_wrk_move_alloc end subroutine s_wrk_move_alloc
subroutine s_wrk_cnv(wk,info,vmold) subroutine s_wrk_cnv(wk,info,vmold)
use psb_base_mod use psb_base_mod
Implicit None Implicit None
! Arguments ! Arguments
class(amg_smlprec_wrk_type), target, intent(inout) :: wk class(amg_smlprec_wrk_type), target, intent(inout) :: wk
integer(psb_ipk_), intent(out) :: info integer(psb_ipk_), intent(out) :: info
class(psb_s_base_vect_type), intent(in), optional :: vmold class(psb_s_base_vect_type), intent(in), optional :: vmold
! !
integer(psb_ipk_) :: i integer(psb_ipk_) :: i
@ -919,7 +921,7 @@ contains
function s_wrk_sizeof(wk) result(val) function s_wrk_sizeof(wk) result(val)
use psb_realloc_mod use psb_realloc_mod
implicit none implicit none
class(amg_smlprec_wrk_type), intent(in) :: wk class(amg_smlprec_wrk_type), intent(in) :: wk
integer(psb_epk_) :: val integer(psb_epk_) :: val
integer :: i integer :: i
@ -938,14 +940,14 @@ contains
end do end do
end if end if
end function s_wrk_sizeof end function s_wrk_sizeof
subroutine s_remap_data_clone(rmp, remap_out, info) subroutine s_remap_data_clone(rmp, remap_out, info)
use psb_base_mod use psb_base_mod
implicit none implicit none
! Arguments ! Arguments
class(amg_s_remap_data_type), target, intent(inout) :: rmp class(amg_s_remap_data_type), target, intent(inout) :: rmp
class(amg_s_remap_data_type), target, intent(inout) :: remap_out class(amg_s_remap_data_type), target, intent(inout) :: remap_out
integer(psb_ipk_), intent(out) :: info integer(psb_ipk_), intent(out) :: info
! !
integer(psb_ipk_) :: i integer(psb_ipk_) :: i
@ -956,7 +958,7 @@ contains
& call rmp%desc_ac_pre_remap%clone(remap_out%desc_ac_pre_remap,info) & call rmp%desc_ac_pre_remap%clone(remap_out%desc_ac_pre_remap,info)
remap_out%idest = rmp%idest remap_out%idest = rmp%idest
call psb_safe_ab_cpy(rmp%isrc,remap_out%isrc,info) call psb_safe_ab_cpy(rmp%isrc,remap_out%isrc,info)
call psb_safe_ab_cpy(rmp%nrsrc,remap_out%nrsrc,info) call psb_safe_ab_cpy(rmp%nrsrc,remap_out%nrsrc,info)
end subroutine s_remap_data_clone end subroutine s_remap_data_clone
end module amg_s_onelev_mod end module amg_s_onelev_mod

295
amgprec/amg_z_onelev_mod.f90
Unescape Escape View File

@ -1,15 +1,15 @@
! !
! !
! AMG4PSBLAS version 1.0 ! AMG4PSBLAS version 1.0
! Algebraic Multigrid Package ! Algebraic Multigrid Package
! based on PSBLAS (Parallel Sparse BLAS version 3.7) ! based on PSBLAS (Parallel Sparse BLAS version 3.7)
! !
! (C) Copyright 2021 ! (C) Copyright 2021
! !
! Salvatore Filippone ! Salvatore Filippone
! Pasqua D'Ambra ! Pasqua D'Ambra
! Fabio Durastante ! Fabio Durastante
! !
! Redistribution and use in source and binary forms, with or without ! Redistribution and use in source and binary forms, with or without
! modification, are permitted provided that the following conditions ! modification, are permitted provided that the following conditions
! are met: ! are met:
@ -21,7 +21,7 @@
! 3. The name of the AMG4PSBLAS group or the names of its contributors may ! 3. The name of the AMG4PSBLAS group or the names of its contributors may
! not be used to endorse or promote products derived from this ! not be used to endorse or promote products derived from this
! software without specific written permission. ! software without specific written permission.
! !
! THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS ! THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
! ``AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED ! ``AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED
! TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR ! TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
@ -33,22 +33,22 @@
! CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ! 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 ! ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
! POSSIBILITY OF SUCH DAMAGE. ! POSSIBILITY OF SUCH DAMAGE.
! !
! !
! File: amg_z_onelev_mod.f90 ! File: amg_z_onelev_mod.f90
! !
! Module: amg_z_onelev_mod ! Module: amg_z_onelev_mod
! !
! This module defines: ! This module defines:
! - the amg_z_onelev_type data structure containing one level ! - the amg_z_onelev_type data structure containing one level
! of a multilevel preconditioner and related ! of a multilevel preconditioner and related
! data structures; ! data structures;
! !
! It contains routines for ! It contains routines for
! - Building and applying; ! - Building and applying;
! - checking if the preconditioner is correctly defined; ! - checking if the preconditioner is correctly defined;
! - printing a description of the preconditioner; ! - printing a description of the preconditioner;
! - deallocating the preconditioner data structure. ! - deallocating the preconditioner data structure.
! !
module amg_z_onelev_mod module amg_z_onelev_mod
@ -56,6 +56,7 @@ module amg_z_onelev_mod
use amg_base_prec_type use amg_base_prec_type
use amg_z_base_smoother_mod use amg_z_base_smoother_mod
use amg_z_dec_aggregator_mod use amg_z_dec_aggregator_mod
use psb_base_mod, only : psb_zspmat_type, psb_z_vect_type, & use psb_base_mod, only : psb_zspmat_type, psb_z_vect_type, &
& psb_z_base_vect_type, psb_lzspmat_type, psb_zlinmap_type, psb_dpk_, & & psb_z_base_vect_type, psb_lzspmat_type, psb_zlinmap_type, psb_dpk_, &
& psb_ipk_, psb_epk_, psb_lpk_, psb_desc_type, psb_i_base_vect_type, & & psb_ipk_, psb_epk_, psb_lpk_, psb_desc_type, psb_i_base_vect_type, &
@ -73,11 +74,11 @@ module amg_z_onelev_mod
! class(amg_z_base_smoother_type), pointer :: sm2 => null() ! class(amg_z_base_smoother_type), pointer :: sm2 => null()
! class(amg_zmlprec_wrk_type), allocatable :: wrk ! class(amg_zmlprec_wrk_type), allocatable :: wrk
! class(amg_z_base_aggregator_type), allocatable :: aggr ! class(amg_z_base_aggregator_type), allocatable :: aggr
! type(amg_dml_parms) :: parms ! type(amg_dml_parms) :: parms
! type(psb_zspmat_type) :: ac ! type(psb_zspmat_type) :: ac
! type(psb_zesc_type) :: desc_ac ! type(psb_zesc_type) :: desc_ac
! type(psb_zspmat_type), pointer :: base_a => null() ! type(psb_zspmat_type), pointer :: base_a => null()
! type(psb_desc_type), pointer :: base_desc => null() ! type(psb_desc_type), pointer :: base_desc => null()
! type(psb_zlinmap_type) :: map ! type(psb_zlinmap_type) :: map
! end type amg_zonelev_type ! end type amg_zonelev_type
! !
@ -93,7 +94,7 @@ module amg_z_onelev_mod
! Workspace for application of preconditioner; may be ! Workspace for application of preconditioner; may be
! pre-allocated to save time in the application within a ! pre-allocated to save time in the application within a
! Krylov solver. ! Krylov solver.
! aggr - class(amg_z_base_aggregator_type), allocatable ! aggr - class(amg_z_base_aggregator_type), allocatable
! The aggregator object: holds the algorithmic choices and ! The aggregator object: holds the algorithmic choices and
! (possibly) additional data for building the aggregation. ! (possibly) additional data for building the aggregation.
! parms - type(amg_dml_parms) ! parms - type(amg_dml_parms)
@ -104,7 +105,7 @@ module amg_z_onelev_mod
! The communication descriptor associated to the matrix ! The communication descriptor associated to the matrix
! stored in ac. ! stored in ac.
! base_a - type(psb_zspmat_type), pointer. ! base_a - type(psb_zspmat_type), pointer.
! Pointer (really a pointer!) to the local part of the current ! Pointer (really a pointer!) to the local part of the current
! matrix (so we have a unified treatment of residuals). ! matrix (so we have a unified treatment of residuals).
! We need this to avoid passing explicitly the current matrix ! We need this to avoid passing explicitly the current matrix
! to the routine which applies the preconditioner. ! to the routine which applies the preconditioner.
@ -115,13 +116,13 @@ module amg_z_onelev_mod
! vector spaces associated to the index spaces of the previous ! vector spaces associated to the index spaces of the previous
! and current levels. ! and current levels.
! !
! Methods: ! Methods:
! Most methods follow the encapsulation hierarchy: they take whatever action ! Most methods follow the encapsulation hierarchy: they take whatever action
! is appropriate for the current object, then call the corresponding method for ! is appropriate for the current object, then call the corresponding method for
! the contained object. ! the contained object.
! As an example: the descr() method prints out a description of the ! As an example: the descr() method prints out a description of the
! level. It starts by invoking the descr() method of the parms object, ! level. It starts by invoking the descr() method of the parms object,
! then calls the descr() method of the smoother object. ! then calls the descr() method of the smoother object.
! !
! descr - Prints a description of the object. ! descr - Prints a description of the object.
! default - Set default values ! default - Set default values
@ -130,14 +131,14 @@ module amg_z_onelev_mod
! it is passed to the smoother object for further processing. ! it is passed to the smoother object for further processing.
! check - Sanity checks. ! check - Sanity checks.
! sizeof - Total memory occupation in bytes ! sizeof - Total memory occupation in bytes
! get_nzeros - Number of nonzeros ! get_nzeros - Number of nonzeros
! get_wrksz - How many workspace vector does apply_vect need ! get_wrksz - How many workspace vector does apply_vect need
! allocate_wrk - Allocate auxiliary workspace ! allocate_wrk - Allocate auxiliary workspace
! free_wrk - Free auxiliary workspace ! free_wrk - Free auxiliary workspace
! bld_tprol - Invoke the aggr method to build the tentative prolongator ! bld_tprol - Invoke the aggr method to build the tentative prolongator
! mat_asb - Build the final (possibly smoothed) prolongator and coarse matrix. ! mat_asb - Build the final (possibly smoothed) prolongator and coarse matrix.
!
! !
!
type amg_zmlprec_wrk_type type amg_zmlprec_wrk_type
complex(psb_dpk_), allocatable :: tx(:), ty(:), x2l(:), y2l(:) complex(psb_dpk_), allocatable :: tx(:), ty(:), x2l(:), y2l(:)
type(psb_z_vect_type) :: vtx, vty, vx2l, vy2l type(psb_z_vect_type) :: vtx, vty, vx2l, vy2l
@ -148,10 +149,10 @@ module amg_z_onelev_mod
procedure, pass(wk) :: clone => z_wrk_clone procedure, pass(wk) :: clone => z_wrk_clone
procedure, pass(wk) :: move_alloc => z_wrk_move_alloc procedure, pass(wk) :: move_alloc => z_wrk_move_alloc
procedure, pass(wk) :: cnv => z_wrk_cnv procedure, pass(wk) :: cnv => z_wrk_cnv
procedure, pass(wk) :: sizeof => z_wrk_sizeof procedure, pass(wk) :: sizeof => z_wrk_sizeof
end type amg_zmlprec_wrk_type end type amg_zmlprec_wrk_type
private :: z_wrk_alloc, z_wrk_free, & private :: z_wrk_alloc, z_wrk_free, &
& z_wrk_clone, z_wrk_move_alloc, z_wrk_cnv, z_wrk_sizeof & z_wrk_clone, z_wrk_move_alloc, z_wrk_cnv, z_wrk_sizeof
type amg_z_remap_data_type type amg_z_remap_data_type
type(psb_zspmat_type) :: ac_pre_remap type(psb_zspmat_type) :: ac_pre_remap
@ -161,19 +162,19 @@ module amg_z_onelev_mod
contains contains
procedure, pass(rmp) :: clone => z_remap_data_clone procedure, pass(rmp) :: clone => z_remap_data_clone
end type amg_z_remap_data_type end type amg_z_remap_data_type
type amg_z_onelev_type type amg_z_onelev_type
class(amg_z_base_smoother_type), allocatable :: sm, sm2a class(amg_z_base_smoother_type), allocatable :: sm, sm2a
class(amg_z_base_smoother_type), pointer :: sm2 => null() class(amg_z_base_smoother_type), pointer :: sm2 => null()
class(amg_zmlprec_wrk_type), allocatable :: wrk class(amg_zmlprec_wrk_type), allocatable :: wrk
class(amg_z_base_aggregator_type), allocatable :: aggr class(amg_z_base_aggregator_type), allocatable :: aggr
type(amg_dml_parms) :: parms type(amg_dml_parms) :: parms
type(psb_zspmat_type) :: ac type(psb_zspmat_type) :: ac
integer(psb_ipk_) :: ac_nz_loc integer(psb_ipk_) :: ac_nz_loc
integer(psb_lpk_) :: ac_nz_tot integer(psb_lpk_) :: ac_nz_tot
type(psb_desc_type) :: desc_ac type(psb_desc_type) :: desc_ac
type(psb_zspmat_type), pointer :: base_a => null() type(psb_zspmat_type), pointer :: base_a => null()
type(psb_desc_type), pointer :: base_desc => null() type(psb_desc_type), pointer :: base_desc => null()
type(psb_lzspmat_type) :: tprol type(psb_lzspmat_type) :: tprol
type(psb_zlinmap_type) :: linmap type(psb_zlinmap_type) :: linmap
type(amg_z_remap_data_type) :: remap_data type(amg_z_remap_data_type) :: remap_data
@ -197,7 +198,7 @@ module amg_z_onelev_mod
procedure, pass(lv) :: setsm => amg_z_base_onelev_setsm procedure, pass(lv) :: setsm => amg_z_base_onelev_setsm
procedure, pass(lv) :: setsv => amg_z_base_onelev_setsv procedure, pass(lv) :: setsv => amg_z_base_onelev_setsv
procedure, pass(lv) :: setag => amg_z_base_onelev_setag procedure, pass(lv) :: setag => amg_z_base_onelev_setag
generic, public :: set => cseti, csetr, csetc, setsm, setsv, setag generic, public :: set => cseti, csetr, csetc, setsm, setsv, setag
procedure, pass(lv) :: sizeof => z_base_onelev_sizeof procedure, pass(lv) :: sizeof => z_base_onelev_sizeof
procedure, pass(lv) :: get_nzeros => z_base_onelev_get_nzeros procedure, pass(lv) :: get_nzeros => z_base_onelev_get_nzeros
procedure, pass(lv) :: get_wrksz => z_base_onelev_get_wrksize procedure, pass(lv) :: get_wrksz => z_base_onelev_get_wrksize
@ -205,8 +206,8 @@ module amg_z_onelev_mod
procedure, pass(lv) :: free_wrk => z_base_onelev_free_wrk procedure, pass(lv) :: free_wrk => z_base_onelev_free_wrk
procedure, nopass :: stringval => amg_stringval procedure, nopass :: stringval => amg_stringval
procedure, pass(lv) :: move_alloc => z_base_onelev_move_alloc procedure, pass(lv) :: move_alloc => z_base_onelev_move_alloc
procedure, pass(lv) :: map_rstr_a => amg_z_base_onelev_map_rstr_a procedure, pass(lv) :: map_rstr_a => amg_z_base_onelev_map_rstr_a
procedure, pass(lv) :: map_prol_a => amg_z_base_onelev_map_prol_a procedure, pass(lv) :: map_prol_a => amg_z_base_onelev_map_prol_a
procedure, pass(lv) :: map_rstr_v => amg_z_base_onelev_map_rstr_v procedure, pass(lv) :: map_rstr_v => amg_z_base_onelev_map_rstr_v
@ -226,11 +227,11 @@ module amg_z_onelev_mod
& z_base_onelev_get_wrksize, z_base_onelev_allocate_wrk, & & z_base_onelev_get_wrksize, z_base_onelev_allocate_wrk, &
& z_base_onelev_free_wrk & z_base_onelev_free_wrk
interface interface
subroutine amg_z_base_onelev_mat_asb(lv,a,desc_a,ilaggr,nlaggr,t_prol,info) subroutine amg_z_base_onelev_mat_asb(lv,a,desc_a,ilaggr,nlaggr,t_prol,info)
import :: psb_zspmat_type, psb_desc_type, psb_dpk_, psb_ipk_, psb_lzspmat_type, psb_lpk_ import :: psb_zspmat_type, psb_desc_type, psb_dpk_, psb_ipk_, psb_lzspmat_type, psb_lpk_
import :: amg_z_onelev_type import :: amg_z_onelev_type
implicit none implicit none
class(amg_z_onelev_type), intent(inout), target :: lv class(amg_z_onelev_type), intent(inout), target :: lv
type(psb_zspmat_type), intent(in) :: a type(psb_zspmat_type), intent(in) :: a
type(psb_desc_type), intent(inout) :: desc_a type(psb_desc_type), intent(inout) :: desc_a
@ -255,7 +256,7 @@ module amg_z_onelev_mod
end subroutine amg_z_base_onelev_build end subroutine amg_z_base_onelev_build
end interface end interface
interface interface
subroutine amg_z_base_onelev_descr(lv,il,nl,ilmin,info,iout, verbosity) subroutine amg_z_base_onelev_descr(lv,il,nl,ilmin,info,iout, verbosity)
import :: psb_zspmat_type, psb_z_vect_type, psb_z_base_vect_type, & import :: psb_zspmat_type, psb_z_vect_type, psb_z_base_vect_type, &
& psb_zlinmap_type, psb_dpk_, amg_z_onelev_type, & & psb_zlinmap_type, psb_dpk_, amg_z_onelev_type, &
@ -270,127 +271,127 @@ module amg_z_onelev_mod
end subroutine amg_z_base_onelev_descr end subroutine amg_z_base_onelev_descr
end interface end interface
interface interface
subroutine amg_z_base_onelev_cnv(lv,info,amold,vmold,imold) subroutine amg_z_base_onelev_cnv(lv,info,amold,vmold,imold)
import :: amg_z_onelev_type, psb_z_base_vect_type, psb_dpk_, & import :: amg_z_onelev_type, psb_z_base_vect_type, psb_dpk_, &
& psb_z_base_sparse_mat, psb_ipk_, psb_i_base_vect_type & psb_z_base_sparse_mat, psb_ipk_, psb_i_base_vect_type
! Arguments ! Arguments
class(amg_z_onelev_type), intent(inout) :: lv class(amg_z_onelev_type), intent(inout) :: lv
integer(psb_ipk_), intent(out) :: info integer(psb_ipk_), intent(out) :: info
class(psb_z_base_sparse_mat), intent(in), optional :: amold class(psb_z_base_sparse_mat), intent(in), optional :: amold
class(psb_z_base_vect_type), intent(in), optional :: vmold class(psb_z_base_vect_type), intent(in), optional :: vmold
class(psb_i_base_vect_type), intent(in), optional :: imold class(psb_i_base_vect_type), intent(in), optional :: imold
end subroutine amg_z_base_onelev_cnv end subroutine amg_z_base_onelev_cnv
end interface end interface
interface interface
subroutine amg_z_base_onelev_free(lv,info) subroutine amg_z_base_onelev_free(lv,info)
import :: psb_zspmat_type, psb_z_vect_type, psb_z_base_vect_type, & import :: psb_zspmat_type, psb_z_vect_type, psb_z_base_vect_type, &
& psb_zlinmap_type, psb_dpk_, amg_z_onelev_type, & & psb_zlinmap_type, psb_dpk_, amg_z_onelev_type, &
& psb_ipk_, psb_epk_, psb_desc_type & psb_ipk_, psb_epk_, psb_desc_type
implicit none implicit none
class(amg_z_onelev_type), intent(inout) :: lv class(amg_z_onelev_type), intent(inout) :: lv
integer(psb_ipk_), intent(out) :: info integer(psb_ipk_), intent(out) :: info
end subroutine amg_z_base_onelev_free end subroutine amg_z_base_onelev_free
end interface end interface
interface interface
subroutine amg_z_base_onelev_check(lv,info) subroutine amg_z_base_onelev_check(lv,info)
import :: psb_zspmat_type, psb_z_vect_type, psb_z_base_vect_type, & import :: psb_zspmat_type, psb_z_vect_type, psb_z_base_vect_type, &
& psb_zlinmap_type, psb_dpk_, amg_z_onelev_type, & & psb_zlinmap_type, psb_dpk_, amg_z_onelev_type, &
& psb_ipk_, psb_epk_, psb_desc_type & psb_ipk_, psb_epk_, psb_desc_type
Implicit None Implicit None
! Arguments ! Arguments
class(amg_z_onelev_type), intent(inout) :: lv class(amg_z_onelev_type), intent(inout) :: lv
integer(psb_ipk_), intent(out) :: info integer(psb_ipk_), intent(out) :: info
end subroutine amg_z_base_onelev_check end subroutine amg_z_base_onelev_check
end interface end interface
interface interface
subroutine amg_z_base_onelev_setsm(lv,val,info,pos) subroutine amg_z_base_onelev_setsm(lv,val,info,pos)
import :: psb_dpk_, amg_z_onelev_type, amg_z_base_smoother_type, & import :: psb_dpk_, amg_z_onelev_type, amg_z_base_smoother_type, &
& psb_ipk_, psb_epk_, psb_desc_type & psb_ipk_, psb_epk_, psb_desc_type
Implicit None Implicit None
! Arguments ! Arguments
class(amg_z_onelev_type), target, intent(inout) :: lv class(amg_z_onelev_type), target, intent(inout) :: lv
class(amg_z_base_smoother_type), intent(in) :: val class(amg_z_base_smoother_type), intent(in) :: val
integer(psb_ipk_), intent(out) :: info integer(psb_ipk_), intent(out) :: info
character(len=*), optional, intent(in) :: pos character(len=*), optional, intent(in) :: pos
end subroutine amg_z_base_onelev_setsm end subroutine amg_z_base_onelev_setsm
end interface end interface
interface interface
subroutine amg_z_base_onelev_setsv(lv,val,info,pos) subroutine amg_z_base_onelev_setsv(lv,val,info,pos)
import :: psb_dpk_, amg_z_onelev_type, amg_z_base_solver_type, & import :: psb_dpk_, amg_z_onelev_type, amg_z_base_solver_type, &
& psb_ipk_, psb_epk_, psb_desc_type & psb_ipk_, psb_epk_, psb_desc_type
Implicit None Implicit None
! Arguments ! Arguments
class(amg_z_onelev_type), target, intent(inout) :: lv class(amg_z_onelev_type), target, intent(inout) :: lv
class(amg_z_base_solver_type), intent(in) :: val class(amg_z_base_solver_type), intent(in) :: val
integer(psb_ipk_), intent(out) :: info integer(psb_ipk_), intent(out) :: info
character(len=*), optional, intent(in) :: pos character(len=*), optional, intent(in) :: pos
end subroutine amg_z_base_onelev_setsv end subroutine amg_z_base_onelev_setsv
end interface end interface
interface interface
subroutine amg_z_base_onelev_setag(lv,val,info,pos) subroutine amg_z_base_onelev_setag(lv,val,info,pos)
import :: psb_dpk_, amg_z_onelev_type, amg_z_base_aggregator_type, & import :: psb_dpk_, amg_z_onelev_type, amg_z_base_aggregator_type, &
& psb_ipk_, psb_epk_, psb_desc_type & psb_ipk_, psb_epk_, psb_desc_type
Implicit None Implicit None
! Arguments ! Arguments
class(amg_z_onelev_type), target, intent(inout) :: lv class(amg_z_onelev_type), target, intent(inout) :: lv
class(amg_z_base_aggregator_type), intent(in) :: val class(amg_z_base_aggregator_type), intent(in) :: val
integer(psb_ipk_), intent(out) :: info integer(psb_ipk_), intent(out) :: info
character(len=*), optional, intent(in) :: pos character(len=*), optional, intent(in) :: pos
end subroutine amg_z_base_onelev_setag end subroutine amg_z_base_onelev_setag
end interface end interface
interface interface
subroutine amg_z_base_onelev_cseti(lv,what,val,info,pos,idx) subroutine amg_z_base_onelev_cseti(lv,what,val,info,pos,idx)
import :: psb_zspmat_type, psb_z_vect_type, psb_z_base_vect_type, & import :: psb_zspmat_type, psb_z_vect_type, psb_z_base_vect_type, &
& psb_zlinmap_type, psb_dpk_, amg_z_onelev_type, & & psb_zlinmap_type, psb_dpk_, amg_z_onelev_type, &
& psb_ipk_, psb_epk_, psb_desc_type & psb_ipk_, psb_epk_, psb_desc_type
Implicit None Implicit None
! Arguments ! Arguments
class(amg_z_onelev_type), intent(inout) :: lv class(amg_z_onelev_type), intent(inout) :: lv
character(len=*), intent(in) :: what character(len=*), intent(in) :: what
integer(psb_ipk_), intent(in) :: val integer(psb_ipk_), intent(in) :: val
integer(psb_ipk_), intent(out) :: info integer(psb_ipk_), intent(out) :: info
character(len=*), optional, intent(in) :: pos character(len=*), optional, intent(in) :: pos
integer(psb_ipk_), intent(in), optional :: idx integer(psb_ipk_), intent(in), optional :: idx
end subroutine amg_z_base_onelev_cseti end subroutine amg_z_base_onelev_cseti
end interface end interface
interface interface
subroutine amg_z_base_onelev_csetc(lv,what,val,info,pos,idx) subroutine amg_z_base_onelev_csetc(lv,what,val,info,pos,idx)
import :: psb_zspmat_type, psb_z_vect_type, psb_z_base_vect_type, & import :: psb_zspmat_type, psb_z_vect_type, psb_z_base_vect_type, &
& psb_zlinmap_type, psb_dpk_, amg_z_onelev_type, & & psb_zlinmap_type, psb_dpk_, amg_z_onelev_type, &
& psb_ipk_, psb_epk_, psb_desc_type & psb_ipk_, psb_epk_, psb_desc_type
Implicit None Implicit None
! Arguments ! Arguments
class(amg_z_onelev_type), intent(inout) :: lv class(amg_z_onelev_type), intent(inout) :: lv
character(len=*), intent(in) :: what character(len=*), intent(in) :: what
character(len=*), intent(in) :: val character(len=*), intent(in) :: val
integer(psb_ipk_), intent(out) :: info integer(psb_ipk_), intent(out) :: info
character(len=*), optional, intent(in) :: pos character(len=*), optional, intent(in) :: pos
integer(psb_ipk_), intent(in), optional :: idx integer(psb_ipk_), intent(in), optional :: idx
end subroutine amg_z_base_onelev_csetc end subroutine amg_z_base_onelev_csetc
end interface end interface
interface interface
subroutine amg_z_base_onelev_csetr(lv,what,val,info,pos,idx) subroutine amg_z_base_onelev_csetr(lv,what,val,info,pos,idx)
import :: psb_zspmat_type, psb_z_vect_type, psb_z_base_vect_type, & import :: psb_zspmat_type, psb_z_vect_type, psb_z_base_vect_type, &
& psb_zlinmap_type, psb_dpk_, amg_z_onelev_type, & & psb_zlinmap_type, psb_dpk_, amg_z_onelev_type, &
& psb_ipk_, psb_epk_, psb_desc_type & psb_ipk_, psb_epk_, psb_desc_type
Implicit None Implicit None
class(amg_z_onelev_type), intent(inout) :: lv class(amg_z_onelev_type), intent(inout) :: lv
character(len=*), intent(in) :: what character(len=*), intent(in) :: what
real(psb_dpk_), intent(in) :: val real(psb_dpk_), intent(in) :: val
integer(psb_ipk_), intent(out) :: info integer(psb_ipk_), intent(out) :: info
character(len=*), optional, intent(in) :: pos character(len=*), optional, intent(in) :: pos
@ -398,13 +399,13 @@ interface
end subroutine amg_z_base_onelev_csetr end subroutine amg_z_base_onelev_csetr
end interface end interface
interface interface
subroutine amg_z_base_onelev_dump(lv,level,info,prefix,head,ac,rp,smoother,& subroutine amg_z_base_onelev_dump(lv,level,info,prefix,head,ac,rp,smoother,&
& solver,tprol,global_num) & solver,tprol,global_num)
import :: psb_zspmat_type, psb_z_vect_type, psb_z_base_vect_type, & import :: psb_zspmat_type, psb_z_vect_type, psb_z_base_vect_type, &
& psb_zlinmap_type, psb_dpk_, amg_z_onelev_type, & & psb_zlinmap_type, psb_dpk_, amg_z_onelev_type, &
& psb_ipk_, psb_epk_, psb_desc_type & psb_ipk_, psb_epk_, psb_desc_type
implicit none implicit none
class(amg_z_onelev_type), intent(in) :: lv class(amg_z_onelev_type), intent(in) :: lv
integer(psb_ipk_), intent(in) :: level integer(psb_ipk_), intent(in) :: level
integer(psb_ipk_), intent(out) :: info integer(psb_ipk_), intent(out) :: info
@ -436,7 +437,7 @@ interface
end subroutine amg_z_base_onelev_map_rstr_v end subroutine amg_z_base_onelev_map_rstr_v
end interface end interface
interface interface
subroutine amg_z_base_onelev_map_prol_a(lv,alpha,v,beta,u,info,work) subroutine amg_z_base_onelev_map_prol_a(lv,alpha,v,beta,u,info,work)
import import
implicit none implicit none
@ -459,15 +460,15 @@ interface
type(psb_z_vect_type), optional, target, intent(inout) :: vtx,vty type(psb_z_vect_type), optional, target, intent(inout) :: vtx,vty
end subroutine amg_z_base_onelev_map_prol_v end subroutine amg_z_base_onelev_map_prol_v
end interface end interface
contains contains
! !
! Function returning the size of the amg_prec_type data structure ! Function returning the size of the amg_prec_type data structure
! in bytes or in number of nonzeros of the operator(s) involved. ! in bytes or in number of nonzeros of the operator(s) involved.
! !
function z_base_onelev_get_nzeros(lv) result(val) function z_base_onelev_get_nzeros(lv) result(val)
implicit none implicit none
class(amg_z_onelev_type), intent(in) :: lv class(amg_z_onelev_type), intent(in) :: lv
integer(psb_epk_) :: val integer(psb_epk_) :: val
integer(psb_ipk_) :: i integer(psb_ipk_) :: i
@ -479,16 +480,16 @@ contains
end function z_base_onelev_get_nzeros end function z_base_onelev_get_nzeros
function z_base_onelev_sizeof(lv) result(val) function z_base_onelev_sizeof(lv) result(val)
implicit none implicit none
class(amg_z_onelev_type), intent(in) :: lv class(amg_z_onelev_type), intent(in) :: lv
integer(psb_epk_) :: val integer(psb_epk_) :: val
integer(psb_ipk_) :: i integer(psb_ipk_) :: i
val = psb_sizeof_ip+psb_sizeof_lp val = psb_sizeof_ip+psb_sizeof_lp
val = val + lv%desc_ac%sizeof() val = val + lv%desc_ac%sizeof()
val = val + lv%ac%sizeof() val = val + lv%ac%sizeof()
val = val + lv%tprol%sizeof() val = val + lv%tprol%sizeof()
val = val + lv%linmap%sizeof() val = val + lv%linmap%sizeof()
if (allocated(lv%sm)) val = val + lv%sm%sizeof() if (allocated(lv%sm)) val = val + lv%sm%sizeof()
if (allocated(lv%sm2a)) val = val + lv%sm2a%sizeof() if (allocated(lv%sm2a)) val = val + lv%sm2a%sizeof()
if (allocated(lv%aggr)) val = val + lv%aggr%sizeof() if (allocated(lv%aggr)) val = val + lv%aggr%sizeof()
@ -497,19 +498,19 @@ contains
subroutine z_base_onelev_nullify(lv) subroutine z_base_onelev_nullify(lv)
implicit none implicit none
class(amg_z_onelev_type), intent(inout) :: lv class(amg_z_onelev_type), intent(inout) :: lv
nullify(lv%base_a) nullify(lv%base_a)
nullify(lv%base_desc) nullify(lv%base_desc)
nullify(lv%sm2) nullify(lv%sm2)
end subroutine z_base_onelev_nullify end subroutine z_base_onelev_nullify
! !
! Multilevel defaults: ! Multilevel defaults:
! multiplicative vs. additive ML framework; ! multiplicative vs. additive ML framework;
! Smoothed decoupled aggregation with zero threshold; ! Smoothed decoupled aggregation with zero threshold;
! distributed coarse matrix; ! distributed coarse matrix;
! damping omega computed with the max-norm estimate of the ! damping omega computed with the max-norm estimate of the
! dominant eigenvalue; ! dominant eigenvalue;
@ -519,10 +520,10 @@ contains
subroutine z_base_onelev_default(lv) subroutine z_base_onelev_default(lv)
Implicit None Implicit None
! Arguments ! Arguments
class(amg_z_onelev_type), target, intent(inout) :: lv class(amg_z_onelev_type), target, intent(inout) :: lv
integer(psb_ipk_) :: info integer(psb_ipk_) :: info
lv%parms%sweeps_pre = 1 lv%parms%sweeps_pre = 1
lv%parms%sweeps_post = 1 lv%parms%sweeps_post = 1
@ -537,7 +538,7 @@ contains
lv%parms%aggr_filter = amg_no_filter_mat_ lv%parms%aggr_filter = amg_no_filter_mat_
lv%parms%aggr_omega_val = dzero lv%parms%aggr_omega_val = dzero
lv%parms%aggr_thresh = 0.01_psb_dpk_ lv%parms%aggr_thresh = 0.01_psb_dpk_
if (allocated(lv%sm)) call lv%sm%default() if (allocated(lv%sm)) call lv%sm%default()
if (allocated(lv%sm2a)) then if (allocated(lv%sm2a)) then
call lv%sm2a%default() call lv%sm2a%default()
@ -547,7 +548,7 @@ contains
end if end if
if (.not.allocated(lv%aggr)) allocate(amg_z_dec_aggregator_type :: lv%aggr,stat=info) if (.not.allocated(lv%aggr)) allocate(amg_z_dec_aggregator_type :: lv%aggr,stat=info)
if (allocated(lv%aggr)) call lv%aggr%default() if (allocated(lv%aggr)) call lv%aggr%default()
return return
end subroutine z_base_onelev_default end subroutine z_base_onelev_default
@ -562,9 +563,9 @@ contains
type(psb_lzspmat_type), intent(out) :: t_prol type(psb_lzspmat_type), intent(out) :: t_prol
type(amg_daggr_data), intent(in) :: ag_data type(amg_daggr_data), intent(in) :: ag_data
integer(psb_ipk_), intent(out) :: info integer(psb_ipk_), intent(out) :: info
call lv%aggr%bld_tprol(lv%parms,ag_data,a,desc_a,ilaggr,nlaggr,t_prol,info) call lv%aggr%bld_tprol(lv%parms,ag_data,a,desc_a,ilaggr,nlaggr,t_prol,info)
end subroutine z_base_onelev_bld_tprol end subroutine z_base_onelev_bld_tprol
@ -574,7 +575,7 @@ contains
integer(psb_ipk_), intent(out) :: info integer(psb_ipk_), intent(out) :: info
call lv%aggr%update_next(lvnext%aggr,info) call lv%aggr%update_next(lvnext%aggr,info)
end subroutine z_base_onelev_update_aggr end subroutine z_base_onelev_update_aggr
@ -583,33 +584,33 @@ contains
Implicit None Implicit None
! Arguments ! Arguments
class(amg_z_onelev_type), target, intent(inout) :: lv class(amg_z_onelev_type), target, intent(inout) :: lv
class(amg_z_onelev_type), target, intent(inout) :: lvout class(amg_z_onelev_type), target, intent(inout) :: lvout
integer(psb_ipk_), intent(out) :: info integer(psb_ipk_), intent(out) :: info
info = psb_success_ info = psb_success_
if (allocated(lv%sm)) then if (allocated(lv%sm)) then
call lv%sm%clone(lvout%sm,info) call lv%sm%clone(lvout%sm,info)
else else
if (allocated(lvout%sm)) then if (allocated(lvout%sm)) then
call lvout%sm%free(info) call lvout%sm%free(info)
if (info==psb_success_) deallocate(lvout%sm,stat=info) if (info==psb_success_) deallocate(lvout%sm,stat=info)
end if end if
end if end if
if (allocated(lv%sm2a)) then if (allocated(lv%sm2a)) then
call lv%sm%clone(lvout%sm2a,info) call lv%sm%clone(lvout%sm2a,info)
lvout%sm2 => lvout%sm2a lvout%sm2 => lvout%sm2a
else else
if (allocated(lvout%sm2a)) then if (allocated(lvout%sm2a)) then
call lvout%sm2a%free(info) call lvout%sm2a%free(info)
if (info==psb_success_) deallocate(lvout%sm2a,stat=info) if (info==psb_success_) deallocate(lvout%sm2a,stat=info)
end if end if
lvout%sm2 => lvout%sm lvout%sm2 => lvout%sm
end if end if
if (allocated(lv%aggr)) then if (allocated(lv%aggr)) then
call lv%aggr%clone(lvout%aggr,info) call lv%aggr%clone(lvout%aggr,info)
else else
if (allocated(lvout%aggr)) then if (allocated(lvout%aggr)) then
call lvout%aggr%free(info) call lvout%aggr%free(info)
if (info==psb_success_) deallocate(lvout%aggr,stat=info) if (info==psb_success_) deallocate(lvout%aggr,stat=info)
end if end if
@ -622,7 +623,7 @@ contains
if (info == psb_success_) call lv%remap_data%clone(lvout%remap_data,info) if (info == psb_success_) call lv%remap_data%clone(lvout%remap_data,info)
lvout%base_a => lv%base_a lvout%base_a => lv%base_a
lvout%base_desc => lv%base_desc lvout%base_desc => lv%base_desc
return return
end subroutine z_base_onelev_clone end subroutine z_base_onelev_clone
@ -631,12 +632,12 @@ contains
use psb_base_mod use psb_base_mod
implicit none implicit none
class(amg_z_onelev_type), target, intent(inout) :: lv, b class(amg_z_onelev_type), target, intent(inout) :: lv, b
integer(psb_ipk_), intent(out) :: info integer(psb_ipk_), intent(out) :: info
call b%free(info) call b%free(info)
b%parms = lv%parms b%parms = lv%parms
b%szratio = lv%szratio b%szratio = lv%szratio
if (associated(lv%sm2,lv%sm2a)) then if (associated(lv%sm2,lv%sm2a)) then
call move_alloc(lv%sm,b%sm) call move_alloc(lv%sm,b%sm)
call move_alloc(lv%sm2a,b%sm2a) call move_alloc(lv%sm2a,b%sm2a)
b%sm2 =>b%sm2a b%sm2 =>b%sm2a
@ -647,18 +648,18 @@ contains
end if end if
call move_alloc(lv%aggr,b%aggr) call move_alloc(lv%aggr,b%aggr)
if (info == psb_success_) call psb_move_alloc(lv%ac,b%ac,info) if (info == psb_success_) call psb_move_alloc(lv%ac,b%ac,info)
if (info == psb_success_) call psb_move_alloc(lv%tprol,b%tprol,info) if (info == psb_success_) call psb_move_alloc(lv%tprol,b%tprol,info)
if (info == psb_success_) call psb_move_alloc(lv%desc_ac,b%desc_ac,info) if (info == psb_success_) call psb_move_alloc(lv%desc_ac,b%desc_ac,info)
if (info == psb_success_) call psb_move_alloc(lv%linmap,b%linmap,info) if (info == psb_success_) call psb_move_alloc(lv%linmap,b%linmap,info)
b%base_a => lv%base_a b%base_a => lv%base_a
b%base_desc => lv%base_desc b%base_desc => lv%base_desc
end subroutine z_base_onelev_move_alloc end subroutine z_base_onelev_move_alloc
function z_base_onelev_get_wrksize(lv) result(val) function z_base_onelev_get_wrksize(lv) result(val)
implicit none implicit none
class(amg_z_onelev_type), intent(inout) :: lv class(amg_z_onelev_type), intent(inout) :: lv
integer(psb_ipk_) :: val integer(psb_ipk_) :: val
@ -679,26 +680,26 @@ contains
select case(lv%parms%ml_cycle) select case(lv%parms%ml_cycle)
case(amg_add_ml_,amg_mult_ml_,amg_vcycle_ml_, amg_wcycle_ml_) case(amg_add_ml_,amg_mult_ml_,amg_vcycle_ml_, amg_wcycle_ml_)
! We're good ! We're good
case(amg_kcycle_ml_, amg_kcyclesym_ml_) case(amg_kcycle_ml_, amg_kcyclesym_ml_)
! !
! We need 7 in inneritkcycle. ! We need 7 in inneritkcycle.
! Can we reuse vtx? ! Can we reuse vtx?
! !
val = val + 7 val = val + 7
case default case default
! Need a better error signaling ? ! Need a better error signaling ?
val = -1 val = -1
end select end select
end function z_base_onelev_get_wrksize end function z_base_onelev_get_wrksize
subroutine z_base_onelev_allocate_wrk(lv,info,vmold) subroutine z_base_onelev_allocate_wrk(lv,info,vmold)
use psb_base_mod use psb_base_mod
implicit none implicit none
class(amg_z_onelev_type), target, intent(inout) :: lv class(amg_z_onelev_type), target, intent(inout) :: lv
integer(psb_ipk_), intent(out) :: info integer(psb_ipk_), intent(out) :: info
class(psb_z_base_vect_type), intent(in), optional :: vmold class(psb_z_base_vect_type), intent(in), optional :: vmold
! !
integer(psb_ipk_) :: nwv, i integer(psb_ipk_) :: nwv, i
@ -711,22 +712,22 @@ contains
! Need to fix this, we need two different allocations ! Need to fix this, we need two different allocations
! !
call lv%wrk%alloc(nwv,lv%base_desc,info,vmold=vmold,& call lv%wrk%alloc(nwv,lv%base_desc,info,vmold=vmold,&
& desc2=lv%remap_data%desc_ac_pre_remap) & desc2=lv%remap_data%desc_ac_pre_remap)
else else
call lv%wrk%alloc(nwv,lv%base_desc,info,vmold=vmold) call lv%wrk%alloc(nwv,lv%base_desc,info,vmold=vmold)
end if end if
end if end if
end subroutine z_base_onelev_allocate_wrk end subroutine z_base_onelev_allocate_wrk
subroutine z_base_onelev_free_wrk(lv,info) subroutine z_base_onelev_free_wrk(lv,info)
use psb_base_mod use psb_base_mod
implicit none implicit none
class(amg_z_onelev_type), target, intent(inout) :: lv class(amg_z_onelev_type), target, intent(inout) :: lv
integer(psb_ipk_), intent(out) :: info integer(psb_ipk_), intent(out) :: info
! !
integer(psb_ipk_) :: nwv,i integer(psb_ipk_) :: nwv,i
info = psb_success_ info = psb_success_
if (allocated(lv%wrk)) then if (allocated(lv%wrk)) then
@ -734,17 +735,17 @@ contains
if (info == 0) deallocate(lv%wrk,stat=info) if (info == 0) deallocate(lv%wrk,stat=info)
end if end if
end subroutine z_base_onelev_free_wrk end subroutine z_base_onelev_free_wrk
subroutine z_wrk_alloc(wk,nwv,desc,info,vmold, desc2) subroutine z_wrk_alloc(wk,nwv,desc,info,vmold, desc2)
use psb_base_mod use psb_base_mod
Implicit None Implicit None
! Arguments ! Arguments
class(amg_zmlprec_wrk_type), target, intent(inout) :: wk class(amg_zmlprec_wrk_type), target, intent(inout) :: wk
integer(psb_ipk_), intent(in) :: nwv integer(psb_ipk_), intent(in) :: nwv
type(psb_desc_type), intent(in) :: desc type(psb_desc_type), intent(in) :: desc
integer(psb_ipk_), intent(out) :: info integer(psb_ipk_), intent(out) :: info
class(psb_z_base_vect_type), intent(in), optional :: vmold class(psb_z_base_vect_type), intent(in), optional :: vmold
type(psb_desc_type), intent(in), optional :: desc2 type(psb_desc_type), intent(in), optional :: desc2
! !
@ -808,14 +809,14 @@ contains
end do end do
end if end if
end subroutine z_wrk_alloc end subroutine z_wrk_alloc
subroutine z_wrk_free(wk,info) subroutine z_wrk_free(wk,info)
Implicit None Implicit None
! Arguments ! Arguments
class(amg_zmlprec_wrk_type), target, intent(inout) :: wk class(amg_zmlprec_wrk_type), target, intent(inout) :: wk
integer(psb_ipk_), intent(out) :: info integer(psb_ipk_), intent(out) :: info
! !
integer(psb_ipk_) :: i integer(psb_ipk_) :: i
info = psb_success_ info = psb_success_
@ -836,7 +837,7 @@ contains
end if end if
end subroutine z_wrk_free end subroutine z_wrk_free
subroutine z_wrk_clone(wk,wkout,info) subroutine z_wrk_clone(wk,wkout,info)
use psb_base_mod use psb_base_mod
Implicit None Implicit None
@ -844,11 +845,11 @@ contains
! Arguments ! Arguments
class(amg_zmlprec_wrk_type), target, intent(inout) :: wk class(amg_zmlprec_wrk_type), target, intent(inout) :: wk
class(amg_zmlprec_wrk_type), target, intent(inout) :: wkout class(amg_zmlprec_wrk_type), target, intent(inout) :: wkout
integer(psb_ipk_), intent(out) :: info integer(psb_ipk_), intent(out) :: info
! !
integer(psb_ipk_) :: i integer(psb_ipk_) :: i
info = psb_success_ info = psb_success_
call psb_safe_ab_cpy(wk%tx,wkout%tx,info) call psb_safe_ab_cpy(wk%tx,wkout%tx,info)
call psb_safe_ab_cpy(wk%ty,wkout%ty,info) call psb_safe_ab_cpy(wk%ty,wkout%ty,info)
call psb_safe_ab_cpy(wk%x2l,wkout%x2l,info) call psb_safe_ab_cpy(wk%x2l,wkout%x2l,info)
@ -870,12 +871,12 @@ contains
return return
end subroutine z_wrk_clone end subroutine z_wrk_clone
subroutine z_wrk_move_alloc(wk, b,info) subroutine z_wrk_move_alloc(wk, b,info)
implicit none implicit none
class(amg_zmlprec_wrk_type), target, intent(inout) :: wk, b class(amg_zmlprec_wrk_type), target, intent(inout) :: wk, b
integer(psb_ipk_), intent(out) :: info integer(psb_ipk_), intent(out) :: info
call b%free(info) call b%free(info)
call move_alloc(wk%tx,b%tx) call move_alloc(wk%tx,b%tx)
call move_alloc(wk%ty,b%ty) call move_alloc(wk%ty,b%ty)
@ -888,17 +889,17 @@ contains
call move_alloc(wk%vx2l%v,b%vx2l%v) call move_alloc(wk%vx2l%v,b%vx2l%v)
call move_alloc(wk%vy2l%v,b%vy2l%v) call move_alloc(wk%vy2l%v,b%vy2l%v)
call move_alloc(wk%wv,b%wv) call move_alloc(wk%wv,b%wv)
end subroutine z_wrk_move_alloc end subroutine z_wrk_move_alloc
subroutine z_wrk_cnv(wk,info,vmold) subroutine z_wrk_cnv(wk,info,vmold)
use psb_base_mod use psb_base_mod
Implicit None Implicit None
! Arguments ! Arguments
class(amg_zmlprec_wrk_type), target, intent(inout) :: wk class(amg_zmlprec_wrk_type), target, intent(inout) :: wk
integer(psb_ipk_), intent(out) :: info integer(psb_ipk_), intent(out) :: info
class(psb_z_base_vect_type), intent(in), optional :: vmold class(psb_z_base_vect_type), intent(in), optional :: vmold
! !
integer(psb_ipk_) :: i integer(psb_ipk_) :: i
@ -919,7 +920,7 @@ contains
function z_wrk_sizeof(wk) result(val) function z_wrk_sizeof(wk) result(val)
use psb_realloc_mod use psb_realloc_mod
implicit none implicit none
class(amg_zmlprec_wrk_type), intent(in) :: wk class(amg_zmlprec_wrk_type), intent(in) :: wk
integer(psb_epk_) :: val integer(psb_epk_) :: val
integer :: i integer :: i
@ -938,14 +939,14 @@ contains
end do end do
end if end if
end function z_wrk_sizeof end function z_wrk_sizeof
subroutine z_remap_data_clone(rmp, remap_out, info) subroutine z_remap_data_clone(rmp, remap_out, info)
use psb_base_mod use psb_base_mod
implicit none implicit none
! Arguments ! Arguments
class(amg_z_remap_data_type), target, intent(inout) :: rmp class(amg_z_remap_data_type), target, intent(inout) :: rmp
class(amg_z_remap_data_type), target, intent(inout) :: remap_out class(amg_z_remap_data_type), target, intent(inout) :: remap_out
integer(psb_ipk_), intent(out) :: info integer(psb_ipk_), intent(out) :: info
! !
integer(psb_ipk_) :: i integer(psb_ipk_) :: i
@ -956,7 +957,7 @@ contains
& call rmp%desc_ac_pre_remap%clone(remap_out%desc_ac_pre_remap,info) & call rmp%desc_ac_pre_remap%clone(remap_out%desc_ac_pre_remap,info)
remap_out%idest = rmp%idest remap_out%idest = rmp%idest
call psb_safe_ab_cpy(rmp%isrc,remap_out%isrc,info) call psb_safe_ab_cpy(rmp%isrc,remap_out%isrc,info)
call psb_safe_ab_cpy(rmp%nrsrc,remap_out%nrsrc,info) call psb_safe_ab_cpy(rmp%nrsrc,remap_out%nrsrc,info)
end subroutine z_remap_data_clone end subroutine z_remap_data_clone
end module amg_z_onelev_mod end module amg_z_onelev_mod

7
amgprec/impl/aggregator/Makefile
Unescape Escape View File

@ -9,7 +9,7 @@ CXXINCLUDES=$(FMFLAG)$(HERE) $(FMFLAG)$(INCDIR) $(FMFLAG)/.
#CINCLUDES= -I${SUPERLU_INCDIR} -I${HSL_INCDIR} -I${SPRAL_INCDIR} -I/home/users/pasqua/Ambra/BootCMatch/include -lBCM -L/home/users/pasqua/Ambra/BootCMatch/lib -lm #CINCLUDES= -I${SUPERLU_INCDIR} -I${HSL_INCDIR} -I${SPRAL_INCDIR} -I/home/users/pasqua/Ambra/BootCMatch/include -lBCM -L/home/users/pasqua/Ambra/BootCMatch/lib -lm
OBJS= \ FOBJS= \
amg_s_dec_aggregator_mat_asb.o \ amg_s_dec_aggregator_mat_asb.o \
amg_s_dec_aggregator_mat_bld.o \ amg_s_dec_aggregator_mat_bld.o \
amg_s_dec_aggregator_tprol.o \ amg_s_dec_aggregator_tprol.o \
@ -64,11 +64,12 @@ amg_s_parmatch_spmm_bld_inner.o
MPCOBJS=MatchBoxPC.o \ MPCOBJS=MatchBoxPC.o \
algoDistEdgeApproxDomEdgesLinearSearchMesgBndlSmallMateC.o algoDistEdgeApproxDomEdgesLinearSearchMesgBndlSmallMateC.o
OBJS = $(FOBJS) $(MPCOBJS)
LIBNAME=libamg_prec.a LIBNAME=libamg_prec.a
lib: $(OBJS) $(MPCOBJS) lib: $(OBJS)
$(AR) $(HERE)/$(LIBNAME) $(OBJS) $(MPCOBJS) $(AR) $(HERE)/$(LIBNAME) $(OBJS)
$(RANLIB) $(HERE)/$(LIBNAME) $(RANLIB) $(HERE)/$(LIBNAME)
mpobjs: mpobjs:

Write Preview
Loading…
Cancel
Save