Update GpuInSet for SPARK-35422 changes (#3190)

Signed-off-by: Jason Lowe <jlowe@nvidia.com>
NVIDIA · Aug 11, 2021 · 3a8dfcd · 3a8dfcd
1 parent 960cef1
commit 3a8dfcd
Showing 1 changed file with 13 additions and 2 deletions.
diff --git a/sql-plugin/src/main/scala/com/nvidia/spark/rapids/GpuInSet.scala b/sql-plugin/src/main/scala/com/nvidia/spark/rapids/GpuInSet.scala
@@ -49,11 +49,22 @@ case class GpuInSet(
   private def buildNeedles: ColumnVector =
     GpuScalar.columnVectorFromLiterals(list, child.dataType)
 
-  override def toString: String = s"$child INSET ${list.mkString("(", ",", ")")}"
+  override def toString: String = {
+    val listString = list
+        .map(elem => Literal(elem, child.dataType).toString)
+        // Sort elements for deterministic behaviours
+        .sorted
+        .mkString(", ")
+    s"$child INSET $listString"
+  }
 
   override def sql: String = {
     val valueSQL = child.sql
-    val listSQL = list.map(Literal(_).sql).mkString(", ")
+    val listSQL = list
+        .map(elem => Literal(elem, child.dataType).sql)
+        // Sort elements for deterministic behaviours
+        .sorted
+        .mkString(", ")
     s"($valueSQL IN ($listSQL))"
   }
 }