GetTimestamp fix.

NVIDIA · Jan 3, 2024 · 02eac1d · 02eac1d
1 parent e4237f9
commit 02eac1d
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/sql-plugin/src/main/scala/org/apache/spark/sql/rapids/datetimeExpressions.scala b/sql-plugin/src/main/scala/org/apache/spark/sql/rapids/datetimeExpressions.scala
@@ -856,7 +856,7 @@ abstract class GpuToTimestamp
     val tmp = lhs.dataType match {
       case _: StringType =>
         // rhs is ignored we already parsed the format
-        if (getTimeParserPolicy == LegacyTimeParserPolicy) {
+        val res = if (getTimeParserPolicy == LegacyTimeParserPolicy) {
           parseStringAsTimestampWithLegacyParserPolicy(
             lhs,
             sparkFormat,
@@ -871,6 +871,11 @@ abstract class GpuToTimestamp
             DType.TIMESTAMP_MICROSECONDS,
             failOnError)
         }
+        if (GpuOverrides.isUTCTimezone(zoneId)) {
+          res
+        } else {
+          GpuTimeZoneDB.fromTimestampToUtcTimestamp(res, zoneId)
+        }
       case _: DateType =>
         timeZoneId match {
           case Some(_) =>