Clarify CPU/GPU infeed error messages.
PiperOrigin-RevId: 312254085
Change-Id: Ic981d72bf59e41b149cf0036a272250d4ea482a3
diff --git a/tensorflow/compiler/xla/service/cpu/cpu_transfer_manager.cc b/tensorflow/compiler/xla/service/cpu/cpu_transfer_manager.cc
index fae9670..e21ed7a 100644
--- a/tensorflow/compiler/xla/service/cpu/cpu_transfer_manager.cc
+++ b/tensorflow/compiler/xla/service/cpu/cpu_transfer_manager.cc
@@ -154,7 +154,8 @@
int64 size,
const void* source) {
if (size > std::numeric_limits<int32>::max()) {
- return InvalidArgument("Infeed shape is too large: needs %d bytes", size);
+ return InvalidArgument("CPU infeed of %d bytes exceeds maximum of %d bytes",
+ size, std::numeric_limits<int32>::max());
}
if (size <= 0) {
diff --git a/tensorflow/compiler/xla/service/gpu/gpu_transfer_manager.cc b/tensorflow/compiler/xla/service/gpu/gpu_transfer_manager.cc
index 05fa798..cb22b4d 100644
--- a/tensorflow/compiler/xla/service/gpu/gpu_transfer_manager.cc
+++ b/tensorflow/compiler/xla/service/gpu/gpu_transfer_manager.cc
@@ -96,7 +96,8 @@
StatusOr<InfeedBuffer> GpuTransferManager::TransferBufferToInfeedInternal(
se::StreamExecutor* executor, int64 size, const void* source) {
if (size > std::numeric_limits<int32>::max()) {
- return InvalidArgument("Infeed shape is too large: needs %d bytes", size);
+ return InvalidArgument("GPU infeed of %d bytes exceeds maximum of %d bytes",
+ size, std::numeric_limits<int32>::max());
}
if (size == 0) {
