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

mld2p4-2:

Browse Source 
examples/pdegen/mld_dexample_1lev.f90
 examples/pdegen/mld_dexample_ml.f90
 examples/pdegen/mld_sexample_1lev.f90
 examples/pdegen/mld_sexample_ml.f90
 tests/newslv/ppde.f90
 tests/newslv/spde.f90
 tests/pdegen/ppde.f90
 tests/pdegen/spde.f90

Fixed examples.
stopcriterion
Salvatore Filippone 13 years ago
parent 79cb0731c4
commit 7c555b3352
8 changed files with 315 additions and 454 deletions
Show Stats Download Patch File Download Diff File

150
examples/pdegen/mld_dexample_1lev.f90
Unescape Escape View File

@ -288,25 +288,25 @@ contains
real(psb_dpk_), allocatable :: b(:),xv(:) real(psb_dpk_), allocatable :: b(:),xv(:)
type(psb_desc_type) :: desc_a type(psb_desc_type) :: desc_a
integer :: ictxt, info integer :: ictxt, info
! local variables character :: afmt*5
type(psb_dspmat_type) :: a type(psb_dspmat_type) :: a
real(psb_dpk_) :: zt(nb),glob_x,glob_y,glob_z real(psb_dpk_) :: zt(nb),x,y,z
integer :: m,n,nnz,glob_row,nlr,i,ii,ib,k,ipoints integer :: m,n,nnz,glob_row,nlr,i,ii,ib,k
integer :: x,y,z,ia,indx_owner integer :: ix,iy,iz,ia,indx_owner, ipoints
integer :: np, iam, nr, nt integer :: np, iam, nr, nt
integer :: element integer :: element
integer, allocatable :: irow(:),icol(:),myidx(:) integer, allocatable :: irow(:),icol(:),myidx(:)
real(psb_dpk_), allocatable :: val(:) real(psb_dpk_), allocatable :: val(:)
! deltah dimension of each grid cell ! deltah dimension of each grid cell
! deltat discretization time ! deltat discretization time
real(psb_dpk_) :: deltah real(psb_dpk_) :: deltah, deltah2
real(psb_dpk_),parameter :: rhs=0.d0,one=1.d0,zero=0.d0 real(psb_dpk_),parameter :: rhs=0.d0,one=1.d0,zero=0.d0
real(psb_dpk_) :: t0, t1, t2, t3, tasb, talc, ttot, tgen real(psb_dpk_) :: t0, t1, t2, t3, tasb, talc, ttot, tgen
real(psb_dpk_) :: a1, a2, a3, a4, b1, b2, b3 real(psb_dpk_) :: a1, a2, a3, a4, b1, b2, b3
external :: a1, a2, a3, a4, b1, b2, b3 external :: a1, a2, a3, a4, b1, b2, b3
integer :: err_act integer :: err_act
character(len=20) :: name character(len=20) :: name, ch_err
info = psb_success_ info = psb_success_
name = 'create_matrix' name = 'create_matrix'
@ -314,16 +314,17 @@ contains
call psb_info(ictxt, iam, np) call psb_info(ictxt, iam, np)
deltah = 1.d0/(idim-1) deltah = 1.d0/(idim-1)
deltah2 = deltah*deltah
! initialize array descriptor and sparse matrix storage; provide an ! initialize array descriptor and sparse matrix storage. provide an
! estimate of the number of non zeroes ! estimate of the number of non zeroes
ipoints=idim-2 ipoints=idim-2
m = ipoints*ipoints*ipoints m = ipoints*ipoints*ipoints
n = m n = m
nnz = ((n*9)/(np)) nnz = ((n*9)/(np))
if(iam == psb_root_) write(0,'("Generating Matrix (size=",i0,")...")')n if(iam == psb_root_) write(psb_out_unit,'("Generating Matrix (size=",i0,")...")')n
! !
! Using a simple BLOCK distribution. ! Using a simple BLOCK distribution.
@ -333,7 +334,7 @@ contains
nt = nr nt = nr
call psb_sum(ictxt,nt) call psb_sum(ictxt,nt)
if (nt /= m) write(0,*) iam, 'Initialization error ',nr,nt,m if (nt /= m) write(psb_err_unit,*) iam, 'Initialization error ',nr,nt,m
call psb_barrier(ictxt) call psb_barrier(ictxt)
t0 = psb_wtime() t0 = psb_wtime()
call psb_cdall(ictxt,desc_a,info,nl=nr) call psb_cdall(ictxt,desc_a,info,nl=nr)
@ -347,7 +348,8 @@ contains
if (info /= psb_success_) then if (info /= psb_success_) then
info=psb_err_from_subroutine_ info=psb_err_from_subroutine_
call psb_errpush(info,name) ch_err='allocation rout.'
call psb_errpush(info,name,a_err=ch_err)
goto 9999 goto 9999
end if end if
@ -384,20 +386,20 @@ contains
glob_row=myidx(i) glob_row=myidx(i)
! compute gridpoint coordinates ! compute gridpoint coordinates
if (mod(glob_row,ipoints*ipoints) == 0) then if (mod(glob_row,ipoints*ipoints) == 0) then
x = glob_row/(ipoints*ipoints) ix = glob_row/(ipoints*ipoints)
else else
x = glob_row/(ipoints*ipoints)+1 ix = glob_row/(ipoints*ipoints)+1
endif endif
if (mod((glob_row-(x-1)*ipoints*ipoints),ipoints) == 0) then if (mod((glob_row-(ix-1)*ipoints*ipoints),ipoints) == 0) then
y = (glob_row-(x-1)*ipoints*ipoints)/ipoints iy = (glob_row-(ix-1)*ipoints*ipoints)/ipoints
else else
y = (glob_row-(x-1)*ipoints*ipoints)/ipoints+1 iy = (glob_row-(ix-1)*ipoints*ipoints)/ipoints+1
endif endif
z = glob_row-(x-1)*ipoints*ipoints-(y-1)*ipoints iz = glob_row-(ix-1)*ipoints*ipoints-(iy-1)*ipoints
! glob_x, glob_y, glob_x coordinates ! x, y, x coordinates
glob_x=x*deltah x=ix*deltah
glob_y=y*deltah y=iy*deltah
glob_z=z*deltah z=iz*deltah
! check on boundary points ! check on boundary points
zt(k) = 0.d0 zt(k) = 0.d0
@ -405,104 +407,68 @@ contains
! !
! term depending on (x-1,y,z) ! term depending on (x-1,y,z)
! !
if (x == 1) then if (ix == 1) then
val(element)=-b1(glob_x,glob_y,glob_z)& val(element) = -b1(x,y,z)/deltah2-a1(x,y,z)/deltah
& -a1(glob_x,glob_y,glob_z) zt(k) = exp(-x**2-y**2-z**2)*(-val(element))
val(element) = val(element)/(deltah*&
& deltah)
zt(k) = exp(-glob_y**2-glob_z**2)*(-val(element))
else else
val(element)=-b1(glob_x,glob_y,glob_z)& val(element) = -b1(x,y,z)/deltah2-a1(x,y,z)/deltah
& -a1(glob_x,glob_y,glob_z) icol(element) = (ix-2)*ipoints*ipoints+(iy-1)*ipoints+(iz)
val(element) = val(element)/(deltah*&
& deltah)
icol(element) = (x-2)*ipoints*ipoints+(y-1)*ipoints+(z)
irow(element) = glob_row irow(element) = glob_row
element = element+1 element = element+1
endif endif
! term depending on (x,y-1,z) ! term depending on (x,y-1,z)
if (y == 1) then if (iy == 1) then
val(element)=-b2(glob_x,glob_y,glob_z)& val(element) = -b2(x,y,z)/deltah2-a2(x,y,z)/deltah
& -a2(glob_x,glob_y,glob_z) zt(k) = exp(-x**2-y**2-z**2)*exp(-x)*(-val(element))
val(element) = val(element)/(deltah*&
& deltah)
zt(k) = exp(-glob_x**2-glob_z**2)*(-val(element))
else else
val(element)=-b2(glob_x,glob_y,glob_z)& val(element) = -b2(x,y,z)/deltah2-a2(x,y,z)/deltah
& -a2(glob_x,glob_y,glob_z) icol(element) = (ix-1)*ipoints*ipoints+(iy-2)*ipoints+(iz)
val(element) = val(element)/(deltah*&
& deltah)
icol(element) = (x-1)*ipoints*ipoints+(y-2)*ipoints+(z)
irow(element) = glob_row irow(element) = glob_row
element = element+1 element = element+1
endif endif
! term depending on (x,y,z-1) ! term depending on (x,y,z-1)
if (z == 1) then if (iz == 1) then
val(element)=-b3(glob_x,glob_y,glob_z)& val(element)=-b3(x,y,z)/deltah2-a3(x,y,z)/deltah
& -a3(glob_x,glob_y,glob_z) zt(k) = exp(-x**2-y**2-z**2)*exp(-x)*(-val(element))
val(element) = val(element)/(deltah*&
& deltah)
zt(k) = exp(-glob_x**2-glob_y**2)*(-val(element))
else else
val(element)=-b3(glob_x,glob_y,glob_z)& val(element)=-b3(x,y,z)/deltah2-a3(x,y,z)/deltah
& -a3(glob_x,glob_y,glob_z) icol(element) = (ix-1)*ipoints*ipoints+(iy-1)*ipoints+(iz-1)
val(element) = val(element)/(deltah*&
& deltah)
icol(element) = (x-1)*ipoints*ipoints+(y-1)*ipoints+(z-1)
irow(element) = glob_row irow(element) = glob_row
element = element+1 element = element+1
endif endif
! term depending on (x,y,z) ! term depending on (x,y,z)
val(element)=2*b1(glob_x,glob_y,glob_z)& val(element)=(2*b1(x,y,z) + 2*b2(x,y,z) + 2*b3(x,y,z))/deltah2&
& +2*b2(glob_x,glob_y,glob_z)& & + (a1(x,y,z) + a2(x,y,z) + a3(x,y,z)+ a4(x,y,z))/deltah
& +2*b3(glob_x,glob_y,glob_z)& icol(element) = (ix-1)*ipoints*ipoints+(iy-1)*ipoints+(iz)
& +a1(glob_x,glob_y,glob_z)&
& +a2(glob_x,glob_y,glob_z)&
& +a3(glob_x,glob_y,glob_z)
val(element) = val(element)/(deltah*&
& deltah)
icol(element) = (x-1)*ipoints*ipoints+(y-1)*ipoints+(z)
irow(element) = glob_row irow(element) = glob_row
element = element+1 element = element+1
! term depending on (x,y,z+1) ! term depending on (x,y,z+1)
if (z == ipoints) then if (iz == ipoints) then
val(element)=-b1(glob_x,glob_y,glob_z) val(element)=-b1(x,y,z)/deltah2
val(element) = val(element)/(deltah*& zt(k) = exp(-x**2-y**2-z**2)*exp(-x)*(-val(element))
& deltah)
zt(k) = exp(-glob_x**2-glob_y**2)*exp(-glob_z)*(-val(element))
else else
val(element)=-b1(glob_x,glob_y,glob_z) val(element)=-b1(x,y,z)/deltah2
val(element) = val(element)/(deltah*& icol(element) = (ix-1)*ipoints*ipoints+(iy-1)*ipoints+(iz+1)
& deltah)
icol(element) = (x-1)*ipoints*ipoints+(y-1)*ipoints+(z+1)
irow(element) = glob_row irow(element) = glob_row
element = element+1 element = element+1
endif endif
! term depending on (x,y+1,z) ! term depending on (x,y+1,z)
if (y == ipoints) then if (iy == ipoints) then
val(element)=-b2(glob_x,glob_y,glob_z) val(element)=-b2(x,y,z)/deltah2
val(element) = val(element)/(deltah*& zt(k) = exp(-x**2-y**2-z**2)*exp(-x)*(-val(element))
& deltah)
zt(k) = exp(-glob_x**2-glob_z**2)*exp(-glob_y)*(-val(element))
else else
val(element)=-b2(glob_x,glob_y,glob_z) val(element)=-b2(x,y,z)/deltah2
val(element) = val(element)/(deltah*& icol(element) = (ix-1)*ipoints*ipoints+(iy)*ipoints+(iz)
& deltah)
icol(element)=(x-1)*ipoints*ipoints+(y)*ipoints+(z)
irow(element) = glob_row irow(element) = glob_row
element = element+1 element = element+1
endif endif
! term depending on (x+1,y,z) ! term depending on (x+1,y,z)
if (x == ipoints) then if (ix==ipoints) then
val(element)=-b3(glob_x,glob_y,glob_z) val(element)=-b3(x,y,z)/deltah2
val(element) = val(element)/(deltah*& zt(k) = exp(-y**2-z**2)*exp(-x)*(-val(element))
& deltah)
zt(k) = exp(-glob_y**2-glob_z**2)*exp(-glob_x)*(-val(element))
else else
val(element)=-b3(glob_x,glob_y,glob_z) val(element)=-b3(x,y,z)/deltah2
val(element) = val(element)/(deltah*& icol(element) = (ix)*ipoints*ipoints+(iy-1)*ipoints+(iz)
& deltah)
icol(element) = (x)*ipoints*ipoints+(y-1)*ipoints+(z)
irow(element) = glob_row irow(element) = glob_row
element = element+1 element = element+1
endif endif

150
examples/pdegen/mld_dexample_ml.f90
Unescape Escape View File

@ -326,25 +326,25 @@ contains
real(psb_dpk_), allocatable :: b(:),xv(:) real(psb_dpk_), allocatable :: b(:),xv(:)
type(psb_desc_type) :: desc_a type(psb_desc_type) :: desc_a
integer :: ictxt, info integer :: ictxt, info
! local variables character :: afmt*5
type(psb_dspmat_type) :: a type(psb_dspmat_type) :: a
real(psb_dpk_) :: zt(nb),glob_x,glob_y,glob_z real(psb_dpk_) :: zt(nb),x,y,z
integer :: m,n,nnz,glob_row,nlr,i,ii,ib,k,ipoints integer :: m,n,nnz,glob_row,nlr,i,ii,ib,k
integer :: x,y,z,ia,indx_owner integer :: ix,iy,iz,ia,indx_owner, ipoints
integer :: np, iam, nr, nt integer :: np, iam, nr, nt
integer :: element integer :: element
integer, allocatable :: irow(:),icol(:),myidx(:) integer, allocatable :: irow(:),icol(:),myidx(:)
real(psb_dpk_), allocatable :: val(:) real(psb_dpk_), allocatable :: val(:)
! deltah dimension of each grid cell ! deltah dimension of each grid cell
! deltat discretization time ! deltat discretization time
real(psb_dpk_) :: deltah real(psb_dpk_) :: deltah, deltah2
real(psb_dpk_),parameter :: rhs=0.d0,one=1.d0,zero=0.d0 real(psb_dpk_),parameter :: rhs=0.d0,one=1.d0,zero=0.d0
real(psb_dpk_) :: t0, t1, t2, t3, tasb, talc, ttot, tgen real(psb_dpk_) :: t0, t1, t2, t3, tasb, talc, ttot, tgen
real(psb_dpk_) :: a1, a2, a3, a4, b1, b2, b3 real(psb_dpk_) :: a1, a2, a3, a4, b1, b2, b3
external :: a1, a2, a3, a4, b1, b2, b3 external :: a1, a2, a3, a4, b1, b2, b3
integer :: err_act integer :: err_act
character(len=20) :: name character(len=20) :: name, ch_err
info = psb_success_ info = psb_success_
name = 'create_matrix' name = 'create_matrix'
@ -352,16 +352,17 @@ contains
call psb_info(ictxt, iam, np) call psb_info(ictxt, iam, np)
deltah = 1.d0/(idim-1) deltah = 1.d0/(idim-1)
deltah2 = deltah*deltah
! initialize array descriptor and sparse matrix storage; provide an ! initialize array descriptor and sparse matrix storage. provide an
! estimate of the number of non zeroes ! estimate of the number of non zeroes
ipoints=idim-2 ipoints=idim-2
m = ipoints*ipoints*ipoints m = ipoints*ipoints*ipoints
n = m n = m
nnz = ((n*9)/(np)) nnz = ((n*9)/(np))
if(iam == psb_root_) write(0,'("Generating Matrix (size=",i0,")...")')n if(iam == psb_root_) write(psb_out_unit,'("Generating Matrix (size=",i0,")...")')n
! !
! Using a simple BLOCK distribution. ! Using a simple BLOCK distribution.
@ -371,7 +372,7 @@ contains
nt = nr nt = nr
call psb_sum(ictxt,nt) call psb_sum(ictxt,nt)
if (nt /= m) write(0,*) iam, 'Initialization error ',nr,nt,m if (nt /= m) write(psb_err_unit,*) iam, 'Initialization error ',nr,nt,m
call psb_barrier(ictxt) call psb_barrier(ictxt)
t0 = psb_wtime() t0 = psb_wtime()
call psb_cdall(ictxt,desc_a,info,nl=nr) call psb_cdall(ictxt,desc_a,info,nl=nr)
@ -385,7 +386,8 @@ contains
if (info /= psb_success_) then if (info /= psb_success_) then
info=psb_err_from_subroutine_ info=psb_err_from_subroutine_
call psb_errpush(info,name) ch_err='allocation rout.'
call psb_errpush(info,name,a_err=ch_err)
goto 9999 goto 9999
end if end if
@ -422,20 +424,20 @@ contains
glob_row=myidx(i) glob_row=myidx(i)
! compute gridpoint coordinates ! compute gridpoint coordinates
if (mod(glob_row,ipoints*ipoints) == 0) then if (mod(glob_row,ipoints*ipoints) == 0) then
x = glob_row/(ipoints*ipoints) ix = glob_row/(ipoints*ipoints)
else else
x = glob_row/(ipoints*ipoints)+1 ix = glob_row/(ipoints*ipoints)+1
endif endif
if (mod((glob_row-(x-1)*ipoints*ipoints),ipoints) == 0) then if (mod((glob_row-(ix-1)*ipoints*ipoints),ipoints) == 0) then
y = (glob_row-(x-1)*ipoints*ipoints)/ipoints iy = (glob_row-(ix-1)*ipoints*ipoints)/ipoints
else else
y = (glob_row-(x-1)*ipoints*ipoints)/ipoints+1 iy = (glob_row-(ix-1)*ipoints*ipoints)/ipoints+1
endif endif
z = glob_row-(x-1)*ipoints*ipoints-(y-1)*ipoints iz = glob_row-(ix-1)*ipoints*ipoints-(iy-1)*ipoints
! glob_x, glob_y, glob_x coordinates ! x, y, x coordinates
glob_x=x*deltah x=ix*deltah
glob_y=y*deltah y=iy*deltah
glob_z=z*deltah z=iz*deltah
! check on boundary points ! check on boundary points
zt(k) = 0.d0 zt(k) = 0.d0
@ -443,104 +445,68 @@ contains
! !
! term depending on (x-1,y,z) ! term depending on (x-1,y,z)
! !
if (x == 1) then if (ix == 1) then
val(element)=-b1(glob_x,glob_y,glob_z)& val(element) = -b1(x,y,z)/deltah2-a1(x,y,z)/deltah
& -a1(glob_x,glob_y,glob_z) zt(k) = exp(-x**2-y**2-z**2)*(-val(element))
val(element) = val(element)/(deltah*&
& deltah)
zt(k) = exp(-glob_y**2-glob_z**2)*(-val(element))
else else
val(element)=-b1(glob_x,glob_y,glob_z)& val(element) = -b1(x,y,z)/deltah2-a1(x,y,z)/deltah
& -a1(glob_x,glob_y,glob_z) icol(element) = (ix-2)*ipoints*ipoints+(iy-1)*ipoints+(iz)
val(element) = val(element)/(deltah*&
& deltah)
icol(element) = (x-2)*ipoints*ipoints+(y-1)*ipoints+(z)
irow(element) = glob_row irow(element) = glob_row
element = element+1 element = element+1
endif endif
! term depending on (x,y-1,z) ! term depending on (x,y-1,z)
if (y == 1) then if (iy == 1) then
val(element)=-b2(glob_x,glob_y,glob_z)& val(element) = -b2(x,y,z)/deltah2-a2(x,y,z)/deltah
& -a2(glob_x,glob_y,glob_z) zt(k) = exp(-x**2-y**2-z**2)*exp(-x)*(-val(element))
val(element) = val(element)/(deltah*&
& deltah)
zt(k) = exp(-glob_x**2-glob_z**2)*(-val(element))
else else
val(element)=-b2(glob_x,glob_y,glob_z)& val(element) = -b2(x,y,z)/deltah2-a2(x,y,z)/deltah
& -a2(glob_x,glob_y,glob_z) icol(element) = (ix-1)*ipoints*ipoints+(iy-2)*ipoints+(iz)
val(element) = val(element)/(deltah*&
& deltah)
icol(element) = (x-1)*ipoints*ipoints+(y-2)*ipoints+(z)
irow(element) = glob_row irow(element) = glob_row
element = element+1 element = element+1
endif endif
! term depending on (x,y,z-1) ! term depending on (x,y,z-1)
if (z == 1) then if (iz == 1) then
val(element)=-b3(glob_x,glob_y,glob_z)& val(element)=-b3(x,y,z)/deltah2-a3(x,y,z)/deltah
& -a3(glob_x,glob_y,glob_z) zt(k) = exp(-x**2-y**2-z**2)*exp(-x)*(-val(element))
val(element) = val(element)/(deltah*&
& deltah)
zt(k) = exp(-glob_x**2-glob_y**2)*(-val(element))
else else
val(element)=-b3(glob_x,glob_y,glob_z)& val(element)=-b3(x,y,z)/deltah2-a3(x,y,z)/deltah
& -a3(glob_x,glob_y,glob_z) icol(element) = (ix-1)*ipoints*ipoints+(iy-1)*ipoints+(iz-1)
val(element) = val(element)/(deltah*&
& deltah)
icol(element) = (x-1)*ipoints*ipoints+(y-1)*ipoints+(z-1)
irow(element) = glob_row irow(element) = glob_row
element = element+1 element = element+1
endif endif
! term depending on (x,y,z) ! term depending on (x,y,z)
val(element)=2*b1(glob_x,glob_y,glob_z)& val(element)=(2*b1(x,y,z) + 2*b2(x,y,z) + 2*b3(x,y,z))/deltah2&
& +2*b2(glob_x,glob_y,glob_z)& & + (a1(x,y,z) + a2(x,y,z) + a3(x,y,z)+ a4(x,y,z))/deltah
& +2*b3(glob_x,glob_y,glob_z)& icol(element) = (ix-1)*ipoints*ipoints+(iy-1)*ipoints+(iz)
& +a1(glob_x,glob_y,glob_z)&
& +a2(glob_x,glob_y,glob_z)&
& +a3(glob_x,glob_y,glob_z)
val(element) = val(element)/(deltah*&
& deltah)
icol(element) = (x-1)*ipoints*ipoints+(y-1)*ipoints+(z)
irow(element) = glob_row irow(element) = glob_row
element = element+1 element = element+1
! term depending on (x,y,z+1) ! term depending on (x,y,z+1)
if (z == ipoints) then if (iz == ipoints) then
val(element)=-b1(glob_x,glob_y,glob_z) val(element)=-b1(x,y,z)/deltah2
val(element) = val(element)/(deltah*& zt(k) = exp(-x**2-y**2-z**2)*exp(-x)*(-val(element))
& deltah)
zt(k) = exp(-glob_x**2-glob_y**2)*exp(-glob_z)*(-val(element))
else else
val(element)=-b1(glob_x,glob_y,glob_z) val(element)=-b1(x,y,z)/deltah2
val(element) = val(element)/(deltah*& icol(element) = (ix-1)*ipoints*ipoints+(iy-1)*ipoints+(iz+1)
& deltah)
icol(element) = (x-1)*ipoints*ipoints+(y-1)*ipoints+(z+1)
irow(element) = glob_row irow(element) = glob_row
element = element+1 element = element+1
endif endif
! term depending on (x,y+1,z) ! term depending on (x,y+1,z)
if (y == ipoints) then if (iy == ipoints) then
val(element)=-b2(glob_x,glob_y,glob_z) val(element)=-b2(x,y,z)/deltah2
val(element) = val(element)/(deltah*& zt(k) = exp(-x**2-y**2-z**2)*exp(-x)*(-val(element))
& deltah)
zt(k) = exp(-glob_x**2-glob_z**2)*exp(-glob_y)*(-val(element))
else else
val(element)=-b2(glob_x,glob_y,glob_z) val(element)=-b2(x,y,z)/deltah2
val(element) = val(element)/(deltah*& icol(element) = (ix-1)*ipoints*ipoints+(iy)*ipoints+(iz)
& deltah)
icol(element)=(x-1)*ipoints*ipoints+(y)*ipoints+(z)
irow(element) = glob_row irow(element) = glob_row
element = element+1 element = element+1
endif endif
! term depending on (x+1,y,z) ! term depending on (x+1,y,z)
if (x == ipoints) then if (ix==ipoints) then
val(element)=-b3(glob_x,glob_y,glob_z) val(element)=-b3(x,y,z)/deltah2
val(element) = val(element)/(deltah*& zt(k) = exp(-y**2-z**2)*exp(-x)*(-val(element))
& deltah)
zt(k) = exp(-glob_y**2-glob_z**2)*exp(-glob_x)*(-val(element))
else else
val(element)=-b3(glob_x,glob_y,glob_z) val(element)=-b3(x,y,z)/deltah2
val(element) = val(element)/(deltah*& icol(element) = (ix)*ipoints*ipoints+(iy-1)*ipoints+(iz)
& deltah)
icol(element) = (x)*ipoints*ipoints+(y-1)*ipoints+(z)
irow(element) = glob_row irow(element) = glob_row
element = element+1 element = element+1
endif endif

