tree_amg_multigrid Module Reference

Implements multigrid using the TreeAMG hierarchy structure. USE: More...

Data Types
type	tamg_solver_t
	Type for the TreeAMG solver. More...

Functions/Subroutines
subroutine	tamg_mg_init (this, ax, xh, coef, msh, gs_h, nlvls_in, blst, max_iter)
	Initialization of the TreeAMG multigrid solver.
subroutine	tamg_mg_solve (this, z, r, n)
	Solver function for the TreeAMG solver object.
recursive subroutine	tamg_mg_cycle (x, b, n, lvl, amg, mgstuff)
	Recrsive multigrid cycle for the TreeAMG solver object.
subroutine	calc_resid (r, x, b, amg, lvl, n)
	Wrapper function to calculate residyal.
subroutine	average_duplicates (u, amg, lvl, n)
	Wrapper function to gather scatter and average the duplicates.
subroutine	print_preagg_info (lvl, nagg)
subroutine	fill_lvl_map (amg)

Detailed Description

type(tamg_hierarchy_t) :: amg type(tamg_solver_t) :: amg_solver

call amginit(ax, Xh, coef, msh, gs_h, 3) call amg_solverinit(amg, niter)

call amg_solversolve(xx, f, n)

Function/Subroutine Documentation

average_duplicates()

subroutine tree_amg_multigrid::average_duplicates ( real(kind=rp), dimension(n), intent(inout)  u, type(tamg_hierarchy_t), intent(inout)  amg, integer, intent(in)  lvl, integer, intent(in)  n  )

private

Parameters

U	The target array
amg	The TreeAMG object
lvl	Current level of the cycle
n	Number of dofs

Definition at line 306 of file tree_amg_multigrid.f90.

Here is the caller graph for this function:

calc_resid()

subroutine tree_amg_multigrid::calc_resid ( real(kind=rp), dimension(n), intent(inout)  r, real(kind=rp), dimension(n), intent(inout)  x, real(kind=rp), dimension(n), intent(inout)  b, type(tamg_hierarchy_t), intent(inout)  amg, integer, intent(in)  lvl, integer, intent(in)  n  )

private

Parameters

r	The residual to be returned
x	The current solution
b	The right-hand side
amg	The TreeAMG object
lvl	Current level of the cycle
n	Number of dofs

Definition at line 285 of file tree_amg_multigrid.f90.

Here is the caller graph for this function:

fill_lvl_map()

subroutine tree_amg_multigrid::fill_lvl_map ( type(tamg_hierarchy_t), intent(inout)  amg )

private

Definition at line 328 of file tree_amg_multigrid.f90.

Here is the caller graph for this function:

print_preagg_info()

subroutine tree_amg_multigrid::print_preagg_info ( integer, intent(in)  lvl, integer, intent(in)  nagg  )

private

Definition at line 318 of file tree_amg_multigrid.f90.

Here is the caller graph for this function:

tamg_mg_cycle()

recursive subroutine tree_amg_multigrid::tamg_mg_cycle ( real(kind=rp), dimension(n), intent(inout)  x, real(kind=rp), dimension(n), intent(inout)  b, integer, intent(in)  n, integer, intent(in)  lvl, type(tamg_hierarchy_t), intent(inout)  amg, type(tamg_solver_t), intent(inout)  mgstuff  )

private

Parameters

x	The solution to be returned
b	The right-hand side
n	Number of dofs
lvl	Current level of the cycle
amg	The TreeAMG object
mgstuff	The Solver object. TODO: rename this

-------—<! SMOOTH <! -------—<!

Is coarsest grid.

-------—<! Residual <! -------—<!

-------—<! Restrict <! -------—<!

----------------—<! Call Coarse solve <! ----------------—<!

-------—<! Project <! -------—<!

-------—<! Correct <! -------—<!

-------—<! SMOOTH <! -------—<!

-------—<! Residual <! -------—<!

Definition at line 201 of file tree_amg_multigrid.f90.

Here is the call graph for this function:

Here is the caller graph for this function:

tamg_mg_init()

subroutine tree_amg_multigrid::tamg_mg_init	(	class(tamg_solver_t), intent(inout), target	this,
		class(ax_t), intent(in), target	ax,
		type(space_t), intent(in), target	xh,
		type(coef_t), intent(in), target	coef,
		type(mesh_t), intent(in), target	msh,
		type(gs_t), intent(in), target	gs_h,
		integer, intent(in)	nlvls_in,
		type(bc_list_t), intent(in), target	blst,
		integer, intent(in)	max_iter
	)

Parameters

ax	Finest level matvec operator
Xh	Finest level field
coef	Finest level coeff thing
msh	Finest level mesh information
gs_h	Finest level gather scatter operator
nlvls_in	Number of levels for the TreeAMG hierarchy
blst	Finest level BC list
max_iter	Number of AMG iterations

Create level 1 (neko elements are level 0)

Create the remaining levels

Create the end point

Definition at line 85 of file tree_amg_multigrid.f90.

Here is the call graph for this function:

tamg_mg_solve()

subroutine tree_amg_multigrid::tamg_mg_solve ( class(tamg_solver_t), intent(inout)  this, real(kind=rp), dimension(n), intent(inout)  z, real(kind=rp), dimension(n), intent(inout)  r, integer, intent(in)  n  )

private

Parameters

z	The solution to be returned
r	The right-hand side
n	Number of dofs

Definition at line 176 of file tree_amg_multigrid.f90.

Here is the call graph for this function: