"nccl_redist_init_repository",
)

nccl_redist_init_repository()

load(
    "@rules_ml_toolchain//gpu/nccl:nccl_configure.bzl",
    "nccl_configure",
)

nccl_configure(name = "local_config_nccl")

load(
    "@rules_ml_toolchain//gpu/nvshmem:nvshmem_json_init_repository.bzl",
    "nvshmem_json_init_repository",
)

nvshmem_json_init_repository()

load(

nccl_configure = use_extension("@rules_ml_toolchain//extensions:nccl_configure.bzl", "nccl_configure_ext")
use_repo(nccl_configure, "local_config_nccl")

sycl_configure = use_extension("@rules_ml_toolchain//extensions:sycl_configure.bzl", "sycl_configure_ext")
use_repo(sycl_configure, "local_config_sycl")

cuda_redist_init_ext = use_extension("@rules_ml_toolchain//extensions:cuda_redist_init.bzl", "cuda_redist_init_ext")
use_repo(

    "@llvm_openmp//:__subpackages__",
    "@llvm_terminfo//:__subpackages__",
    "@llvm_zlib//:__subpackages__",
    "@local_config_cuda//:__subpackages__",
    "@local_config_git//:__subpackages__",
    "@local_config_nccl//:__subpackages__",
    "@local_config_rocm//:__subpackages__",
    "@local_config_tensorrt//:__subpackages__",
    "@mkl_dnn_acl_compatible//:__subpackages__",
    "@ml_dtypes_py//ml_dtypes:__subpackages__",

common:rocm_ci_hermetic --repo_env="ROCM_VERSION=6.2.0"
common:rocm_ci_hermetic --@local_config_rocm//rocm:use_rocm_hermetic_rpath=True

common:sycl --config=clang_local
common:sycl --crosstool_top=@local_config_sycl//crosstool:toolchain
common:sycl --define=using_sycl=true
common:sycl --define=tensorflow_mkldnn_contraction_kernel=0
common:sycl --repo_env TF_NEED_SYCL=1

# Options to disable default on features