Source code for firedrake.parameters

"""The parameters dictionary contains global parameter settings."""
from pyop2.configuration import configuration, target as pyop2_target
from tsfc import default_parameters
import sys
from firedrake.utils import ScalarType, ScalarType_c

max_float = sys.float_info[0]

__all__ = ['Parameters', 'parameters', 'disable_performance_optimisations']


[docs] class Parameters(dict): def __init__(self, name=None, **kwargs): self._name = name self._update_function = None for key, value in kwargs.items(): self.add(key, value)
[docs] def add(self, key, value=None): if isinstance(key, Parameters): self[key.name()] = key else: self[key] = value
def __setitem__(self, key, value): super(Parameters, self).__setitem__(key, value) if hasattr(self, "_update_function") and self._update_function: self._update_function(key, value)
[docs] def name(self): return self._name
[docs] def rename(self, name): self._name = name
def __getstate__(self): # Remove non-picklable update function slot d = self.__dict__.copy() del d["_update_function"] return d
[docs] def set_update_function(self, callable): """Set a function to be called whenever a dictionary entry is changed. :arg callable: the function. The function receives two arguments, the key-value pair of updated entries.""" self._update_function = callable
parameters = Parameters() """A nested dictionary of parameters used by Firedrake""" # Default to the values of PyOP2 configuration dictionary pyop2_opts = Parameters("pyop2_options", **configuration) pyop2_opts.set_update_function(lambda k, v: configuration.unsafe_reconfigure(**{k: v})) # Override values pyop2_opts["type_check"] = True target = pyop2_target parameters.add(pyop2_opts) parameters.add(Parameters("form_compiler", **default_parameters())) parameters["form_compiler"]['scalar_type'] = ScalarType parameters["form_compiler"]['scalar_type_c'] = ScalarType_c parameters["reorder_meshes"] = True # One of nest, aij, baij or matfree parameters["default_matrix_type"] = "aij" # One of aij or baij parameters["default_sub_matrix_type"] = "baij" parameters["type_check_safe_par_loops"] = False parameters.add(Parameters("slate_compiler")) parameters["slate_compiler"]["optimise"] = True # Should a Slate multiplication be replaced by an action? parameters["slate_compiler"]["replace_mul"] = False
[docs] def disable_performance_optimisations(): """Switches off performance optimisations in Firedrake. This is mostly useful for debugging purposes. This enables PyOP2's runtime checking of par_loop arguments in all cases (even those where they are claimed safe). Additionally, it switches to compiling generated code in debug mode. Returns a function that can be called with no arguments, to restore the state of the parameters dict.""" check = parameters["pyop2_options"]["type_check"] debug = parameters["pyop2_options"]["debug"] safe_check = parameters["type_check_safe_par_loops"] def restore(): parameters["pyop2_options"]["type_check"] = check parameters["pyop2_options"]["debug"] = debug parameters["type_check_safe_par_loops"] = safe_check parameters["pyop2_options"]["type_check"] = True parameters["pyop2_options"]["debug"] = True parameters["type_check_safe_par_loops"] = True return restore