Fix

apache · Mar 29, 2024 · 2c4597a · 2c4597a
1 parent c804574
commit 2c4597a
Showing 1 changed file with 7 additions and 3 deletions.
diff --git a/spark/src/main/scala/org/apache/comet/CometSparkSessionExtensions.scala b/spark/src/main/scala/org/apache/comet/CometSparkSessionExtensions.scala
@@ -459,9 +459,13 @@ class CometSparkSessionExtensions
               }
             case other => other
           }
-          val newPlan = transform(plan.withNewChildren(newChildren))
-          if (isCometNative(newPlan) || isCometBroadCastForceEnabled(conf)) {
-            newPlan
+          if (!newChildren.exists(_.isInstanceOf[BroadcastExchangeExec])) {
+            val newPlan = transform(plan.withNewChildren(newChildren))
+            if (isCometNative(newPlan) || isCometBroadCastForceEnabled(conf)) {
+              newPlan
+            } else {
+              plan
+            }
           } else {
             plan
           }