150
examples/pdegen/mld_sexample_1lev.f90
Unescape Escape View File

@ -289,25 +289,25 @@ contains
real(psb_spk_), allocatable :: b(:),xv(:) real(psb_spk_), allocatable :: b(:),xv(:)
type(psb_desc_type) :: desc_a type(psb_desc_type) :: desc_a
integer :: ictxt, info integer :: ictxt, info
! local variables character :: afmt*5
type(psb_sspmat_type) :: a type(psb_sspmat_type) :: a
real(psb_spk_) :: zt(nb),glob_x,glob_y,glob_z real(psb_spk_) :: zt(nb),x,y,z
integer :: m,n,nnz,glob_row,nlr,i,ii,ib,k,ipoints integer :: m,n,nnz,glob_row,nlr,i,ii,ib,k
integer :: x,y,z,ia,indx_owner integer :: ix,iy,iz,ia,indx_owner, ipoints
integer :: np, iam, nr, nt integer :: np, iam, nr, nt
integer :: element integer :: element
integer, allocatable :: irow(:),icol(:),myidx(:) integer, allocatable :: irow(:),icol(:),myidx(:)
real(psb_spk_), allocatable :: val(:) real(psb_spk_), allocatable :: val(:)
! deltah dimension of each grid cell ! deltah dimension of each grid cell
! deltat discretization time ! deltat discretization time
real(psb_spk_) :: deltah real(psb_spk_) :: deltah, deltah2
real(psb_spk_),parameter :: rhs=0.e0,one=1.e0,zero=0.e0 real(psb_spk_),parameter :: rhs=0.0,one=1.0,zero=0.0
real(psb_dpk_) :: t0, t1, t2, t3, tasb, talc, ttot, tgen real(psb_dpk_) :: t0, t1, t2, t3, tasb, talc, ttot, tgen
real(psb_spk_) :: a1, a2, a3, a4, b1, b2, b3 real(psb_spk_) :: a1, a2, a3, a4, b1, b2, b3
external :: a1, a2, a3, a4, b1, b2, b3 external :: a1, a2, a3, a4, b1, b2, b3
integer :: err_act integer :: err_act
character(len=20) :: name character(len=20) :: name, ch_err
info = psb_success_ info = psb_success_
name = 'create_matrix' name = 'create_matrix'
@ -315,16 +315,17 @@ contains
call psb_info(ictxt, iam, np) call psb_info(ictxt, iam, np)
deltah = 1.d0/(idim-1) deltah = 1.d0/(idim-1)
deltah2 = deltah*deltah
! initialize array descriptor and sparse matrix storage; provide an ! initialize array descriptor and sparse matrix storage. provide an
! estimate of the number of non zeroes ! estimate of the number of non zeroes
ipoints=idim-2 ipoints=idim-2
m = ipoints*ipoints*ipoints m = ipoints*ipoints*ipoints
n = m n = m
nnz = ((n*9)/(np)) nnz = ((n*9)/(np))
if(iam == psb_root_) write(0,'("Generating Matrix (size=",i0,")...")')n if(iam == psb_root_) write(psb_out_unit,'("Generating Matrix (size=",i0,")...")')n
! !
! Using a simple BLOCK distribution. ! Using a simple BLOCK distribution.
@ -334,7 +335,7 @@ contains
nt = nr nt = nr
call psb_sum(ictxt,nt) call psb_sum(ictxt,nt)
if (nt /= m) write(0,*) iam, 'Initialization error ',nr,nt,m if (nt /= m) write(psb_err_unit,*) iam, 'Initialization error ',nr,nt,m
call psb_barrier(ictxt) call psb_barrier(ictxt)
t0 = psb_wtime() t0 = psb_wtime()
call psb_cdall(ictxt,desc_a,info,nl=nr) call psb_cdall(ictxt,desc_a,info,nl=nr)
@ -348,7 +349,8 @@ contains
if (info /= psb_success_) then if (info /= psb_success_) then
info=psb_err_from_subroutine_ info=psb_err_from_subroutine_
call psb_errpush(info,name) ch_err='allocation rout.'
call psb_errpush(info,name,a_err=ch_err)
goto 9999 goto 9999
end if end if
@ -385,20 +387,20 @@ contains
glob_row=myidx(i) glob_row=myidx(i)
! compute gridpoint coordinates ! compute gridpoint coordinates
if (mod(glob_row,ipoints*ipoints) == 0) then if (mod(glob_row,ipoints*ipoints) == 0) then
x = glob_row/(ipoints*ipoints) ix = glob_row/(ipoints*ipoints)
else else
x = glob_row/(ipoints*ipoints)+1 ix = glob_row/(ipoints*ipoints)+1
endif endif
if (mod((glob_row-(x-1)*ipoints*ipoints),ipoints) == 0) then if (mod((glob_row-(ix-1)*ipoints*ipoints),ipoints) == 0) then
y = (glob_row-(x-1)*ipoints*ipoints)/ipoints iy = (glob_row-(ix-1)*ipoints*ipoints)/ipoints
else else
y = (glob_row-(x-1)*ipoints*ipoints)/ipoints+1 iy = (glob_row-(ix-1)*ipoints*ipoints)/ipoints+1
endif endif
z = glob_row-(x-1)*ipoints*ipoints-(y-1)*ipoints iz = glob_row-(ix-1)*ipoints*ipoints-(iy-1)*ipoints
! glob_x, glob_y, glob_x coordinates ! x, y, x coordinates
glob_x=x*deltah x=ix*deltah
glob_y=y*deltah y=iy*deltah
glob_z=z*deltah z=iz*deltah
! check on boundary points ! check on boundary points
zt(k) = 0.d0 zt(k) = 0.d0
@ -406,104 +408,68 @@ contains
! !
! term depending on (x-1,y,z) ! term depending on (x-1,y,z)
! !
if (x == 1) then if (ix == 1) then
val(element)=-b1(glob_x,glob_y,glob_z)& val(element) = -b1(x,y,z)/deltah2-a1(x,y,z)/deltah
& -a1(glob_x,glob_y,glob_z) zt(k) = exp(-x**2-y**2-z**2)*(-val(element))
val(element) = val(element)/(deltah*&
& deltah)
zt(k) = exp(-glob_y**2-glob_z**2)*(-val(element))
else else
val(element)=-b1(glob_x,glob_y,glob_z)& val(element) = -b1(x,y,z)/deltah2-a1(x,y,z)/deltah
& -a1(glob_x,glob_y,glob_z) icol(element) = (ix-2)*ipoints*ipoints+(iy-1)*ipoints+(iz)
val(element) = val(element)/(deltah*&
& deltah)
icol(element) = (x-2)*ipoints*ipoints+(y-1)*ipoints+(z)
irow(element) = glob_row irow(element) = glob_row
element = element+1 element = element+1
endif endif
! term depending on (x,y-1,z) ! term depending on (x,y-1,z)
if (y == 1) then if (iy == 1) then
val(element)=-b2(glob_x,glob_y,glob_z)& val(element) = -b2(x,y,z)/deltah2-a2(x,y,z)/deltah
& -a2(glob_x,glob_y,glob_z) zt(k) = exp(-x**2-y**2-z**2)*exp(-x)*(-val(element))
val(element) = val(element)/(deltah*&
& deltah)
zt(k) = exp(-glob_x**2-glob_z**2)*(-val(element))
else else
val(element)=-b2(glob_x,glob_y,glob_z)& val(element) = -b2(x,y,z)/deltah2-a2(x,y,z)/deltah
& -a2(glob_x,glob_y,glob_z) icol(element) = (ix-1)*ipoints*ipoints+(iy-2)*ipoints+(iz)
val(element) = val(element)/(deltah*&
& deltah)
icol(element) = (x-1)*ipoints*ipoints+(y-2)*ipoints+(z)
irow(element) = glob_row irow(element) = glob_row
element = element+1 element = element+1
endif endif
! term depending on (x,y,z-1) ! term depending on (x,y,z-1)
if (z == 1) then if (iz == 1) then
val(element)=-b3(glob_x,glob_y,glob_z)& val(element)=-b3(x,y,z)/deltah2-a3(x,y,z)/deltah
& -a3(glob_x,glob_y,glob_z) zt(k) = exp(-x**2-y**2-z**2)*exp(-x)*(-val(element))
val(element) = val(element)/(deltah*&
& deltah)
zt(k) = exp(-glob_x**2-glob_y**2)*(-val(element))
else else
val(element)=-b3(glob_x,glob_y,glob_z)& val(element)=-b3(x,y,z)/deltah2-a3(x,y,z)/deltah
& -a3(glob_x,glob_y,glob_z) icol(element) = (ix-1)*ipoints*ipoints+(iy-1)*ipoints+(iz-1)
val(element) = val(element)/(deltah*&
& deltah)
icol(element) = (x-1)*ipoints*ipoints+(y-1)*ipoints+(z-1)
irow(element) = glob_row irow(element) = glob_row
element = element+1 element = element+1
endif endif
! term depending on (x,y,z) ! term depending on (x,y,z)
val(element)=2*b1(glob_x,glob_y,glob_z)& val(element)=(2*b1(x,y,z) + 2*b2(x,y,z) + 2*b3(x,y,z))/deltah2&
& +2*b2(glob_x,glob_y,glob_z)& & + (a1(x,y,z) + a2(x,y,z) + a3(x,y,z)+ a4(x,y,z))/deltah
& +2*b3(glob_x,glob_y,glob_z)& icol(element) = (ix-1)*ipoints*ipoints+(iy-1)*ipoints+(iz)
& +a1(glob_x,glob_y,glob_z)&
& +a2(glob_x,glob_y,glob_z)&
& +a3(glob_x,glob_y,glob_z)
val(element) = val(element)/(deltah*&
& deltah)
icol(element) = (x-1)*ipoints*ipoints+(y-1)*ipoints+(z)
irow(element) = glob_row irow(element) = glob_row
element = element+1 element = element+1
! term depending on (x,y,z+1) ! term depending on (x,y,z+1)
if (z == ipoints) then if (iz == ipoints) then
val(element)=-b1(glob_x,glob_y,glob_z) val(element)=-b1(x,y,z)/deltah2
val(element) = val(element)/(deltah*& zt(k) = exp(-x**2-y**2-z**2)*exp(-x)*(-val(element))
& deltah)
zt(k) = exp(-glob_x**2-glob_y**2)*exp(-glob_z)*(-val(element))
else else
val(element)=-b1(glob_x,glob_y,glob_z) val(element)=-b1(x,y,z)/deltah2
val(element) = val(element)/(deltah*& icol(element) = (ix-1)*ipoints*ipoints+(iy-1)*ipoints+(iz+1)
& deltah)
icol(element) = (x-1)*ipoints*ipoints+(y-1)*ipoints+(z+1)
irow(element) = glob_row irow(element) = glob_row
element = element+1 element = element+1
endif endif
! term depending on (x,y+1,z) ! term depending on (x,y+1,z)
if (y == ipoints) then if (iy == ipoints) then
val(element)=-b2(glob_x,glob_y,glob_z) val(element)=-b2(x,y,z)/deltah2
val(element) = val(element)/(deltah*& zt(k) = exp(-x**2-y**2-z**2)*exp(-x)*(-val(element))
& deltah)
zt(k) = exp(-glob_x**2-glob_z**2)*exp(-glob_y)*(-val(element))
else else
val(element)=-b2(glob_x,glob_y,glob_z) val(element)=-b2(x,y,z)/deltah2
val(element) = val(element)/(deltah*& icol(element) = (ix-1)*ipoints*ipoints+(iy)*ipoints+(iz)
& deltah)
icol(element)=(x-1)*ipoints*ipoints+(y)*ipoints+(z)
irow(element) = glob_row irow(element) = glob_row
element = element+1 element = element+1
endif endif
! term depending on (x+1,y,z) ! term depending on (x+1,y,z)
if (x == ipoints) then if (ix==ipoints) then
val(element)=-b3(glob_x,glob_y,glob_z) val(element)=-b3(x,y,z)/deltah2
val(element) = val(element)/(deltah*& zt(k) = exp(-y**2-z**2)*exp(-x)*(-val(element))
& deltah)
zt(k) = exp(-glob_y**2-glob_z**2)*exp(-glob_x)*(-val(element))
else else
val(element)=-b3(glob_x,glob_y,glob_z) val(element)=-b3(x,y,z)/deltah2
val(element) = val(element)/(deltah*& icol(element) = (ix)*ipoints*ipoints+(iy-1)*ipoints+(iz)
& deltah)
icol(element) = (x)*ipoints*ipoints+(y-1)*ipoints+(z)
irow(element) = glob_row irow(element) = glob_row
element = element+1 element = element+1
endif endif

