commit feadb9ae1f09b1431cf149c66f8a78c40447be63
author Trent Lo <trentl@nvidia.com> Mon Oct 12 17:34:53 2020 -0700
committer Trent Lo <trentl@nvidia.com> Mon Oct 12 17:34:53 2020 -0700
tree daa8bea0eadbb863f9da925e5f1153e323b59d25
parent 438cb49f13863f622dde5df81c968753e4c26c6a

[XLA/GPU] remove unnecessary check.

tensorflow/compiler/xla/service/buffer_assignment.cc

diff --git a/tensorflow/compiler/xla/service/buffer_assignment.cc b/tensorflow/compiler/xla/service/buffer_assignment.cc
index cc7dd13..bab28bb 100644
--- a/tensorflow/compiler/xla/service/buffer_assignment.cc
+++ b/tensorflow/compiler/xla/service/buffer_assignment.cc
@@ -1542,10 +1542,8 @@
     }
     // Compute peak_buffers only when the multiheap mode is off. Simply return
     // an empty vector in the multiheap mode.
-    if (assignment->multiheap_size_constraint_per_heap() == -1) {
-      allocation->peak_buffers_ =
-          ComputePeakMemoryLogicalBuffers(*allocation, result.debug_trace);
-    }
+    allocation->peak_buffers_ =
+        ComputePeakMemoryLogicalBuffers(*allocation, result.debug_trace);
 
     XLA_VLOG_LINES(2, allocation->ToString());