)

load(
    "@rules_ml_toolchain//third_party/gpus/cuda/hermetic:cuda_configure.bzl",
    "cuda_configure",
)

cuda_configure(name = "local_config_cuda")

load(
    "@rules_ml_toolchain//third_party/nccl/hermetic:nccl_redist_init_repository.bzl",
    "nccl_redist_init_repository",
)

nccl_redist_init_repository()

load(
    "@rules_ml_toolchain//third_party/nccl/hermetic:nccl_configure.bzl",

# These are used to generate the ML wheel version string.
# See the explanation in the file comment of
# @local_xla//third_party/py/python_wheel.bzl.
# The generated version suffix is used in
# third_party/tensorflow/core/public/release_version.h and
# third_party/tensorflow/tools/pip_package/setup.oss.py.tpl
common --repo_env=ML_WHEEL_TYPE="snapshot"
common --repo_env=ML_WHEEL_BUILD_DATE=""