[SCALA] Fix ColumnarBroadcastExchange didn't fallback issue when DPP …

…is enabled Signed-off-by: Chendi Xue <[email protected]>
oap-project · Jan 4, 2021 · 93f8525 · 93f8525
1 parent cea3f04
commit 93f8525
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/core/src/main/scala/com/intel/oap/ColumnarGuardRule.scala b/core/src/main/scala/com/intel/oap/ColumnarGuardRule.scala
@@ -151,6 +151,10 @@ case class ColumnarGuardRule(conf: SparkConf) extends Rule[SparkPlan] {
    */
   private def insertRowGuardRecursive(plan: SparkPlan): SparkPlan = {
     plan match {
+      case p: ShuffleExchangeExec =>
+        RowGuard(p.withNewChildren(p.children.map(insertRowGuardOrNot)))
+      case p: BroadcastExchangeExec =>
+        RowGuard(p.withNewChildren(p.children.map(insertRowGuardOrNot)))
       case p: ShuffledHashJoinExec =>
         RowGuard(p.withNewChildren(p.children.map(insertRowGuardRecursive)))
       case p if !supportCodegen(p) =>