[XLA] Slightly improve a CHECK-fail message.

PiperOrigin-RevId: 180855522

diff --git a/tensorflow/compiler/xla/service/buffer_assignment.cc b/tensorflow/compiler/xla/service/buffer_assignment.cc
index 7ece79d781acfaffc21d6a29e8a12e68622a1617..b01b6587689df8dc071a2004923a4be7ac8fee05 100644 (file)
--- a/tensorflow/compiler/xla/service/buffer_assignment.cc
+++ b/tensorflow/compiler/xla/service/buffer_assignment.cc
@@ -75,7 +75,7 @@ void BufferAllocation::AddAssignment(const LogicalBuffer& buffer, int64 offset,
   CHECK_LE(offset + size, size_)
       << "LogicalBuffer " << buffer << " size out of range";
   CHECK_EQ(buffer.color(), color())
-      << "Buffer color " << buffer.color()
+      << "Buffer color " << buffer.color() << " for buffer " << buffer
       << " does not match allocation color " << color() << ".";
   OffsetSize offset_size;
   offset_size.offset = offset;