Skip to content

Commit 657cac8

Browse files
authored
chore: Improvements of fallback info (#2450)
1 parent e205ab8 commit 657cac8

File tree

2 files changed

+3
-2
lines changed

2 files changed

+3
-2
lines changed

spark/src/main/scala/org/apache/comet/ExtendedExplainInfo.scala

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -25,6 +25,7 @@ import org.apache.spark.sql.ExtendedExplainGenerator
2525
import org.apache.spark.sql.catalyst.trees.{TreeNode, TreeNodeTag}
2626
import org.apache.spark.sql.execution.{InputAdapter, SparkPlan, WholeStageCodegenExec}
2727
import org.apache.spark.sql.execution.adaptive.{AdaptiveSparkPlanExec, QueryStageExec}
28+
import org.apache.spark.sql.execution.exchange.ReusedExchangeExec
2829

2930
import org.apache.comet.CometExplainInfo.getActualPlan
3031

@@ -158,6 +159,7 @@ object CometExplainInfo {
158159
case p: InputAdapter => getActualPlan(p.child)
159160
case p: QueryStageExec => getActualPlan(p.plan)
160161
case p: WholeStageCodegenExec => getActualPlan(p.child)
162+
case p: ReusedExchangeExec => getActualPlan(p.child)
161163
case p => p
162164
}
163165
}

spark/src/main/scala/org/apache/comet/rules/CometExecRule.scala

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -535,8 +535,7 @@ case class CometExecRule(session: SparkSession) extends Rule[SparkPlan] {
535535

536536
case op =>
537537
op match {
538-
case _: CometExec | _: AQEShuffleReadExec | _: BroadcastExchangeExec |
539-
_: CometBroadcastExchangeExec | _: CometShuffleExchangeExec |
538+
case _: CometPlan | _: AQEShuffleReadExec | _: BroadcastExchangeExec |
540539
_: BroadcastQueryStageExec | _: AdaptiveSparkPlanExec =>
541540
// Some execs should never be replaced. We include
542541
// these cases specially here so we do not add a misleading 'info' message

0 commit comments

Comments
 (0)