def preprocess_gem(expressions, replace_delta=True, remove_componenttensors=True): """Lower GEM nodes that cannot be translated to C directly.""" if remove_componenttensors: expressions = optimise.remove_componenttensors(expressions) if replace_delta: expressions = optimise.replace_delta(expressions) return expressions
def Integrals(expressions, quadrature_multiindex, argument_multiindices, parameters): """Constructs an integral representation for each GEM integrand expression. :arg expressions: integrand multiplied with quadrature weight; multi-root GEM expression DAG :arg quadrature_multiindex: quadrature multiindex (tuple) :arg argument_multiindices: tuple of argument multiindices, one multiindex for each argument :arg parameters: parameters dictionary :returns: list of integral representations """ # Unroll max_extent = parameters["unroll_indexsum"] if max_extent: def predicate(index): return index.extent <= max_extent expressions = unroll_indexsum(expressions, predicate=predicate) # Choose GEM expression as the integral representation expressions = [index_sum(e, quadrature_multiindex) for e in expressions] expressions = replace_delta(expressions) expressions = remove_componenttensors(expressions) expressions = replace_division(expressions) argument_indices = tuple(itertools.chain(*argument_multiindices)) return optimise_expressions(expressions, argument_indices)
def preprocess_gem(expressions, replace_delta=True, remove_componenttensors=True): """Lower GEM nodes that cannot be translated to C directly.""" if remove_componenttensors: expressions = optimise.remove_componenttensors(expressions) if replace_delta: expressions = optimise.replace_delta(expressions) return expressions
def _evaluate_delta(e, self): """Lower delta and evaluate.""" e, = replace_delta((e, )) return self(e)
def _evaluate_delta(e, self): """Lower delta and evaluate.""" e, = replace_delta((e,)) return self(e)