From 9726d9bd2193c701dd27f1e915096699752812c5 Mon Sep 17 00:00:00 2001 From: Matthias Diener Date: Wed, 27 Nov 2024 14:11:08 -0600 Subject: [PATCH] opt ancestors --- loopy/schedule/tree.py | 6 ++---- 1 file changed, 2 insertions(+), 4 deletions(-) diff --git a/loopy/schedule/tree.py b/loopy/schedule/tree.py index 590f62be1..cc580713c 100644 --- a/loopy/schedule/tree.py +++ b/loopy/schedule/tree.py @@ -97,13 +97,11 @@ def ancestors(self, node: NodeT) -> tuple[NodeT, ...]: """ Returns a :class:`tuple` of nodes that are ancestors of *node*. """ - if self.is_root(node): + parent = self.parent(node) + if parent is None: # => root return () - parent = self._child_to_parent[node] - assert parent is not None - return (parent, *self.ancestors(parent)) def parent(self, node: NodeT) -> NodeT | None: