From 1f20030df06646eb103da8a716fefe1548383bdd Mon Sep 17 00:00:00 2001
From: taiyang-li <654010905@qq.com>
Date: Tue, 20 Feb 2024 18:36:07 +0800
Subject: [PATCH] fix issue of actions dag split

---
 src/Interpreters/ActionsDAG.cpp | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/src/Interpreters/ActionsDAG.cpp b/src/Interpreters/ActionsDAG.cpp
index 7240679abb7..324a59a9470 100644
--- a/src/Interpreters/ActionsDAG.cpp
+++ b/src/Interpreters/ActionsDAG.cpp
@@ -1713,6 +1713,10 @@ ActionsDAG::SplitResult ActionsDAG::split(std::unordered_set<const Node *> split
         }
     }
 
+    /// All input nodes are needed by split to make sure first DAG and initial DAG have equal inputs,
+    for (const auto * input: inputs)
+        data[input].needed_by_split_node = true;
+
     /// DFS. Move nodes to one of the DAGs.
     for (const auto & node : nodes)
     {