@@ -30,7 +30,7 @@ import org.apache.spark.sql.catalyst.rules.Rule
3030import org .apache .spark .sql .comet ._
3131import org .apache .spark .sql .comet .execution .shuffle .{CometColumnarShuffle , CometNativeShuffle , CometShuffleExchangeExec , CometShuffleManager }
3232import org .apache .spark .sql .execution ._
33- import org .apache .spark .sql .execution .adaptive .{AQEShuffleReadExec , BroadcastQueryStageExec , ShuffleQueryStageExec }
33+ import org .apache .spark .sql .execution .adaptive .{AdaptiveSparkPlanExec , AQEShuffleReadExec , BroadcastQueryStageExec , ShuffleQueryStageExec }
3434import org .apache .spark .sql .execution .aggregate .{BaseAggregateExec , HashAggregateExec , ObjectHashAggregateExec }
3535import org .apache .spark .sql .execution .exchange .{BroadcastExchangeExec , ReusedExchangeExec , ShuffleExchangeExec }
3636import org .apache .spark .sql .execution .joins .{BroadcastHashJoinExec , ShuffledHashJoinExec , SortMergeJoinExec }
@@ -534,7 +534,8 @@ case class CometExecRule(session: SparkSession) extends Rule[SparkPlan] {
534534 case op =>
535535 op match {
536536 case _ : CometExec | _ : AQEShuffleReadExec | _ : BroadcastExchangeExec |
537- _ : CometBroadcastExchangeExec | _ : CometShuffleExchangeExec =>
537+ _ : CometBroadcastExchangeExec | _ : CometShuffleExchangeExec |
538+ _ : BroadcastQueryStageExec | _ : AdaptiveSparkPlanExec =>
538539 // Some execs should never be replaced. We include
539540 // these cases specially here so we do not add a misleading 'info' message
540541 op
@@ -799,7 +800,7 @@ case class CometExecRule(session: SparkSession) extends Rule[SparkPlan] {
799800 }
800801
801802 if (! isCometPlan(s.child)) {
802- withInfo(s, s " Child ${s. child.getClass.getName} is not native " )
803+ // we do not need to report a fallback reason if the child plan is not a Comet plan
803804 return false
804805 }
805806
0 commit comments