Use original input for null check rather than sanitized input

Signed-off-by: Andy Grove <andygrove@nvidia.com>
NVIDIA · Jan 20, 2021 · 55db986 · 55db986
1 parent 0a99b00
commit 55db986
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/sql-plugin/src/main/scala/com/nvidia/spark/rapids/GpuCast.scala b/sql-plugin/src/main/scala/com/nvidia/spark/rapids/GpuCast.scala
@@ -791,7 +791,7 @@ case class GpuCast(
       // When ANSI mode is enabled, we need to throw an exception if any values could not be
       // converted
       if (ansiMode) {
-        val wasNotNull = withResource(sanitizedInput.isNull()) { wasNull =>
+        val wasNotNull = withResource(input.isNull()) { wasNull =>
           wasNull.not()
         }
         withResource(wasNotNull) { wasNotNull =>