diff --git a/src/pystencilssfg/ir/postprocessing.py b/src/pystencilssfg/ir/postprocessing.py index d86dedc140a7cde8d18c78781aae1dc039757dad..0626e2ee3716b81675e31b6cded42594a744584e 100644 --- a/src/pystencilssfg/ir/postprocessing.py +++ b/src/pystencilssfg/ir/postprocessing.py @@ -27,38 +27,6 @@ from ..lang import ( ) -class FlattenSequences: - """Flattens any nested sequences occuring in a kernel call tree.""" - - def __call__(self, node: SfgCallTreeNode) -> None: - self.visit(node) - - def visit(self, node: SfgCallTreeNode): - match node: - case SfgSequence(): - self.flatten(node) - case _: - for c in node.children: - self.visit(c) - - def flatten(self, sequence: SfgSequence) -> None: - children_flattened: list[SfgCallTreeNode] = [] - - def flatten(seq: SfgSequence): - for c in seq.children: - if isinstance(c, SfgSequence): - flatten(c) - else: - children_flattened.append(c) - - flatten(sequence) - - for c in children_flattened: - self.visit(c) - - sequence.children = children_flattened - - class PostProcessingContext: def __init__(self) -> None: self._live_variables: dict[str, SfgVar] = dict() @@ -129,9 +97,6 @@ class PostProcessingResult: class CallTreePostProcessing: - def __init__(self): - self._flattener = FlattenSequences() - def __call__(self, ast: SfgCallTreeNode) -> PostProcessingResult: live_vars = self.get_live_variables(ast) return PostProcessingResult(live_vars)