more

viirya · Sep 17, 2024 · 902587c · 902587c
1 parent fd7be01
commit 902587c
Showing 1 changed file with 10 additions and 1 deletion.
diff --git a/spark/src/main/scala/org/apache/spark/sql/comet/CometRowToColumnarExec.scala b/spark/src/main/scala/org/apache/spark/sql/comet/CometRowToColumnarExec.scala
@@ -19,6 +19,7 @@
 
 package org.apache.spark.sql.comet
 
+import org.apache.spark.broadcast
 import org.apache.spark.rdd.RDD
 import org.apache.spark.sql.catalyst.InternalRow
 import org.apache.spark.sql.catalyst.expressions.{Attribute, SortOrder, UnsafeRow}
@@ -66,9 +67,17 @@ case class CometRowToColumnarExec(override val output: Seq[Attribute], child: Sp
 
   override def hashCode(): Int = Objects.hashCode(output, child)
 
-  override protected def doExecute(): RDD[InternalRow] = throw new UnsupportedOperationException
+  override def doExecute(): RDD[InternalRow] = {
+    child.execute()
+  }
+
+  override def doExecuteBroadcast[T](): broadcast.Broadcast[T] = {
+    child.doExecuteBroadcast()
+  }
 
   override protected def doExecuteColumnar(): RDD[ColumnarBatch] = {
+    // scalastyle:off println
+    println("CometRowToColumnarExec")
     child
       .execute()
       .mapPartitionsInternal(iter => {