experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Add dependency for the direct TPU driver back into the tpu_driver target

Browse Source 
PiperOrigin-RevId: 295804517
Change-Id: I264b897ef17ff38d0c2a98dec1e6de49f8283556
This commit is contained in:
Frank Chen 2020-02-18 13:10:55 -08:00 committed by TensorFlower Gardener
parent ecb8befb32
commit bdcb2782c3
2 changed files with 3 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
tensorflow/compiler/xla/python/tpu_driver/client/BUILD
View File

@ -22,6 +22,7 @@ cc_library(
"//tensorflow/compiler/xla/python:local_client",
"//tensorflow/compiler/xla/python:semaphore",
"//tensorflow/compiler/xla/python/tpu_driver",
"//tensorflow/compiler/xla/python/tpu_driver:direct_tpu_driver",
"//tensorflow/compiler/xla/python/tpu_driver:grpc_tpu_driver",
"//tensorflow/compiler/xla/python/tpu_driver:recording_tpu_driver",
"//tensorflow/compiler/xla/python/tpu_driver:tpu_driver_proto_cc",

3
tensorflow/compiler/xla/python/tpu_driver/direct_tpu_driver.cc
View File

@ -27,7 +27,8 @@
namespace tpu_driver {
namespace {
// Enable the macro by default in the env where the libtpu.so is available.
// Enable the macro by default in the Google internal environment where the
// libtpu.so is linked in statically.
#ifdef PLATFORM_GOOGLE
#define TPU_SHARED_LIBRARY_COMPILE_LINK 1
#endif