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

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