diff --git a/python/dune/codegen/sumfact/vectorization.py b/python/dune/codegen/sumfact/vectorization.py index 2ac0ffb5a41eac1503a3a50b1809c971de330008..38996a90d0dfc018bbb538ddf70772092b8d7f57 100644 --- a/python/dune/codegen/sumfact/vectorization.py +++ b/python/dune/codegen/sumfact/vectorization.py @@ -144,6 +144,12 @@ class PrimitiveApproximateOpcounter(FlopCounter): def map_tagged_variable(self, expr): return self.map_variable(expr) + def map_loopy_function_identifier(self, expr): + if hasattr(expr, "operations"): + return expr.operations() + + raise NotImplementedError("The class {} should implement a symbolic flopcounter.".format(type(expr))) + @kernel_cached def store_operation_count(expr, count):