Fix leak when concatenating batches

NVIDIA · Sep 14, 2021 · a626e65 · a626e65
1 parent 9786055
commit a626e65
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/sql-plugin/src/main/scala/com/nvidia/spark/rapids/aggregate.scala b/sql-plugin/src/main/scala/com/nvidia/spark/rapids/aggregate.scala
@@ -637,7 +637,9 @@ class GpuHashAggregateIterator(
         c => batchesToConcat.head.column(c).dataType
       }.toArray
       withResource(batchesToConcat.map(GpuColumnVector.from)) { tbl =>
-        GpuColumnVector.from(cudf.Table.concatenate(tbl:_*), dataTypes)
+        withResource(cudf.Table.concatenate(tbl: _*)) { concatenated =>
+          GpuColumnVector.from(concatenated, dataTypes)
+        }
       }
     }
   }