150
examples/pdegen/mld_sexample_ml.f90
Unescape Escape View File

@ -327,25 +327,25 @@ contains
real(psb_spk_), allocatable :: b(:),xv(:) real(psb_spk_), allocatable :: b(:),xv(:)
type(psb_desc_type) :: desc_a type(psb_desc_type) :: desc_a
integer :: ictxt, info integer :: ictxt, info
! local variables character :: afmt*5
type(psb_sspmat_type) :: a type(psb_sspmat_type) :: a
real(psb_spk_) :: zt(nb),glob_x,glob_y,glob_z real(psb_spk_) :: zt(nb),x,y,z
integer :: m,n,nnz,glob_row,nlr,i,ii,ib,k,ipoints integer :: m,n,nnz,glob_row,nlr,i,ii,ib,k
integer :: x,y,z,ia,indx_owner integer :: ix,iy,iz,ia,indx_owner, ipoints
integer :: np, iam, nr, nt integer :: np, iam, nr, nt
integer :: element integer :: element
integer, allocatable :: irow(:),icol(:),myidx(:) integer, allocatable :: irow(:),icol(:),myidx(:)
real(psb_spk_), allocatable :: val(:) real(psb_spk_), allocatable :: val(:)
! deltah dimension of each grid cell ! deltah dimension of each grid cell
! deltat discretization time ! deltat discretization time
real(psb_spk_) :: deltah real(psb_spk_) :: deltah, deltah2
real(psb_spk_),parameter :: rhs=0.e0,one=1.e0,zero=0.e0 real(psb_spk_),parameter :: rhs=0.0,one=1.0,zero=0.0
real(psb_dpk_) :: t0, t1, t2, t3, tasb, talc, ttot, tgen real(psb_dpk_) :: t0, t1, t2, t3, tasb, talc, ttot, tgen
real(psb_spk_) :: a1, a2, a3, a4, b1, b2, b3 real(psb_spk_) :: a1, a2, a3, a4, b1, b2, b3
external :: a1, a2, a3, a4, b1, b2, b3 external :: a1, a2, a3, a4, b1, b2, b3
integer :: err_act integer :: err_act
character(len=20) :: name character(len=20) :: name, ch_err
info = psb_success_ info = psb_success_
name = 'create_matrix' name = 'create_matrix'
@ -353,16 +353,17 @@ contains
call psb_info(ictxt, iam, np) call psb_info(ictxt, iam, np)
deltah = 1.d0/(idim-1) deltah = 1.d0/(idim-1)
deltah2 = deltah*deltah
! initialize array descriptor and sparse matrix storage; provide an ! initialize array descriptor and sparse matrix storage. provide an
! estimate of the number of non zeroes ! estimate of the number of non zeroes
ipoints=idim-2 ipoints=idim-2
m = ipoints*ipoints*ipoints m = ipoints*ipoints*ipoints
n = m n = m
nnz = ((n*9)/(np)) nnz = ((n*9)/(np))
if(iam == psb_root_) write(0,'("Generating Matrix (size=",i0,")...")')n if(iam == psb_root_) write(psb_out_unit,'("Generating Matrix (size=",i0,")...")')n
! !
! Using a simple BLOCK distribution. ! Using a simple BLOCK distribution.
@ -372,7 +373,7 @@ contains
nt = nr nt = nr
call psb_sum(ictxt,nt) call psb_sum(ictxt,nt)
if (nt /= m) write(0,*) iam, 'Initialization error ',nr,nt,m if (nt /= m) write(psb_err_unit,*) iam, 'Initialization error ',nr,nt,m
call psb_barrier(ictxt) call psb_barrier(ictxt)
t0 = psb_wtime() t0 = psb_wtime()
call psb_cdall(ictxt,desc_a,info,nl=nr) call psb_cdall(ictxt,desc_a,info,nl=nr)
@ -386,7 +387,8 @@ contains
if (info /= psb_success_) then if (info /= psb_success_) then
info=psb_err_from_subroutine_ info=psb_err_from_subroutine_
call psb_errpush(info,name) ch_err='allocation rout.'
call psb_errpush(info,name,a_err=ch_err)
goto 9999 goto 9999
end if end if
@ -423,20 +425,20 @@ contains
glob_row=myidx(i) glob_row=myidx(i)
! compute gridpoint coordinates ! compute gridpoint coordinates
if (mod(glob_row,ipoints*ipoints) == 0) then if (mod(glob_row,ipoints*ipoints) == 0) then
x = glob_row/(ipoints*ipoints) ix = glob_row/(ipoints*ipoints)
else else
x = glob_row/(ipoints*ipoints)+1 ix = glob_row/(ipoints*ipoints)+1
endif endif
if (mod((glob_row-(x-1)*ipoints*ipoints),ipoints) == 0) then if (mod((glob_row-(ix-1)*ipoints*ipoints),ipoints) == 0) then
y = (glob_row-(x-1)*ipoints*ipoints)/ipoints iy = (glob_row-(ix-1)*ipoints*ipoints)/ipoints
else else
y = (glob_row-(x-1)*ipoints*ipoints)/ipoints+1 iy = (glob_row-(ix-1)*ipoints*ipoints)/ipoints+1
endif endif
z = glob_row-(x-1)*ipoints*ipoints-(y-1)*ipoints iz = glob_row-(ix-1)*ipoints*ipoints-(iy-1)*ipoints
! glob_x, glob_y, glob_x coordinates ! x, y, x coordinates
glob_x=x*deltah x=ix*deltah
glob_y=y*deltah y=iy*deltah
glob_z=z*deltah z=iz*deltah
! check on boundary points ! check on boundary points
zt(k) = 0.d0 zt(k) = 0.d0
@ -444,104 +446,68 @@ contains
! !
! term depending on (x-1,y,z) ! term depending on (x-1,y,z)
! !
if (x == 1) then if (ix == 1) then
val(element)=-b1(glob_x,glob_y,glob_z)& val(element) = -b1(x,y,z)/deltah2-a1(x,y,z)/deltah
& -a1(glob_x,glob_y,glob_z) zt(k) = exp(-x**2-y**2-z**2)*(-val(element))
val(element) = val(element)/(deltah*&
& deltah)
zt(k) = exp(-glob_y**2-glob_z**2)*(-val(element))
else else
val(element)=-b1(glob_x,glob_y,glob_z)& val(element) = -b1(x,y,z)/deltah2-a1(x,y,z)/deltah
& -a1(glob_x,glob_y,glob_z) icol(element) = (ix-2)*ipoints*ipoints+(iy-1)*ipoints+(iz)
val(element) = val(element)/(deltah*&
& deltah)
icol(element) = (x-2)*ipoints*ipoints+(y-1)*ipoints+(z)
irow(element) = glob_row irow(element) = glob_row
element = element+1 element = element+1
endif endif
! term depending on (x,y-1,z) ! term depending on (x,y-1,z)
if (y == 1) then if (iy == 1) then
val(element)=-b2(glob_x,glob_y,glob_z)& val(element) = -b2(x,y,z)/deltah2-a2(x,y,z)/deltah
& -a2(glob_x,glob_y,glob_z) zt(k) = exp(-x**2-y**2-z**2)*exp(-x)*(-val(element))
val(element) = val(element)/(deltah*&
& deltah)
zt(k) = exp(-glob_x**2-glob_z**2)*(-val(element))
else else
val(element)=-b2(glob_x,glob_y,glob_z)& val(element) = -b2(x,y,z)/deltah2-a2(x,y,z)/deltah
& -a2(glob_x,glob_y,glob_z) icol(element) = (ix-1)*ipoints*ipoints+(iy-2)*ipoints+(iz)
val(element) = val(element)/(deltah*&
& deltah)
icol(element) = (x-1)*ipoints*ipoints+(y-2)*ipoints+(z)
irow(element) = glob_row irow(element) = glob_row
element = element+1 element = element+1
endif endif
! term depending on (x,y,z-1) ! term depending on (x,y,z-1)
if (z == 1) then if (iz == 1) then
val(element)=-b3(glob_x,glob_y,glob_z)& val(element)=-b3(x,y,z)/deltah2-a3(x,y,z)/deltah
& -a3(glob_x,glob_y,glob_z) zt(k) = exp(-x**2-y**2-z**2)*exp(-x)*(-val(element))
val(element) = val(element)/(deltah*&
& deltah)
zt(k) = exp(-glob_x**2-glob_y**2)*(-val(element))
else else
val(element)=-b3(glob_x,glob_y,glob_z)& val(element)=-b3(x,y,z)/deltah2-a3(x,y,z)/deltah
& -a3(glob_x,glob_y,glob_z) icol(element) = (ix-1)*ipoints*ipoints+(iy-1)*ipoints+(iz-1)
val(element) = val(element)/(deltah*&
& deltah)
icol(element) = (x-1)*ipoints*ipoints+(y-1)*ipoints+(z-1)
irow(element) = glob_row irow(element) = glob_row
element = element+1 element = element+1
endif endif
! term depending on (x,y,z) ! term depending on (x,y,z)
val(element)=2*b1(glob_x,glob_y,glob_z)& val(element)=(2*b1(x,y,z) + 2*b2(x,y,z) + 2*b3(x,y,z))/deltah2&
& +2*b2(glob_x,glob_y,glob_z)& & + (a1(x,y,z) + a2(x,y,z) + a3(x,y,z)+ a4(x,y,z))/deltah
& +2*b3(glob_x,glob_y,glob_z)& icol(element) = (ix-1)*ipoints*ipoints+(iy-1)*ipoints+(iz)
& +a1(glob_x,glob_y,glob_z)&
& +a2(glob_x,glob_y,glob_z)&
& +a3(glob_x,glob_y,glob_z)
val(element) = val(element)/(deltah*&
& deltah)
icol(element) = (x-1)*ipoints*ipoints+(y-1)*ipoints+(z)
irow(element) = glob_row irow(element) = glob_row
element = element+1 element = element+1
! term depending on (x,y,z+1) ! term depending on (x,y,z+1)
if (z == ipoints) then if (iz == ipoints) then
val(element)=-b1(glob_x,glob_y,glob_z) val(element)=-b1(x,y,z)/deltah2
val(element) = val(element)/(deltah*& zt(k) = exp(-x**2-y**2-z**2)*exp(-x)*(-val(element))
& deltah)
zt(k) = exp(-glob_x**2-glob_y**2)*exp(-glob_z)*(-val(element))
else else
val(element)=-b1(glob_x,glob_y,glob_z) val(element)=-b1(x,y,z)/deltah2
val(element) = val(element)/(deltah*& icol(element) = (ix-1)*ipoints*ipoints+(iy-1)*ipoints+(iz+1)
& deltah)
icol(element) = (x-1)*ipoints*ipoints+(y-1)*ipoints+(z+1)
irow(element) = glob_row irow(element) = glob_row
element = element+1 element = element+1
endif endif
! term depending on (x,y+1,z) ! term depending on (x,y+1,z)
if (y == ipoints) then if (iy == ipoints) then
val(element)=-b2(glob_x,glob_y,glob_z) val(element)=-b2(x,y,z)/deltah2
val(element) = val(element)/(deltah*& zt(k) = exp(-x**2-y**2-z**2)*exp(-x)*(-val(element))
& deltah)
zt(k) = exp(-glob_x**2-glob_z**2)*exp(-glob_y)*(-val(element))
else else
val(element)=-b2(glob_x,glob_y,glob_z) val(element)=-b2(x,y,z)/deltah2
val(element) = val(element)/(deltah*& icol(element) = (ix-1)*ipoints*ipoints+(iy)*ipoints+(iz)
& deltah)
icol(element)=(x-1)*ipoints*ipoints+(y)*ipoints+(z)
irow(element) = glob_row irow(element) = glob_row
element = element+1 element = element+1
endif endif
! term depending on (x+1,y,z) ! term depending on (x+1,y,z)
if (x == ipoints) then if (ix==ipoints) then
val(element)=-b3(glob_x,glob_y,glob_z) val(element)=-b3(x,y,z)/deltah2
val(element) = val(element)/(deltah*& zt(k) = exp(-y**2-z**2)*exp(-x)*(-val(element))
& deltah)
zt(k) = exp(-glob_y**2-glob_z**2)*exp(-glob_x)*(-val(element))
else else
val(element)=-b3(glob_x,glob_y,glob_z) val(element)=-b3(x,y,z)/deltah2
val(element) = val(element)/(deltah*& icol(element) = (ix)*ipoints*ipoints+(iy-1)*ipoints+(iz)
& deltah)
icol(element) = (x)*ipoints*ipoints+(y-1)*ipoints+(z)
irow(element) = glob_row irow(element) = glob_row
element = element+1 element = element+1
endif endif

