From 9bef7eea2aff255ff92790bdbabf0d750e51e923 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Vladim=C3=ADr=20=C5=A0till?= <git@vstill.eu>
Date: Tue, 19 Nov 2024 16:26:31 +0100
Subject: [PATCH] Fix modifyAllMatching visitor helper (#5020)
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

Signed-off-by: Vladimír Štill <vladimir.still@intel.com>
---
 ir/visitor.h | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/ir/visitor.h b/ir/visitor.h
index 9b935ab6c18..c203e99892c 100644
--- a/ir/visitor.h
+++ b/ir/visitor.h
@@ -820,7 +820,7 @@ const RootType *modifyAllMatching(const RootType *root, Func &&function) {
         Func function;
         void postorder(NodeType *node) override { function(node); }
     };
-    return root->apply(NodeVisitor(std::forward<Func>(function)));
+    return root->apply(NodeVisitor(std::forward<Func>(function)))->template checkedTo<RootType>();
 }
 
 /**