commit 49e4e94f207938f3009175360467f0c8cb3f3a9a
author Frank Chen <frankchn@google.com> Thu Jan 16 17:21:48 2020 -0800
committer TensorFlower Gardener <gardener@tensorflow.org> Thu Jan 16 17:25:37 2020 -0800
tree 59e8823c4e4613d8ea572590837159bf2b5a0abc
parent 6383a35f7ce89292646d286995e0667a2d56ef40

Minor bugfix after a change in DeviceAssignment types

PiperOrigin-RevId: 290178875
Change-Id: I2e4b215668b14d158c4011e2f42a47305f3bed2e

tensorflow/compiler/xla/python/tpu_driver/client/libtpu_client.c

diff --git a/tensorflow/compiler/xla/python/tpu_driver/client/libtpu_client.c b/tensorflow/compiler/xla/python/tpu_driver/client/libtpu_client.c
index d7bdcf3..8e3617d 100644
--- a/tensorflow/compiler/xla/python/tpu_driver/client/libtpu_client.c
+++ b/tensorflow/compiler/xla/python/tpu_driver/client/libtpu_client.c
@@ -114,7 +114,7 @@
         /*eventc=*/1, /*eventv=*/allocate_buf_b_events);
 
   fprintf(stdout, "------ Going to Execute a TPU program ------\n");
-  DeviceAssignment device_assignment = {1, 1};
+  DeviceAssignment device_assignment = {NULL, 0};
   TpuBufferHandle* input_buffer_handle[] = {buf_a_handle, buf_b_handle};
   TpuBufferHandle* output_buffer_handle[] = {buf_sum_handle};
   TpuEvent* transfer_events[] = {transfer_ev1, transfer_ev2};