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

[bugfix] Prevent double visit of sum/product children

parent 8c4d996a
No related branches found
No related tags found
No related merge requests found
......@@ -282,7 +282,7 @@ class UFL2LoopyVisitor(ModifiedTerminalTracker):
ops = tuple(self.call(op) for op in o.ufl_operands)
if all(isinstance(op, (int, float)) for op in ops):
return ptproduct(ops)
return prim.flattened_product(tuple(self.call(op) for op in o.ufl_operands))
return prim.flattened_product(ops)
def float_value(self, o):
return o.value()
......@@ -297,7 +297,7 @@ class UFL2LoopyVisitor(ModifiedTerminalTracker):
ops = tuple(self.call(op) for op in o.ufl_operands)
if all(isinstance(op, (int, float)) for op in ops):
return sum(ops)
return prim.flattened_sum(tuple(self.call(op) for op in o.ufl_operands))
return prim.flattened_sum(ops)
def zero(self, o):
# UFL has Zeroes with shape. We ignore those indices.
......
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