Enable wscg for agg with child plan supporting codegen

oap-project · Aug 16, 2022 · 59e43e4 · 59e43e4
1 parent cce191a
commit 59e43e4
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/...ne/core/src/main/scala/org/apache/spark/sql/execution/ColumnarCollapseCodegenStages.scala b/...ne/core/src/main/scala/org/apache/spark/sql/execution/ColumnarCollapseCodegenStages.scala
@@ -156,7 +156,8 @@ case class ColumnarCollapseCodegenStages(
         if (containsSubquery(p.condition) || containsSubquery(p.projectList)) =>
       false
     case p: ColumnarCodegenSupport if p.supportColumnarCodegen =>
-      plan.children.map(existsJoins(_, count)).exists(_ == true)
+      if (count >= 1) true
+      else plan.children.map(existsJoins(_, count)).exists(_ == true)
     case _ =>
       false
   }