Update rules_ml_toolchain to integrate mirrored_http_archive repo rule.

If the repository env var `USE_MIRRORED_TAR_ARCHIVE_FILES` is set to `1`, the `.tar` files are downloaded from the mirror instead of `.tar.xz`.

`USE_MIRRORED_TAR_ARCHIVE_FILES` is not set by default.

PiperOrigin-RevId: 794631542
This commit is contained in:
A. Unique TensorFlower 2025-08-13 10:13:02 -07:00 committed by TensorFlower Gardener
parent c224f0cdc5
commit be308cd0b2
4 changed files with 8 additions and 6 deletions

View File

@ -589,6 +589,7 @@ common --experimental_repo_remote_exec
build:use_tar_archive_files --repo_env=USE_CUDA_TAR_ARCHIVE_FILES=1
build:use_tar_archive_files --repo_env=USE_NVSHMEM_TAR_ARCHIVE_FILES=1
build:use_tar_archive_files --repo_env=USE_LLVM_TAR_ARCHIVE_FILES=1
build:use_tar_archive_files --repo_env=USE_MIRRORED_TAR_ARCHIVE_FILES=1
# Make Bazel not try to probe the host system for a C++ toolchain.
build:rbe_base --config=use_tar_archive_files

View File

@ -140,10 +140,10 @@ def workspace():
# Details: https://github.com/google-ml-infra/rules_ml_toolchain
http_archive(
name = "rules_ml_toolchain",
sha256 = "b15e9159fb9c42a1a54319456fc7e8f2855261dedc90927a60f4fdc5dbfdcf58",
strip_prefix = "rules_ml_toolchain-09961bd590a5d3627b0044cd71b69b8cb9ba3145",
sha256 = "59d7eb36a02cbe3c2e2fa67fda5e8f1ab7e274bc4773bbd207c51fe199e11c19",
strip_prefix = "rules_ml_toolchain-ffd9e3d7b84e43c2686c803cb08ce790ffd58baa",
urls = [
"https://github.com/google-ml-infra/rules_ml_toolchain/archive/09961bd590a5d3627b0044cd71b69b8cb9ba3145.tar.gz",
"https://github.com/google-ml-infra/rules_ml_toolchain/archive/ffd9e3d7b84e43c2686c803cb08ce790ffd58baa.tar.gz",
],
)

View File

@ -470,6 +470,7 @@ common --experimental_repo_remote_exec
build:use_tar_archive_files --repo_env=USE_CUDA_TAR_ARCHIVE_FILES=1
build:use_tar_archive_files --repo_env=USE_NVSHMEM_TAR_ARCHIVE_FILES=1
build:use_tar_archive_files --repo_env=USE_LLVM_TAR_ARCHIVE_FILES=1
build:use_tar_archive_files --repo_env=USE_MIRRORED_TAR_ARCHIVE_FILES=1
# Make Bazel not try to probe the host system for a C++ toolchain.
build:rbe_base --config=use_tar_archive_files

View File

@ -138,10 +138,10 @@ def workspace():
# Details: https://github.com/google-ml-infra/rules_ml_toolchain
http_archive(
name = "rules_ml_toolchain",
sha256 = "4e7ff653635d3a20c63075da6201077345fde3fde67a5a36ee95323b203da9c6",
strip_prefix = "rules_ml_toolchain-aaf55866f2a6c42eb9eb2449a9211df9c3420682",
sha256 = "59d7eb36a02cbe3c2e2fa67fda5e8f1ab7e274bc4773bbd207c51fe199e11c19",
strip_prefix = "rules_ml_toolchain-ffd9e3d7b84e43c2686c803cb08ce790ffd58baa",
urls = [
"https://github.com/google-ml-infra/rules_ml_toolchain/archive/aaf55866f2a6c42eb9eb2449a9211df9c3420682.tar.gz",
"https://github.com/google-ml-infra/rules_ml_toolchain/archive/ffd9e3d7b84e43c2686c803cb08ce790ffd58baa.tar.gz",
],
)