Skip to content

Commit ce76e9e

Browse files
committed
chore: Improvements of fallback info
1 parent 8f9d276 commit ce76e9e

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
@@ -585,8 +585,7 @@ case class CometExecRule(session: SparkSession) extends Rule[SparkPlan] {
585585

586586
case op =>
587587
op match {
588-
case _: CometExec | _: AQEShuffleReadExec | _: BroadcastExchangeExec |
589-
_: CometBroadcastExchangeExec | _: CometShuffleExchangeExec |
588+
case _: CometPlan | _: AQEShuffleReadExec | _: BroadcastExchangeExec |
590589
_: BroadcastQueryStageExec | _: AdaptiveSparkPlanExec =>
591590
// Some execs should never be replaced. We include
592591
// these cases specially here so we do not add a misleading 'info' message

0 commit comments

Comments
 (0)