aboutsummaryrefslogtreecommitdiffhomepage
path: root/tensorflow/core/common_runtime/simple_placer_test.cc
diff options
context:
space:
mode:
authorGravatar A. Unique TensorFlower <gardener@tensorflow.org>2016-08-05 10:21:36 -0800
committerGravatar TensorFlower Gardener <gardener@tensorflow.org>2016-08-05 11:32:27 -0700
commitd2b5fb93820c981f6b7f67e46f8839fa79b26913 (patch)
treeae23fba7a60c8f59312766661cf3b38e4c1a6cfd /tensorflow/core/common_runtime/simple_placer_test.cc
parent27a4509d91a6087b2d0103c6abff579ed093a112 (diff)
More verbose errors for unregistered kernels.
Change: 129462457
Diffstat (limited to 'tensorflow/core/common_runtime/simple_placer_test.cc')
-rw-r--r--tensorflow/core/common_runtime/simple_placer_test.cc3
1 files changed, 3 insertions, 0 deletions
diff --git a/tensorflow/core/common_runtime/simple_placer_test.cc b/tensorflow/core/common_runtime/simple_placer_test.cc
index 64442622dc..e17a40d720 100644
--- a/tensorflow/core/common_runtime/simple_placer_test.cc
+++ b/tensorflow/core/common_runtime/simple_placer_test.cc
@@ -856,6 +856,8 @@ TEST_F(SimplePlacerTest, TestNoKernelsRegistered) {
StringPiece(s.error_message())
.contains(
"No OpKernel was registered to support Op 'VariableNoKernels'"));
+ EXPECT_TRUE(
+ StringPiece(s.error_message()).contains("<no registered kernels>"));
}
// Test that placement fails when a kernel is registered but no known
@@ -878,6 +880,7 @@ TEST_F(SimplePlacerTest, TestNoDevicesRegistered) {
EXPECT_TRUE(StringPiece(s.error_message())
.contains("No OpKernel was registered to support "
"Op 'VariableGPU'"));
+ EXPECT_TRUE(StringPiece(s.error_message()).contains("device='GPU'"));
}
// Test that placement fails when a requested device is malformed.