79
tests/newslv/ppde.f90
Unescape Escape View File

@ -145,7 +145,13 @@ program ppde
if(psb_get_errstatus() /= 0) goto 9999 if(psb_get_errstatus() /= 0) goto 9999
name='pde90' name='pde90'
call psb_set_errverbosity(2) call psb_set_errverbosity(2)
!
! Hello world
!
if (iam == psb_root_) then
write(*,*) 'Welcome to MLD2P4 version: ',mld_version_string_
write(*,*) 'This is the ',trim(name),' sample program'
end if
! !
! get parameters ! get parameters
@ -190,7 +196,8 @@ program ppde
call mld_precset(prec,mld_aggr_alg_, prectype%aggr_alg,info) call mld_precset(prec,mld_aggr_alg_, prectype%aggr_alg,info)
call mld_precset(prec,mld_ml_type_, prectype%mltype, info) call mld_precset(prec,mld_ml_type_, prectype%mltype, info)
call mld_precset(prec,mld_smoother_pos_, prectype%smthpos, info) call mld_precset(prec,mld_smoother_pos_, prectype%smthpos, info)
call mld_precset(prec,mld_aggr_thresh_, prectype%athres, info) if (prectype%athres >= dzero) &
& call mld_precset(prec,mld_aggr_thresh_, prectype%athres, info)
call mld_precset(prec,mld_coarse_solve_, prectype%csolve, info) call mld_precset(prec,mld_coarse_solve_, prectype%csolve, info)
call mld_precset(prec,mld_coarse_subsolve_, prectype%csbsolve,info) call mld_precset(prec,mld_coarse_subsolve_, prectype%csbsolve,info)
call mld_precset(prec,mld_coarse_mat_, prectype%cmat, info) call mld_precset(prec,mld_coarse_mat_, prectype%cmat, info)
@ -447,8 +454,8 @@ contains
real(psb_dpk_), allocatable :: val(:) real(psb_dpk_), allocatable :: val(:)
! deltah dimension of each grid cell ! deltah dimension of each grid cell
! deltat discretization time ! deltat discretization time
real(psb_dpk_) :: deltah real(psb_dpk_) :: deltah, deltah2
real(psb_dpk_),parameter :: rhs=0.d0,one=1.d0,zero=0.d0 real(psb_dpk_),parameter :: rhs=0.d0,one=1.d0,zero=0.d0
real(psb_dpk_) :: t0, t1, t2, t3, tasb, talc, ttot, tgen real(psb_dpk_) :: t0, t1, t2, t3, tasb, talc, ttot, tgen
real(psb_dpk_) :: a1, a2, a3, a4, b1, b2, b3 real(psb_dpk_) :: a1, a2, a3, a4, b1, b2, b3
external :: a1, a2, a3, a4, b1, b2, b3 external :: a1, a2, a3, a4, b1, b2, b3
@ -462,7 +469,8 @@ contains
call psb_info(ictxt, iam, np) call psb_info(ictxt, iam, np)
deltah = 1.d0/(idim-1) deltah = 1.d0/(idim-1)
deltah2 = deltah*deltah
! initialize array descriptor and sparse matrix storage. provide an ! initialize array descriptor and sparse matrix storage. provide an
! estimate of the number of non zeroes ! estimate of the number of non zeroes
@ -470,7 +478,7 @@ contains
m = idim*idim*idim m = idim*idim*idim
n = m n = m
nnz = ((n*9)/(np)) nnz = ((n*9)/(np))
if(iam == psb_root_) write(*,'("Generating Matrix (size=",i0,")...")')n if(iam == psb_root_) write(psb_out_unit,'("Generating Matrix (size=",i0,")...")')n
! !
! Using a simple BLOCK distribution. ! Using a simple BLOCK distribution.
@ -480,7 +488,7 @@ contains
nt = nr nt = nr
call psb_sum(ictxt,nt) call psb_sum(ictxt,nt)
if (nt /= m) write(0,*) iam, 'Initialization error ',nr,nt,m if (nt /= m) write(psb_err_unit,*) iam, 'Initialization error ',nr,nt,m
call psb_barrier(ictxt) call psb_barrier(ictxt)
t0 = psb_wtime() t0 = psb_wtime()
call psb_cdall(ictxt,desc_a,info,nl=nr) call psb_cdall(ictxt,desc_a,info,nl=nr)
@ -554,79 +562,66 @@ contains
! term depending on (x-1,y,z) ! term depending on (x-1,y,z)
! !
if (ix == 1) then if (ix == 1) then
val(element)=-b1(x,y,z)-a1(x,y,z) val(element) = -b1(x,y,z)/deltah2-a1(x,y,z)/deltah
val(element) = val(element)/(deltah*& zt(k) = exp(-x**2-y**2-z**2)*(-val(element))
& deltah)
zt(k) = exp(-y**2-z**2)*(-val(element))
else else
val(element)=-b1(x,y,z)-a1(x,y,z) val(element) = -b1(x,y,z)/deltah2-a1(x,y,z)/deltah
val(element) = val(element)/(deltah*&
& deltah)
icol(element) = (ix-2)*idim*idim+(iy-1)*idim+(iz) icol(element) = (ix-2)*idim*idim+(iy-1)*idim+(iz)
irow(element) = glob_row irow(element) = glob_row
element = element+1 element = element+1
endif endif
! term depending on (x,y-1,z) ! term depending on (x,y-1,z)
if (iy == 1) then if (iy == 1) then
val(element)=-b2(x,y,z)-a2(x,y,z) val(element) = -b2(x,y,z)/deltah2-a2(x,y,z)/deltah
val(element) = val(element)/(deltah*& zt(k) = exp(-x**2-y**2-z**2)*exp(-x)*(-val(element))
& deltah)
zt(k) = exp(-y**2-z**2)*exp(-x)*(-val(element))
else else
val(element)=-b2(x,y,z)-a2(x,y,z) val(element) = -b2(x,y,z)/deltah2-a2(x,y,z)/deltah
val(element) = val(element)/(deltah*deltah)
icol(element) = (ix-1)*idim*idim+(iy-2)*idim+(iz) icol(element) = (ix-1)*idim*idim+(iy-2)*idim+(iz)
irow(element) = glob_row irow(element) = glob_row
element = element+1 element = element+1
endif endif
! term depending on (x,y,z-1) ! term depending on (x,y,z-1)
if (iz == 1) then if (iz == 1) then
val(element)=-b3(x,y,z)-a3(x,y,z) val(element)=-b3(x,y,z)/deltah2-a3(x,y,z)/deltah
val(element) = val(element)/(deltah*deltah) zt(k) = exp(-x**2-y**2-z**2)*exp(-x)*(-val(element))
zt(k) = exp(-y**2-z**2)*exp(-x)*(-val(element))
else else
val(element)=-b3(x,y,z)-a3(x,y,z) val(element)=-b3(x,y,z)/deltah2-a3(x,y,z)/deltah
val(element) = val(element)/(deltah*deltah)
icol(element) = (ix-1)*idim*idim+(iy-1)*idim+(iz-1) icol(element) = (ix-1)*idim*idim+(iy-1)*idim+(iz-1)
irow(element) = glob_row irow(element) = glob_row
element = element+1 element = element+1
endif endif
! term depending on (x,y,z) ! term depending on (x,y,z)
val(element)=2*b1(x,y,z) + 2*b2(x,y,z)& val(element)=(2*b1(x,y,z) + 2*b2(x,y,z) + 2*b3(x,y,z))/deltah2&
& + 2*b3(x,y,z) + a1(x,y,z)& & + (a1(x,y,z) + a2(x,y,z) + a3(x,y,z)+ a4(x,y,z))/deltah
& + a2(x,y,z) + a3(x,y,z)
val(element) = val(element)/(deltah*deltah)
icol(element) = (ix-1)*idim*idim+(iy-1)*idim+(iz) icol(element) = (ix-1)*idim*idim+(iy-1)*idim+(iz)
irow(element) = glob_row irow(element) = glob_row
element = element+1 element = element+1
! term depending on (x,y,z+1) ! term depending on (x,y,z+1)
if (iz == idim) then if (iz == idim) then
val(element)=-b1(x,y,z) val(element)=-b1(x,y,z)/deltah2
val(element) = val(element)/(deltah*deltah) zt(k) = exp(-x**2-y**2-z**2)*exp(-x)*(-val(element))
zt(k) = exp(-y**2-z**2)*exp(-x)*(-val(element))
else else
val(element)=-b1(x,y,z) val(element)=-b1(x,y,z)/deltah2
val(element) = val(element)/(deltah*deltah)
icol(element) = (ix-1)*idim*idim+(iy-1)*idim+(iz+1) icol(element) = (ix-1)*idim*idim+(iy-1)*idim+(iz+1)
irow(element) = glob_row irow(element) = glob_row
element = element+1 element = element+1
endif endif
! term depending on (x,y+1,z) ! term depending on (x,y+1,z)
if (iy == idim) then if (iy == idim) then
val(element)=-b2(x,y,z) val(element)=-b2(x,y,z)/deltah2
val(element) = val(element)/(deltah*deltah) zt(k) = exp(-x**2-y**2-z**2)*exp(-x)*(-val(element))
zt(k) = exp(-y**2-z**2)*exp(-x)*(-val(element))
else else
val(element)=-b2(x,y,z) val(element)=-b2(x,y,z)/deltah2
val(element) = val(element)/(deltah*deltah)
icol(element) = (ix-1)*idim*idim+(iy)*idim+(iz) icol(element) = (ix-1)*idim*idim+(iy)*idim+(iz)
irow(element) = glob_row irow(element) = glob_row
element = element+1 element = element+1
endif endif
! term depending on (x+1,y,z) ! term depending on (x+1,y,z)
if (ix<idim) then if (ix==idim) then
val(element)=-b3(x,y,z) val(element)=-b3(x,y,z)/deltah2
val(element) = val(element)/(deltah*deltah) zt(k) = exp(-y**2-z**2)*exp(-x)*(-val(element))
else
val(element)=-b3(x,y,z)/deltah2
icol(element) = (ix)*idim*idim+(iy-1)*idim+(iz) icol(element) = (ix)*idim*idim+(iy-1)*idim+(iz)
irow(element) = glob_row irow(element) = glob_row
element = element+1 element = element+1

