diff options
Diffstat (limited to 'tensorflow/core/distributed_runtime/executor_test.cc')
| -rw-r--r-- | tensorflow/core/distributed_runtime/executor_test.cc | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/tensorflow/core/distributed_runtime/executor_test.cc b/tensorflow/core/distributed_runtime/executor_test.cc index 1a4980a61b..5b115f9a4d 100644 --- a/tensorflow/core/distributed_runtime/executor_test.cc +++ b/tensorflow/core/distributed_runtime/executor_test.cc @@ -140,7 +140,7 @@ Rendezvous::ParsedKey Key(const string& sender, const uint64 incarnation, } #define ALICE "/job:j/replica:0/task:0/cpu:0" -#define BOB "/job:j/replica:0/task:0/gpu:0" +#define BOB "/job:j/replica:0/task:0/device:GPU:0" TEST_F(ExecutorTest, SimpleAdd) { // c = a + b |