Undo a piece of code changes

oap-project · Mar 28, 2022 · c91cac0 · c91cac0
1 parent c56ef48
commit c91cac0
Showing 1 changed file with 1 addition 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
@@ -248,7 +248,7 @@ case class ColumnarCollapseCodegenStages(
             .isInstanceOf[ColumnarSortMergeJoinExec]) =>
         // we don't support any ColumnarSortMergeJoin whose both children are ColumnarSortMergeJoin
         j.withNewChildren(j.children.map(c => {
-          if (c.isInstanceOf[ColumnarSortMergeJoinExec]) {
+          if (c.equals(j.buildPlan)) {
             new ColumnarInputAdapter(insertWholeStageCodegen(c))
           } else {
             insertInputAdapter(c)