Merge pull request #87586 from tensorflow/r2.19-2b775610a7b

r2.19 cherry-pick: 2b77561 "Update libtpu installation index path"

Author: learning-to-play

ci/official/envs/linux_x86_tpu

@@ -20,4 +20,4 @@ TFCI_LIB_SUFFIX="-tpu-linux-x86_64"
 TFCI_WHL_BAZEL_TEST_ENABLE=0
 TFCI_WHL_IMPORT_TEST_ENABLE=0
 TFCI_WHL_SIZE_LIMIT=580M
-TFCI_PYTHON_VERIFY_PIP_INSTALL_ARGS="-f https://storage.googleapis.com/libtpu-tf-releases/index.html"
+TFCI_PYTHON_VERIFY_PIP_INSTALL_ARGS="-f https://storage.googleapis.com/libtpu-wheels/index.html"

tensorflow/tools/pip_package/setup.py

@@ -50,6 +50,9 @@
 # tensorflow/core/public/version.h
 _VERSION = '2.19.0-rc0'
 
+# Update this version when a new libtpu stable version is released.
+LATEST_RELEASE_LIBTPU_VERSION = '0.0.9'
+NEXT_LIBTPU_VERSION = '0.0.10'
 
 # We use the same setup.py for all tensorflow_* packages and for the nightly
 # equivalents (tf_nightly_*). The package is controlled from the argument line
@@ -310,17 +313,14 @@ def find_files(pattern, root):
   # timing of these tests, the UTC date from eight hours ago is expected to be a
   # valid version.
   _libtpu_version = standard_or_nightly(
-      _VERSION.replace('-', ''),
-      '0.1.dev'
+      LATEST_RELEASE_LIBTPU_VERSION,
+      NEXT_LIBTPU_VERSION + '.dev'
       + (
           datetime.datetime.now(tz=datetime.timezone.utc)
           - datetime.timedelta(hours=8)
-      ).strftime('%Y%m%d'),
+      ).strftime('%Y%m%d') + '+nightly',
   )
-  if _libtpu_version.startswith('0.1'):
-    REQUIRED_PACKAGES.append([f'libtpu-nightly=={_libtpu_version}'])
-  else:
-    REQUIRED_PACKAGES.append([f'libtpu=={_libtpu_version}'])
+  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 = '