fix: Avoid resursive call to canonicalizePlans

apache · Jun 17, 2024 · 3b3de17 · 3b3de17
1 parent c87fabe
commit 3b3de17
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/spark/src/main/scala/org/apache/spark/sql/comet/operators.scala b/spark/src/main/scala/org/apache/spark/sql/comet/operators.scala
@@ -381,7 +381,8 @@ abstract class CometNativeExec extends CometExec {
    */
   protected def canonicalizePlans(): CometNativeExec = {
     def transform(arg: Any): AnyRef = arg match {
-      case sparkPlan: SparkPlan => sparkPlan.canonicalized
+      case sparkPlan: SparkPlan if !sparkPlan.isInstanceOf[CometNativeExec] =>
+        sparkPlan.canonicalized
       case other: AnyRef => other
       case null => null
     }