memory viewer external fragmentation shouldn't include padding.

PiperOrigin-RevId: 557830844
tensorflow · Aug 17, 2023 · 9106124 · 9106124
1 parent a9ddc74
commit 9106124
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/frontend/app/components/memory_viewer/memory_usage/memory_usage.ts b/frontend/app/components/memory_viewer/memory_usage/memory_usage.ts
@@ -141,8 +141,8 @@ export class MemoryUsage {
         (preprocess.entryComputationParametersMib || 0) * 1024 * 1024;
     this.hloTempSizeBytes = this.peakHeapSizeBytes -
         (preprocess.indefiniteBufferAllocationMib || 0) * 1024 * 1024;
-    const framenentationSizeBytes = this.peakHeapSizeBytes -
-        (preprocess.peakUnpaddedHeapMib || 0) * 1024 * 1024;
+    const framenentationSizeBytes =
+        this.peakHeapSizeBytes - (preprocess.peakHeapMib || 0) * 1024 * 1024;
     if (this.hloTempSizeBytes) {
       this.hloTempFragmentation =
           framenentationSizeBytes / this.hloTempSizeBytes;