[GLUTEN-4723]Fix issue of actions dag split (#482)

(cherry picked from commit 21fa961)
Kyligence · Dec 17, 2024 · 2048027 · 2048027
1 parent e0ab4d8
commit 2048027
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/src/Interpreters/ActionsDAG.cpp b/src/Interpreters/ActionsDAG.cpp
@@ -1874,6 +1874,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)
     {