diff options
author | Kanglan Tang <kanglan@google.com> | 2024-02-20 17:55:58 -0800 |
---|---|---|
committer | TensorFlow Release Automation <jenkins@tensorflow.org> | 2024-02-21 07:29:32 +0000 |
commit | ca6afcc61c167c9401486237b790a48b912f9802 (patch) | |
tree | 1c04bbe5796f96ca5f380ff2b85333baad0ba411 | |
parent | 70108028c9123bb591a147847397c4ac64770a65 (diff) | |
download | tensorflow-upstream-r2.16-a0d7ace8e2f.tar.gz |
Update libtpu index and release versionupstream-r2.16-a0d7ace8e2f
PiperOrigin-RevId: 608803465
-rw-r--r-- | ci/official/envs/linux_x86_tpu | 2 | ||||
-rw-r--r-- | tensorflow/tools/pip_package/setup.py | 9 | ||||
-rw-r--r-- | tensorflow/tools/pip_package/v2/setup.py | 9 |
3 files changed, 13 insertions, 7 deletions
diff --git a/ci/official/envs/linux_x86_tpu b/ci/official/envs/linux_x86_tpu index 8a3cbe271c5..3c7d61b2ac3 100644 --- a/ci/official/envs/linux_x86_tpu +++ b/ci/official/envs/linux_x86_tpu @@ -19,4 +19,4 @@ TFCI_BUILD_PIP_PACKAGE_ARGS="--repo_env=WHEEL_NAME=tensorflow_tpu" TFCI_LIB_SUFFIX="-tpu-linux-x86_64" TFCI_WHL_BAZEL_TEST_ENABLE=0 TFCI_WHL_SIZE_LIMIT=580M -TFCI_PYTHON_VERIFY_PIP_INSTALL_ARGS="-f https://storage.googleapis.com/libtpu-releases/index.html" +TFCI_PYTHON_VERIFY_PIP_INSTALL_ARGS="-f https://storage.googleapis.com/libtpu-tf-releases/index.html" diff --git a/tensorflow/tools/pip_package/setup.py b/tensorflow/tools/pip_package/setup.py index a55424c4198..e2c71d2f83f 100644 --- a/tensorflow/tools/pip_package/setup.py +++ b/tensorflow/tools/pip_package/setup.py @@ -330,19 +330,22 @@ for path in so_lib_paths: # $ pip install <tf-tpu project> -f https://storage.googleapis.com/libtpu-releases/index.html # libtpu is built and uploaded to this link every night (PST). if '_tpu' in project_name: - # For tensorflow-tpu releases, use a set libtpu-nightly version; + # For tensorflow-tpu releases, use a set libtpu version; # For tf-nightly-tpu, use the most recent libtpu-nightly. Because of the # timing of these tests, the UTC date from eight hours ago is expected to be a # valid version. _libtpu_version = standard_or_nightly( - '0.1.dev20231018', + '2.16.0rc0', '0.1.dev' + ( datetime.datetime.now(tz=datetime.timezone.utc) - datetime.timedelta(hours=8) ).strftime('%Y%m%d'), ) - REQUIRED_PACKAGES.append([f'libtpu-nightly=={_libtpu_version}']) + if _libtpu_version.startswith('0.1'): + REQUIRED_PACKAGES.append([f'libtpu-nightly=={_libtpu_version}']) + else: + REQUIRED_PACKAGES.append([f'libtpu=={_libtpu_version}']) CONSOLE_SCRIPTS.extend([ 'start_grpc_tpu_worker = tensorflow.python.tools.grpc_tpu_worker:run', ('start_grpc_tpu_service = ' diff --git a/tensorflow/tools/pip_package/v2/setup.py b/tensorflow/tools/pip_package/v2/setup.py index 3833f38c828..53a1eb4e1c5 100644 --- a/tensorflow/tools/pip_package/v2/setup.py +++ b/tensorflow/tools/pip_package/v2/setup.py @@ -317,19 +317,22 @@ for path in so_lib_paths: # https://storage.googleapis.com/libtpu-releases/index.html # libtpu is built and uploaded to this link every night (PST). if '_tpu' in project_name: - # For tensorflow-tpu releases, use a set libtpu-nightly version; + # For tensorflow-tpu releases, use a set libtpu version; # For tf-nightly-tpu, use the most recent libtpu-nightly. Because of the # timing of these tests, the UTC date from eight hours ago is expected to be a # valid version. _libtpu_version = standard_or_nightly( - '0.1.dev20231018', + '2.16.0rc0', '0.1.dev' + ( datetime.datetime.now(tz=datetime.timezone.utc) - datetime.timedelta(hours=8) ).strftime('%Y%m%d'), ) - REQUIRED_PACKAGES.append([f'libtpu-nightly=={_libtpu_version}']) + if _libtpu_version.startswith('0.1'): + REQUIRED_PACKAGES.append([f'libtpu-nightly=={_libtpu_version}']) + else: + REQUIRED_PACKAGES.append([f'libtpu=={_libtpu_version}']) CONSOLE_SCRIPTS.extend([ 'start_grpc_tpu_worker = tensorflow.python.tools.grpc_tpu_worker:run', ('start_grpc_tpu_service = ' |