From 117bcd9cb5f3e55ce1fcc09a0bb4963c32bad8ce Mon Sep 17 00:00:00 2001 From: Rohan Jain Date: Wed, 1 Nov 2017 16:15:20 -0700 Subject: Adding support for local device names for ProcessFLR. Now one can specify a remote target as /device:CPU:0 or /device:GPU:0 etc. PiperOrigin-RevId: 174252575 --- tensorflow/core/common_runtime/function_test.cc | 5 ++--- 1 file changed, 2 insertions(+), 3 deletions(-) (limited to 'tensorflow/core/common_runtime/function_test.cc') diff --git a/tensorflow/core/common_runtime/function_test.cc b/tensorflow/core/common_runtime/function_test.cc index b77a8f50c4..d183bf7c97 100644 --- a/tensorflow/core/common_runtime/function_test.cc +++ b/tensorflow/core/common_runtime/function_test.cc @@ -939,9 +939,8 @@ TEST_F(FunctionLibraryRuntimeTest, Gradient_AddSum) { TEST_F(FunctionLibraryRuntimeTest, CrossDevice) { Init({test::function::FindDevice()}); FunctionLibraryRuntime::Handle handle; - TF_CHECK_OK(Instantiate( - flr0_, "FindDevice", - {{"_target", "/job:localhost/replica:0/task:0/cpu:1"}}, &handle)); + TF_CHECK_OK(Instantiate(flr0_, "FindDevice", {{"_target", "/device:CPU:1"}}, + &handle)); Tensor y; FunctionLibraryRuntime::Options opts; -- cgit v1.2.3