Fix Spark 3.1.0 build for HashJoin changes (NVIDIA#490)

* Fix Spark 3.1.0 build for HashJoin changes Signed-off-by: Thomas Graves <[email protected]> * Fix missing space Signed-off-by: Thomas Graves <[email protected]> Co-authored-by: Thomas Graves <[email protected]>
nartal1 · Jul 31, 2020 · 8ae7e94 · 8ae7e94
1 parent 5236999
commit 8ae7e94
Showing 1 changed file with 12 additions and 0 deletions.
diff --git a/shims/spark310/src/main/scala/com/nvidia/spark/rapids/shims/spark310/GpuHashJoin.scala b/shims/spark310/src/main/scala/com/nvidia/spark/rapids/shims/spark310/GpuHashJoin.scala
@@ -19,7 +19,10 @@ import ai.rapids.cudf.{NvtxColor, Table}
 import com.nvidia.spark.rapids._
 
 import org.apache.spark.TaskContext
+import org.apache.spark.rdd.RDD
+import org.apache.spark.sql.catalyst.InternalRow
 import org.apache.spark.sql.catalyst.expressions.{Attribute, Expression}
+import org.apache.spark.sql.catalyst.expressions.codegen.CodegenContext
 import org.apache.spark.sql.catalyst.optimizer.{BuildLeft, BuildRight}
 import org.apache.spark.sql.catalyst.plans.{ExistenceJoin, FullOuter, InnerLike, JoinType, LeftAnti, LeftExistence, LeftOuter, LeftSemi, RightOuter}
 import org.apache.spark.sql.execution.joins.HashJoin
@@ -242,4 +245,13 @@ trait GpuHashJoin extends GpuExec with HashJoin {
       joinedTable.close()
     }
   }
+
+  override def inputRDDs(): Seq[RDD[InternalRow]] = {
+    throw new UnsupportedOperationException("inputRDDs is used by codegen which we don't support")
+  }
+
+  protected override def prepareRelation(ctx: CodegenContext): (String, Boolean) = {
+    throw new UnsupportedOperationException(
+      "prepareRelation is used by codegen which we don't support")
+  }
 }