From 17a0742e9295ceb1755138c209f6c044c5a32f8d Mon Sep 17 00:00:00 2001 From: Jason Lowe Date: Thu, 10 Aug 2023 14:59:12 -0500 Subject: [PATCH] Fix host memory buffer leaks in SerializationSuite Signed-off-by: Jason Lowe --- .../nvidia/spark/rapids/SerializationSuite.scala | 14 ++++++++------ 1 file changed, 8 insertions(+), 6 deletions(-) diff --git a/tests/src/test/scala/com/nvidia/spark/rapids/SerializationSuite.scala b/tests/src/test/scala/com/nvidia/spark/rapids/SerializationSuite.scala index e4fd7917a6e..56ecb1a8c57 100644 --- a/tests/src/test/scala/com/nvidia/spark/rapids/SerializationSuite.scala +++ b/tests/src/test/scala/com/nvidia/spark/rapids/SerializationSuite.scala @@ -154,9 +154,10 @@ class SerializationSuite extends AnyFunSuite withBroadcast(broadcast) { _ => withResource(broadcast.hostBatch) { hostBatch1 => TestUtils.compareBatches(expectedHostBatch, hostBatch1) - val clonedObj = SerializationUtils.clone(broadcast) - withResource(clonedObj.hostBatch) { hostBatch2 => - TestUtils.compareBatches(expectedHostBatch, hostBatch2) + withBroadcast(SerializationUtils.clone(broadcast)) { clonedObj => + withResource(clonedObj.hostBatch) { hostBatch2 => + TestUtils.compareBatches(expectedHostBatch, hostBatch2) + } } } } @@ -176,9 +177,10 @@ class SerializationSuite extends AnyFunSuite // we materialized withResource(broadcast.hostBatch) { hostBatch1 => TestUtils.compareBatches(expectedHostBatch, hostBatch1) - val clonedObj = SerializationUtils.clone(broadcast) - withResource(clonedObj.hostBatch) { hostBatch2 => - TestUtils.compareBatches(expectedHostBatch, hostBatch2) + withBroadcast(SerializationUtils.clone(broadcast)) { clonedObj => + withResource(clonedObj.hostBatch) { hostBatch2 => + TestUtils.compareBatches(expectedHostBatch, hostBatch2) + } } } }