From df70407e547f88ec9fc3d750005449ff109bb934 Mon Sep 17 00:00:00 2001 From: JackCaoG <59073027+JackCaoG@users.noreply.github.com> Date: Fri, 22 Apr 2022 11:50:19 -0700 Subject: [PATCH 1/2] Revert "Revert "Update tf submodule to 04/07/2022 (#3507)" (#3522)" This reverts commit da2d988794cc2b1fe420f14689a5cc91843793e8. --- third_party/tensorflow | 2 +- third_party/xla_client/xrt_computation_client.cc | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/third_party/tensorflow b/third_party/tensorflow index fe5345ed0a56..ace2ad8d5a94 160000 --- a/third_party/tensorflow +++ b/third_party/tensorflow @@ -1 +1 @@ -Subproject commit fe5345ed0a56e9b6deb9b46b0ab1a99fc8934547 +Subproject commit ace2ad8d5a9473f49547cd2fc70064dd54c67de1 diff --git a/third_party/xla_client/xrt_computation_client.cc b/third_party/xla_client/xrt_computation_client.cc index 35d0ed2733ea..7f56603a438b 100644 --- a/third_party/xla_client/xrt_computation_client.cc +++ b/third_party/xla_client/xrt_computation_client.cc @@ -25,7 +25,7 @@ #include "tensorflow/compiler/xrt/xrt_util.h" #include "tensorflow/core/framework/allocator.h" #include "tensorflow/core/profiler/lib/traceme.h" -#include "tensorflow/core/tpu/tpu_api_dlsym_initializer.h" +#include "tensorflow/core/tpu/tpu_initializer_helper.h" #include "tensorflow/core/util/device_name_utils.h" namespace xla { From e8c48395f9b036b479a5e9a20cec79f94b019544 Mon Sep 17 00:00:00 2001 From: JackCaoG Date: Thu, 21 Apr 2022 23:25:43 +0000 Subject: [PATCH 2/2] Update libtpu link --- setup.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/setup.py b/setup.py index 8139b47aa1b5..10fe761885a0 100644 --- a/setup.py +++ b/setup.py @@ -53,7 +53,7 @@ base_dir = os.path.dirname(os.path.abspath(__file__)) third_party_path = os.path.join(base_dir, 'third_party') -_libtpu_version = '0.1.dev20220303' +_libtpu_version = '0.1.dev20220413' _litbpu_storage_path = f'https://storage.googleapis.com/cloud-tpu-tpuvm-artifacts/wheels/libtpu-nightly/libtpu_nightly-{_libtpu_version}-py3-none-any.whl'