File tree Expand file tree Collapse file tree 2 files changed +3
-2
lines changed
spark/src/main/scala/org/apache/comet Expand file tree Collapse file tree 2 files changed +3
-2
lines changed Original file line number Diff line number Diff line change @@ -25,6 +25,7 @@ import org.apache.spark.sql.ExtendedExplainGenerator
2525import org .apache .spark .sql .catalyst .trees .{TreeNode , TreeNodeTag }
2626import org .apache .spark .sql .execution .{InputAdapter , SparkPlan , WholeStageCodegenExec }
2727import org .apache .spark .sql .execution .adaptive .{AdaptiveSparkPlanExec , QueryStageExec }
28+ import org .apache .spark .sql .execution .exchange .ReusedExchangeExec
2829
2930import 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 }
Original file line number Diff line number Diff 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
You can’t perform that action at this time.
0 commit comments