From af1de1f019838aad3b657a4d09a319abac7c2c87 Mon Sep 17 00:00:00 2001 From: jparkerh Date: Wed, 29 Jul 2026 13:35:47 -0700 Subject: [PATCH] Add tpu-raiden wheel build to torch_tpu wheel build pipeline - What the problem was: Nightly wheel builds only produced torch_tpu wheels, requiring a separate build process for tpu-raiden wheels with mismatched timestamp tags. - Why it happened: tpu-raiden wheel generation was not integrated into the main torch_tpu wheel build script (build_wheel.sh). - How this change resolves it: Updated third_party/py/torch_tpu/oss/ci/build_wheel.sh and third_party/tpu_raiden/ci/build_wheel.sh to build tpu-raiden wheels alongside torch_tpu wheels using a matching pre-set WHEEL_VERSION_EXTRAS timestamp, and updated upload_wheel.sh to publish all generated wheels. PiperOrigin-RevId: 956060724 --- ci/build_wheel.sh | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/ci/build_wheel.sh b/ci/build_wheel.sh index 5645a12a..eccfbf67 100755 --- a/ci/build_wheel.sh +++ b/ci/build_wheel.sh @@ -45,7 +45,7 @@ set -exu -o pipefail REPO_ROOT="$(cd "$(dirname "${BASH_SOURCE[0]}")/.." && pwd)" cd "${REPO_ROOT}" -WHEEL_VERSION_EXTRAS=".dev$(date +%Y%m%d%H%M%S)" +WHEEL_VERSION_EXTRAS="${WHEEL_VERSION_EXTRAS:-.dev$(date +%Y%m%d%H%M%S)}" export WHEEL_VERSION_EXTRAS echo "WHEEL_VERSION_EXTRAS: ${WHEEL_VERSION_EXTRAS}"