diff options
Diffstat (limited to 'tensorflow/core/debug/grpc_session_debug_test.cc')
| -rw-r--r-- | tensorflow/core/debug/grpc_session_debug_test.cc | 5 |
1 files changed, 4 insertions, 1 deletions
diff --git a/tensorflow/core/debug/grpc_session_debug_test.cc b/tensorflow/core/debug/grpc_session_debug_test.cc index 3827596a67..d6f35fe24c 100644 --- a/tensorflow/core/debug/grpc_session_debug_test.cc +++ b/tensorflow/core/debug/grpc_session_debug_test.cc @@ -279,9 +279,12 @@ TEST_F(GrpcSessionDebugTest, MultiDevices_String) { DeleteDumpDir(); } else { + // CUDA and SYCL devices do not have an Identity op for strings LOG(ERROR) << "Error: " << s; ASSERT_TRUE((a_dev.device_type() == DEVICE_GPU) || - (b_dev.device_type() == DEVICE_GPU)); + (a_dev.device_type() == DEVICE_SYCL) || + (b_dev.device_type() == DEVICE_GPU) || + (b_dev.device_type() == DEVICE_SYCL)); ASSERT_FALSE(s.ok()); } } |