From ce3412e5401b5e2c14d14cd878020e3c169f33ab Mon Sep 17 00:00:00 2001 From: jax authors Date: Tue, 4 Mar 2025 08:12:27 -0800 Subject: [PATCH] Remove redundant `BUILD_TAG` from `JAX` wheels build rule. PiperOrigin-RevId: 733334423 --- jaxlib/jax.bzl | 7 ++----- 1 file changed, 2 insertions(+), 5 deletions(-) diff --git a/jaxlib/jax.bzl b/jaxlib/jax.bzl index d69532956..633cd07ab 100644 --- a/jaxlib/jax.bzl +++ b/jaxlib/jax.bzl @@ -17,7 +17,7 @@ load("@bazel_skylib//rules:common_settings.bzl", "BuildSettingInfo") load("@com_github_google_flatbuffers//:build_defs.bzl", _flatbuffer_cc_library = "flatbuffer_cc_library") load("@jax_wheel//:wheel.bzl", "WHEEL_VERSION") -load("@jax_wheel_version_suffix//:wheel_version_suffix.bzl", "BUILD_TAG", "WHEEL_VERSION_SUFFIX") +load("@jax_wheel_version_suffix//:wheel_version_suffix.bzl", "WHEEL_VERSION_SUFFIX") load("@local_config_cuda//cuda:build_defs.bzl", _cuda_library = "cuda_library", _if_cuda_is_configured = "if_cuda_is_configured") load("@local_config_rocm//rocm:build_defs.bzl", _if_rocm_is_configured = "if_rocm_is_configured", _rocm_library = "rocm_library") load("@python_version_repo//:py_version.bzl", "HERMETIC_PYTHON_VERSION") @@ -365,10 +365,7 @@ def _jax_wheel_impl(ctx): full_wheel_version = (WHEEL_VERSION + WHEEL_VERSION_SUFFIX) env["WHEEL_VERSION_SUFFIX"] = WHEEL_VERSION_SUFFIX - if BUILD_TAG: - env["WHEEL_VERSION_SUFFIX"] = ".dev{}+selfbuilt".format(BUILD_TAG) - full_wheel_version += env["WHEEL_VERSION_SUFFIX"] - if not WHEEL_VERSION_SUFFIX and not BUILD_TAG: + if not WHEEL_VERSION_SUFFIX: env["JAX_RELEASE"] = "1" cpu = ctx.attr.cpu