[VL] Following #6959, Fix leak memory dump is not correctly printed (#…

…6985)
apache · Aug 23, 2024 · 50ca8a0 · 50ca8a0
1 parent d12bf1f
commit 50ca8a0
Showing 1 changed file with 8 additions and 7 deletions.
diff --git a/gluten-data/src/main/scala/org/apache/gluten/runtime/Runtime.scala b/gluten-data/src/main/scala/org/apache/gluten/runtime/Runtime.scala
@@ -98,16 +98,17 @@ object Runtime {
           s"Runtime instance already released: $handle, ${resourceName()}, ${priority()}")
       }
 
-      def dump(): KnownNameAndStats = {
-        new KnownNameAndStats() {
-          override def name: String = resourceName()
-          override def stats: MemoryUsageStats = collectMemoryUsage()
-        }
+      def dump(): String = {
+        SparkMemoryUtil.prettyPrintStats(
+          s"[${resourceName()}]",
+          new KnownNameAndStats() {
+            override def name: String = resourceName()
+            override def stats: MemoryUsageStats = collectMemoryUsage()
+          })
       }
 
       if (LOGGER.isDebugEnabled) {
-        LOGGER.debug(
-          SparkMemoryUtil.prettyPrintStats("About to release memory manager, usage dump:", dump()))
+        LOGGER.debug("About to release memory manager, " + dump())
       }
 
       RuntimeJniWrapper.releaseRuntime(handle)