one log for GPU reuse fixup

Signed-off-by: Firestarman <[email protected]>
firestarman · Dec 20, 2023 · 4447886 · 4447886
1 parent 5cd0dd6
commit 4447886
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/sql-plugin/src/main/scala/com/nvidia/spark/rapids/GpuTransitionOverrides.scala b/sql-plugin/src/main/scala/com/nvidia/spark/rapids/GpuTransitionOverrides.scala
@@ -694,7 +694,7 @@ class GpuTransitionOverrides extends Rule[SparkPlan] {
           }.getOrElse(g)
       }
     }
-
+    logInfo(s"==> start fixupAdaptiveExchangeReuse, input plan is: \n    $p")
     // If an exchange is at the top of the plan being remapped, this is likely due to AQE
     // re-planning, and we're not allowed to change an exchange to a reused exchange in that case.
     p match {

diff --git a/sql-plugin/src/main/scala/com/nvidia/spark/rapids/SupportPlanningMark.scala b/sql-plugin/src/main/scala/com/nvidia/spark/rapids/SupportPlanningMark.scala
@@ -22,10 +22,10 @@ import scala.ref.WeakReference
 import org.apache.spark.internal.Logging
 import org.apache.spark.sql.execution.exchange.Exchange
 
-/**A trait mixed with a GPU version of Exchange to support to mark if it is optimized by GPU*/
+/** A trait mixed with a GPU version of Exchange to support to mark if it is optimized by GPU */
 trait SupportPlanningMark extends Logging { this: Exchange =>
 
-  // Some GPU Shuffle exchanges are used internally without a CPU instance, it is an Option.
+  // Some GPU Shuffle exchanges are used internally without a CPU instance, So it is an Option.
   val cpuCanonicalExec: Option[Exchange]
 
   private var _isGpuPlanningComplete = false