Add GpuSubstring handling to SortOrder canonicalization (#1302)

Signed-off-by: Jason Lowe <jlowe@nvidia.com>
NVIDIA · Dec 7, 2020 · 4cb4f5b · 4cb4f5b
1 parent 6a53452
commit 4cb4f5b
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/sql-plugin/src/main/scala/com/nvidia/spark/rapids/GpuOverrides.scala b/sql-plugin/src/main/scala/com/nvidia/spark/rapids/GpuOverrides.scala
@@ -413,6 +413,12 @@ object GpuOverrides {
         g.qualifier,
         g.explicitMetadata)
           .canonicalized
+    case g: GpuSubstring =>
+      Substring(
+        canonicalizeToCpuForSortOrder(g.str),
+        canonicalizeToCpuForSortOrder(g.pos),
+        canonicalizeToCpuForSortOrder(g.len))
+          .canonicalized
     case o: GpuExpression =>
       throw new IllegalStateException(s"${o.getClass} is not expected to be a part of a SortOrder")
     case other => other.canonicalized