78
tests/newslv/spde.f90
Unescape Escape View File

@ -143,6 +143,13 @@ program spde
if(psb_get_errstatus() /= 0) goto 9999 if(psb_get_errstatus() /= 0) goto 9999
name='pde90' name='pde90'
call psb_set_errverbosity(2) call psb_set_errverbosity(2)
!
! Hello world
!
if (iam == psb_root_) then
write(*,*) 'Welcome to MLD2P4 version: ',mld_version_string_
write(*,*) 'This is the ',trim(name),' sample program'
end if
! !
! get parameters ! get parameters
@ -186,7 +193,8 @@ program spde
call mld_precset(prec,mld_aggr_alg_, prectype%aggr_alg,info) call mld_precset(prec,mld_aggr_alg_, prectype%aggr_alg,info)
call mld_precset(prec,mld_ml_type_, prectype%mltype, info) call mld_precset(prec,mld_ml_type_, prectype%mltype, info)
call mld_precset(prec,mld_smoother_pos_, prectype%smthpos, info) call mld_precset(prec,mld_smoother_pos_, prectype%smthpos, info)
call mld_precset(prec,mld_aggr_thresh_, prectype%athres, info) if (prectype%athres >= szero) &
& call mld_precset(prec,mld_aggr_thresh_, prectype%athres, info)
call mld_precset(prec,mld_coarse_solve_, prectype%csolve, info) call mld_precset(prec,mld_coarse_solve_, prectype%csolve, info)
call mld_precset(prec,mld_coarse_subsolve_, prectype%csbsolve,info) call mld_precset(prec,mld_coarse_subsolve_, prectype%csbsolve,info)
call mld_precset(prec,mld_coarse_mat_, prectype%cmat, info) call mld_precset(prec,mld_coarse_mat_, prectype%cmat, info)
@ -441,8 +449,8 @@ contains
real(psb_spk_), allocatable :: val(:) real(psb_spk_), allocatable :: val(:)
! deltah dimension of each grid cell ! deltah dimension of each grid cell
! deltat discretization time ! deltat discretization time
real(psb_spk_) :: deltah real(psb_spk_) :: deltah, deltah2
real(psb_spk_),parameter :: rhs=0.0,one=1.0,zero=0.0 real(psb_spk_),parameter :: rhs=0.0,one=1.0,zero=0.0
real(psb_dpk_) :: t0, t1, t2, t3, tasb, talc, ttot, tgen real(psb_dpk_) :: t0, t1, t2, t3, tasb, talc, ttot, tgen
real(psb_spk_) :: a1, a2, a3, a4, b1, b2, b3 real(psb_spk_) :: a1, a2, a3, a4, b1, b2, b3
external :: a1, a2, a3, a4, b1, b2, b3 external :: a1, a2, a3, a4, b1, b2, b3
@ -456,7 +464,8 @@ contains
call psb_info(ictxt, iam, np) call psb_info(ictxt, iam, np)
deltah = 1.0/(idim-1) deltah = 1.d0/(idim-1)
deltah2 = deltah*deltah
! initialize array descriptor and sparse matrix storage. provide an ! initialize array descriptor and sparse matrix storage. provide an
! estimate of the number of non zeroes ! estimate of the number of non zeroes
@ -464,7 +473,7 @@ contains
m = idim*idim*idim m = idim*idim*idim
n = m n = m
nnz = ((n*9)/(np)) nnz = ((n*9)/(np))
if(iam == psb_root_) write(0,'("Generating Matrix (size=",i0,")...")')n if(iam == psb_root_) write(psb_out_unit,'("Generating Matrix (size=",i0,")...")')n
! !
! Using a simple BLOCK distribution. ! Using a simple BLOCK distribution.
@ -474,7 +483,7 @@ contains
nt = nr nt = nr
call psb_sum(ictxt,nt) call psb_sum(ictxt,nt)
if (nt /= m) write(0,*) iam, 'Initialization error ',nr,nt,m if (nt /= m) write(psb_err_unit,*) iam, 'Initialization error ',nr,nt,m
call psb_barrier(ictxt) call psb_barrier(ictxt)
t0 = psb_wtime() t0 = psb_wtime()
call psb_cdall(ictxt,desc_a,info,nl=nr) call psb_cdall(ictxt,desc_a,info,nl=nr)
@ -548,79 +557,66 @@ contains
! term depending on (x-1,y,z) ! term depending on (x-1,y,z)
! !
if (ix == 1) then if (ix == 1) then
val(element)=-b1(x,y,z)-a1(x,y,z) val(element) = -b1(x,y,z)/deltah2-a1(x,y,z)/deltah
val(element) = val(element)/(deltah*& zt(k) = exp(-x**2-y**2-z**2)*(-val(element))
& deltah)
zt(k) = exp(-y**2-z**2)*(-val(element))
else else
val(element)=-b1(x,y,z)-a1(x,y,z) val(element) = -b1(x,y,z)/deltah2-a1(x,y,z)/deltah
val(element) = val(element)/(deltah*&
& deltah)
icol(element) = (ix-2)*idim*idim+(iy-1)*idim+(iz) icol(element) = (ix-2)*idim*idim+(iy-1)*idim+(iz)
irow(element) = glob_row irow(element) = glob_row
element = element+1 element = element+1
endif endif
! term depending on (x,y-1,z) ! term depending on (x,y-1,z)
if (iy == 1) then if (iy == 1) then
val(element)=-b2(x,y,z)-a2(x,y,z) val(element) = -b2(x,y,z)/deltah2-a2(x,y,z)/deltah
val(element) = val(element)/(deltah*& zt(k) = exp(-x**2-y**2-z**2)*exp(-x)*(-val(element))
& deltah)
zt(k) = exp(-y**2-z**2)*exp(-x)*(-val(element))
else else
val(element)=-b2(x,y,z)-a2(x,y,z) val(element) = -b2(x,y,z)/deltah2-a2(x,y,z)/deltah
val(element) = val(element)/(deltah*deltah)
icol(element) = (ix-1)*idim*idim+(iy-2)*idim+(iz) icol(element) = (ix-1)*idim*idim+(iy-2)*idim+(iz)
irow(element) = glob_row irow(element) = glob_row
element = element+1 element = element+1
endif endif
! term depending on (x,y,z-1) ! term depending on (x,y,z-1)
if (iz == 1) then if (iz == 1) then
val(element)=-b3(x,y,z)-a3(x,y,z) val(element)=-b3(x,y,z)/deltah2-a3(x,y,z)/deltah
val(element) = val(element)/(deltah*deltah) zt(k) = exp(-x**2-y**2-z**2)*exp(-x)*(-val(element))
zt(k) = exp(-y**2-z**2)*exp(-x)*(-val(element))
else else
val(element)=-b3(x,y,z)-a3(x,y,z) val(element)=-b3(x,y,z)/deltah2-a3(x,y,z)/deltah
val(element) = val(element)/(deltah*deltah)
icol(element) = (ix-1)*idim*idim+(iy-1)*idim+(iz-1) icol(element) = (ix-1)*idim*idim+(iy-1)*idim+(iz-1)
irow(element) = glob_row irow(element) = glob_row
element = element+1 element = element+1
endif endif
! term depending on (x,y,z) ! term depending on (x,y,z)
val(element)=2*b1(x,y,z) + 2*b2(x,y,z)& val(element)=(2*b1(x,y,z) + 2*b2(x,y,z) + 2*b3(x,y,z))/deltah2&
& + 2*b3(x,y,z) + a1(x,y,z)& & + (a1(x,y,z) + a2(x,y,z) + a3(x,y,z)+ a4(x,y,z))/deltah
& + a2(x,y,z) + a3(x,y,z)
val(element) = val(element)/(deltah*deltah)
icol(element) = (ix-1)*idim*idim+(iy-1)*idim+(iz) icol(element) = (ix-1)*idim*idim+(iy-1)*idim+(iz)
irow(element) = glob_row irow(element) = glob_row
element = element+1 element = element+1
! term depending on (x,y,z+1) ! term depending on (x,y,z+1)
if (iz == idim) then if (iz == idim) then
val(element)=-b1(x,y,z) val(element)=-b1(x,y,z)/deltah2
val(element) = val(element)/(deltah*deltah) zt(k) = exp(-x**2-y**2-z**2)*exp(-x)*(-val(element))
zt(k) = exp(-y**2-z**2)*exp(-x)*(-val(element))
else else
val(element)=-b1(x,y,z) val(element)=-b1(x,y,z)/deltah2
val(element) = val(element)/(deltah*deltah)
icol(element) = (ix-1)*idim*idim+(iy-1)*idim+(iz+1) icol(element) = (ix-1)*idim*idim+(iy-1)*idim+(iz+1)
irow(element) = glob_row irow(element) = glob_row
element = element+1 element = element+1
endif endif
! term depending on (x,y+1,z) ! term depending on (x,y+1,z)
if (iy == idim) then if (iy == idim) then
val(element)=-b2(x,y,z) val(element)=-b2(x,y,z)/deltah2
val(element) = val(element)/(deltah*deltah) zt(k) = exp(-x**2-y**2-z**2)*exp(-x)*(-val(element))
zt(k) = exp(-y**2-z**2)*exp(-x)*(-val(element))
else else
val(element)=-b2(x,y,z) val(element)=-b2(x,y,z)/deltah2
val(element) = val(element)/(deltah*deltah)
icol(element) = (ix-1)*idim*idim+(iy)*idim+(iz) icol(element) = (ix-1)*idim*idim+(iy)*idim+(iz)
irow(element) = glob_row irow(element) = glob_row
element = element+1 element = element+1
endif endif
! term depending on (x+1,y,z) ! term depending on (x+1,y,z)
if (ix<idim) then if (iy == idim) then
val(element)=-b3(x,y,z) val(element)=-b2(x,y,z)/deltah2
val(element) = val(element)/(deltah*deltah) zt(k) = exp(-x**2-y**2-z**2)*exp(-x)*(-val(element))
else
val(element)=-b3(x,y,z)/deltah2
icol(element) = (ix)*idim*idim+(iy-1)*idim+(iz) icol(element) = (ix)*idim*idim+(iy-1)*idim+(iz)
irow(element) = glob_row irow(element) = glob_row
element = element+1 element = element+1

7
tests/pdegen/ppde.f90
Unescape Escape View File

@ -422,7 +422,7 @@ contains
! discretize the partial diferential equation ! discretize the partial diferential equation
! !
! b1 dd(u) b2 dd(u) b3 dd(u) a1 d(u) a2 d(u) a3 d(u) ! b1 dd(u) b2 dd(u) b3 dd(u) a1 d(u) a2 d(u) a3 d(u)
! - ------ - ------ - ------ - ----- - ------ - ------ + a4 u ! - ------ - ------ - ------ - ----- - ------ - ------ + a4 u = 0
! dxdx dydy dzdz dx dy dz ! dxdx dydy dzdz dx dy dz
! !
! with Dirichlet boundary conditions, on the unit cube 0<=x,y,z<=1. ! with Dirichlet boundary conditions, on the unit cube 0<=x,y,z<=1.
@ -615,7 +615,10 @@ contains
element = element+1 element = element+1
endif endif
! term depending on (x+1,y,z) ! term depending on (x+1,y,z)
if (ix<idim) then if (ix==idim) then
val(element)=-b3(x,y,z)/deltah2
zt(k) = exp(-y**2-z**2)*exp(-x)*(-val(element))
else
val(element)=-b3(x,y,z)/deltah2 val(element)=-b3(x,y,z)/deltah2
icol(element) = (ix)*idim*idim+(iy-1)*idim+(iz) icol(element) = (ix)*idim*idim+(iy-1)*idim+(iz)
irow(element) = glob_row irow(element) = glob_row

5
tests/pdegen/spde.f90
Unescape Escape View File

@ -613,7 +613,10 @@ contains
element = element+1 element = element+1
endif endif
! term depending on (x+1,y,z) ! term depending on (x+1,y,z)
if (ix<idim) then if (iy == idim) then
val(element)=-b2(x,y,z)/deltah2
zt(k) = exp(-x**2-y**2-z**2)*exp(-x)*(-val(element))
else
val(element)=-b3(x,y,z)/deltah2 val(element)=-b3(x,y,z)/deltah2
icol(element) = (ix)*idim*idim+(iy-1)*idim+(iz) icol(element) = (ix)*idim*idim+(iy-1)*idim+(iz)
irow(element) = glob_row irow(element) = glob_row

Write Preview
Loading…
Cancel
Save