From 96a58d121a5af7ef956196a6141fe7777277e95e Mon Sep 17 00:00:00 2001 From: Kuhu Shukla Date: Mon, 9 Dec 2024 19:47:01 -0600 Subject: [PATCH] Fix leak in isTimeStamp (#11845) Signed-off-by: Kuhu Shukla --- .../org/apache/spark/sql/rapids/datetimeExpressions.scala | 4 +++- 1 file changed, 3 insertions(+), 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 index 0f382a7b6e6..d08c598cba4 100644 --- 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 @@ -707,7 +707,9 @@ object GpuToTimestamp { case _ => // this is the incompatibleDateFormats case where we do not guarantee compatibility with // Spark and assume that all non-null inputs are valid - ColumnVector.fromScalar(Scalar.fromBool(true), col.getRowCount.toInt) + withResource(Scalar.fromBool(true)) { s => + ColumnVector.fromScalar(s, col.getRowCount.toInt) + } } }