)

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=""

@rules_ml_toolchain//
@rules_python//
@stablehlo//stablehlo/experimental
EOF

  # grep patterns for targets which are allowed to be extra licenses
  cat > $BATS_TEST_TMPDIR/allowed_to_be_extra <<EOF
@local_xla//third_party/mkl
@local_xla//third_party/mkl_dnn
@absl_py//
@bazel_tools//src
@bazel_tools//platforms
@bazel_tools//tools
@org_tensorflow//tensorflow
@com_google_absl//
@com_google_protobuf//
@internal_platforms_do_not_use//host