New global option for sizeof()

This commit is contained in:
Salvatore Filippone
2021-03-15 15:24:22 +01:00
parent c51414f7ab
commit 1159659b4f
4 changed files with 72 additions and 8 deletions
+18 -2
View File
@@ -425,11 +425,22 @@ contains
end if
end function amg_c_get_nzeros
function amg_cprec_sizeof(prec) result(val)
function amg_cprec_sizeof(prec, global) result(val)
implicit none
class(amg_cprec_type), intent(in) :: prec
integer(psb_epk_) :: val
logical, intent(in), optional :: global
integer(psb_epk_) :: val
integer(psb_ipk_) :: i
type(psb_ctxt_type) :: ctxt
logical :: global_
if (present(global)) then
global_ = global
else
global_ = .false.
end if
val = 0
val = val + psb_sizeof_ip
if (allocated(prec%precv)) then
@@ -437,6 +448,11 @@ contains
val = val + prec%precv(i)%sizeof()
end do
end if
if (global_) then
ctxt = prec%ctxt
call psb_sum(ctxt,val)
end if
end function amg_cprec_sizeof
!
+18 -2
View File
@@ -425,11 +425,22 @@ contains
end if
end function amg_d_get_nzeros
function amg_dprec_sizeof(prec) result(val)
function amg_dprec_sizeof(prec, global) result(val)
implicit none
class(amg_dprec_type), intent(in) :: prec
integer(psb_epk_) :: val
logical, intent(in), optional :: global
integer(psb_epk_) :: val
integer(psb_ipk_) :: i
type(psb_ctxt_type) :: ctxt
logical :: global_
if (present(global)) then
global_ = global
else
global_ = .false.
end if
val = 0
val = val + psb_sizeof_ip
if (allocated(prec%precv)) then
@@ -437,6 +448,11 @@ contains
val = val + prec%precv(i)%sizeof()
end do
end if
if (global_) then
ctxt = prec%ctxt
call psb_sum(ctxt,val)
end if
end function amg_dprec_sizeof
!
+18 -2
View File
@@ -425,11 +425,22 @@ contains
end if
end function amg_s_get_nzeros
function amg_sprec_sizeof(prec) result(val)
function amg_sprec_sizeof(prec, global) result(val)
implicit none
class(amg_sprec_type), intent(in) :: prec
integer(psb_epk_) :: val
logical, intent(in), optional :: global
integer(psb_epk_) :: val
integer(psb_ipk_) :: i
type(psb_ctxt_type) :: ctxt
logical :: global_
if (present(global)) then
global_ = global
else
global_ = .false.
end if
val = 0
val = val + psb_sizeof_ip
if (allocated(prec%precv)) then
@@ -437,6 +448,11 @@ contains
val = val + prec%precv(i)%sizeof()
end do
end if
if (global_) then
ctxt = prec%ctxt
call psb_sum(ctxt,val)
end if
end function amg_sprec_sizeof
!
+18 -2
View File
@@ -425,11 +425,22 @@ contains
end if
end function amg_z_get_nzeros
function amg_zprec_sizeof(prec) result(val)
function amg_zprec_sizeof(prec, global) result(val)
implicit none
class(amg_zprec_type), intent(in) :: prec
integer(psb_epk_) :: val
logical, intent(in), optional :: global
integer(psb_epk_) :: val
integer(psb_ipk_) :: i
type(psb_ctxt_type) :: ctxt
logical :: global_
if (present(global)) then
global_ = global
else
global_ = .false.
end if
val = 0
val = val + psb_sizeof_ip
if (allocated(prec%precv)) then
@@ -437,6 +448,11 @@ contains
val = val + prec%precv(i)%sizeof()
end do
end if
if (global_) then
ctxt = prec%ctxt
call psb_sum(ctxt,val)
end if
end function amg_zprec_sizeof
!