Skip to content
Snippets Groups Projects
Commit 763d27dc authored by Dominic Kempf's avatar Dominic Kempf
Browse files

pep8

parent 3f831a48
No related branches found
No related tags found
No related merge requests found
......@@ -107,7 +107,6 @@ def vectorize_quadrature_loop(knl):
# and precompute them
#
quantities = []
for insn in insns:
for expr in DependencyMapper()(insn.expression):
......@@ -222,7 +221,7 @@ def vectorize_quadrature_loop(knl):
tags = set(get_pymbolic_tag(expr) for expr in quantity_exprs)
if tags and tags.pop() == "operator_precomputed":
expr, = quantity_exprs
shape=(ceildiv(product(s for s in arg.shape), vec_size), vec_size)
shape = (ceildiv(product(s for s in arg.shape), vec_size), vec_size)
name = loopy_class_member(quantity,
shape=shape,
dim_tags="f,vec",
......
......@@ -65,6 +65,7 @@ class GeoCornersInput(SumfactKernelInputBase, ImmutableRecord):
tags=frozenset({"sumfact_stage{}".format(sf.stage)}),
)
@kernel_cached
@backend(interface="spatial_coordinate", name="default")
def pymbolic_spatial_coordinate_multilinear(visitor_indices):
......
......@@ -42,6 +42,7 @@ class ModifiedArgument(Record):
else:
return self.index
class ModifiedTerminalTracker(MultiFunction):
""" A multifunction base that defines handler for
grad, reference_grad, positive_restricted and negative_restricted.
......
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