memory leak

Signed-off-by: Haoyang Li <[email protected]>
NVIDIA · Dec 5, 2024 · 14900dd · 14900dd
1 parent cedcd58
commit 14900dd
Showing 1 changed file with 4 additions and 6 deletions.
diff --git a/sql-plugin/src/main/scala/org/apache/spark/sql/rapids/stringFunctions.scala b/sql-plugin/src/main/scala/org/apache/spark/sql/rapids/stringFunctions.scala
@@ -1249,13 +1249,11 @@ case class GpuContainsAny(input: Expression, targets: Seq[UTF8String])
     val boolCvs = withResource(ColumnVector.fromUTF8Strings(targetsBytes: _*)) { targetsCv =>
       input.getBase.stringContains(targetsCv)
     }
-    // boolCvs is a sequence of ColumnVectors, we need to OR them together
-    boolCvs.reduce {
-      (cv1, cv2) => withResource(cv1) { cv1 =>
-        withResource(cv2) { cv2 =>
-          cv1.or(cv2)
-        }
+    withResource(boolCvs) { _ =>
+      val falseCv = withResource(Scalar.fromBool(false)) { falseScalar =>
+        ColumnVector.fromScalar(falseScalar, input.getRowCount.toInt)
       }
+      boolCvs.foldLeft(falseCv)((l, r) => withResource(l) { _ => l.or(r)})
     }
   }
 }