From 0f9215244016a51ab1b933366e710125d3b62f77 Mon Sep 17 00:00:00 2001 From: Jason Lowe Date: Mon, 7 Dec 2020 17:38:49 -0600 Subject: [PATCH] Add GpuSubstring handling to SortOrder canonicalization (#1302) Signed-off-by: Jason Lowe --- .../main/scala/com/nvidia/spark/rapids/GpuOverrides.scala | 6 ++++++ 1 file changed, 6 insertions(+) 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 index cc15b76b2d2..7a951a9d603 100644 --- 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