Could not stop when not in loop boundary

2026-10-22 10:50:32 +02:00
parent 62f20a157c
commit cb39326b40
1 changed files with 1 additions and 0 deletions
@@ -362,6 +362,7 @@ class UnifiedMathVisitor(MathExprVisitor):
        self._owned_tensors = weakref.WeakValueDictionary()

    def visit(self, tree):
+        throw_exception_if_processing_interrupted()
        if tree is None:
            return None