tlm_adjoint.firedrake.caches
This module implements finite element assembly and linear solver data caching.
Module Contents
- class tlm_adjoint.firedrake.caches.AssemblyCache
A
Cache
for finite element assembly data.- assemble(form, *, bcs=None, form_compiler_parameters=None, linear_solver_parameters=None, replace_map=None)
Perform finite element assembly and cache the result, or return a previously cached result.
- Parameters:
form – The
ufl.Form
to assemble.bcs – Dirichlet boundary conditions.
form_compiler_parameters – Form compiler parameters.
linear_solver_parameters – Linear solver parameters. Required for assembly parameters which appear in the linear solver parameters.
replace_map – A
Mapping
defining a map from symbolic variables to values.
- Returns:
A
tuple
(value_ref, value), where value is the result of the finite element assembly, and value_ref is aCacheRef
storing a reference to value.For an arity zero or arity one form value_ref stores the assembled value.
For an arity two form value_ref is a tuple (A, b_bc). A is the assembled matrix, and b_bc is a boundary condition right-hand-side term which should be added after assembling a right-hand-side with homogeneous boundary conditions applied. b_bc may be None to indicate that this term is zero.
- class tlm_adjoint.firedrake.caches.LinearSolverCache
A
Cache
for linear solver data.- linear_solver(form, *, bcs=None, form_compiler_parameters=None, linear_solver_parameters=None, replace_map=None, assembly_cache=None)
Construct a linear solver and cache the result, or return a previously cached result.
- Parameters:
form – An arity two
ufl.Form
, defining the matrix.bcs – Dirichlet boundary conditions.
form_compiler_parameters – Form compiler parameters.
linear_solver_parameters – Linear solver parameters.
replace_map – A
Mapping
defining a map from symbolic variables to values.assembly_cache –
AssemblyCache
to use for finite element assembly. Defaults to assembly_cache().
- Returns:
A
tuple
(value_ref, value). value is a tuple (solver, A, b_bc), where solver is the linear solver, A is the assembled matrix, and b_bc is a boundary condition right-hand-side term which should be added after assembling a right-hand-side with homogeneous boundary conditions applied. b_bc may be None to indicate that this term is zero. value_ref is aCacheRef
storing a reference to value.
- class tlm_adjoint.firedrake.caches.LocalSolverCache
A
Cache
for element-wise local block diagonal linear solver data.- local_solver(form, *, form_compiler_parameters=None, replace_map=None)
Compute data for an element-wise local block diagonal linear solver and cache the result, or return a previously cached result.
- Parameters:
form – An arity two
ufl.Form
, defining the element-wise local block diagonal matrix.form_compiler_parameters – Form compiler parameters.
replace_map – A
Mapping
defining a map from symbolic variables to values.
- Returns:
A
tuple
(value_ref, value). value is a tlm_adjoint.firedrake.backend_interface.LocalSolver and value_ref is aCacheRef
storing a reference to value.
- tlm_adjoint.firedrake.caches.assembly_cache()
- Returns:
The default
AssemblyCache
.
- tlm_adjoint.firedrake.caches.set_assembly_cache(assembly_cache)
Set the default
AssemblyCache
.- Parameters:
assembly_cache – The new default
AssemblyCache
.
- tlm_adjoint.firedrake.caches.linear_solver_cache()
- Returns:
The default
LinearSolverCache
.
- tlm_adjoint.firedrake.caches.set_linear_solver_cache(linear_solver_cache)
Set the default
LinearSolverCache
.- Parameters:
linear_solver_cache – The new default
LinearSolverCache
.
- tlm_adjoint.firedrake.caches.local_solver_cache()
- Returns:
The default
LocalSolverCache
.
- tlm_adjoint.firedrake.caches.set_local_solver_cache(local_solver_cache)
Set the default
LocalSolverCache
.- Parameters:
local_solver_cache – The new default
LocalSolverCache
.