opt ancestors

inducer · Nov 28, 2024 · 9726d9b · 9726d9b
1 parent 706e873
commit 9726d9b
Showing 1 changed file with 2 additions and 4 deletions.
diff --git 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: