Skip to content
Snippets Groups Projects
Commit 210020e0 authored by Marcel Koch's avatar Marcel Koch
Browse files

let loopy manage reference gradients

parent 8956ce3b
No related branches found
No related tags found
No related merge requests found
......@@ -197,16 +197,7 @@ def evaluate_coefficient_gradient(visitor, element, name, container, restriction
from ufl import FiniteElement
assert isinstance(sub_element, FiniteElement)
if get_option("vectorization_blockstructured"):
temporary_variable(name,
shape=(element.cell().geometric_dimension(),),
managed=True
)
else:
temporary_variable(name,
shape=(element.cell().geometric_dimension(),),
shape_impl=('arr',)
)
temporary_variable(name, shape=(element.cell().geometric_dimension(),), managed=True)
dimindex = component_iname(count=0)
......@@ -230,6 +221,5 @@ def evaluate_coefficient_gradient(visitor, element, name, container, restriction
instruction(expression=Reduction("sum", basisindex, reduction_expr, allow_simultaneous=True),
assignee=assignee,
forced_iname_deps=frozenset(get_backend("quad_inames")()).union(frozenset({dimindex})),
forced_iname_deps_is_final=True,
tags=frozenset({'blockvec'})
forced_iname_deps_is_final=True
)
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment