# Copyright 2026 The OpenXLA Authors. All Rights Reserved.
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
#     http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.
# =============================================================================

"""Module extension for third party dependencies."""

load("//third_party/benchmark:workspace.bzl", benchmark = "repo")
load("//third_party/brotli:workspace.bzl", brotli = "repo")
load("//third_party/compute_library:workspace.bzl", compute_library = "repo")
load("//third_party/cpuinfo:workspace.bzl", cpuinfo = "repo")
load("//third_party/cudnn_frontend:workspace.bzl", cudnn_frontend = "repo")
load("//third_party/cutlass:workspace.bzl", cutlass = "repo")
load("//third_party/dlpack:workspace.bzl", dlpack = "repo")
load("//third_party/ducc:workspace.bzl", ducc = "repo")
load("//third_party/eigen3:workspace.bzl", eigen3 = "repo")
load("//third_party/farmhash:workspace.bzl", farmhash = "repo")
load("//third_party/fmt:workspace.bzl", fmt = "repo")
load("//third_party/FP16:workspace.bzl", FP16 = "repo")
load("//third_party/fxdiv:workspace.bzl", fxdiv = "repo")
load("//third_party/gemmlowp:workspace.bzl", gemmlowp = "repo")
load("//third_party/gloo:workspace.bzl", gloo = "repo")
load("//third_party/highwayhash:workspace.bzl", highwayhash = "repo")
load("//third_party/hwloc:workspace.bzl", hwloc = "repo")
load("//third_party/implib_so:workspace.bzl", implib_so = "repo")
load("//third_party/kleidiai:workspace.bzl", kleidiai = "repo")
load("//third_party/llvm:workspace.bzl", llvm = "repo")
load("//third_party/llvm_openmp:workspace.bzl", llvm_openmp = "repo")
load("//third_party/mkl_dnn:workspace.bzl", onednn = "repo")
load("//third_party/mpitrampoline:workspace.bzl", mpitrampoline = "repo")
load("//third_party/nanobind:workspace.bzl", nanobind = "repo")
load("//third_party/nasm:workspace.bzl", nasm = "repo")
load("//third_party/nccl:workspace.bzl", nccl = "repo")
load("//third_party/net_zstd:workspace.bzl", net_zstd = "repo")
load("//third_party/nvshmem:workspace.bzl", nvshmem = "repo")
load("//third_party/nvtx:workspace.bzl", nvtx = "repo")
load("//third_party/oneccl:workspace.bzl", oneccl = "repo_v2", oneccl_v1 = "repo_v1")
load("//third_party/pthreadpool:workspace.bzl", pthreadpool = "repo")
load("//third_party/py/ml_dtypes:workspace.bzl", ml_dtypes = "repo")
load("//third_party/raft:workspace.bzl", raft = "xla_repo")
load("//third_party/rapids_logger:workspace.bzl", rapids_logger = "xla_repo")
load("//third_party/riegeli:workspace.bzl", riegeli = "repo")
load("//third_party/rmm:workspace.bzl", rmm = "xla_repo")
load("//third_party/robin_map:workspace.bzl", robin_map = "repo")
load("//third_party/rocm_device_libs:workspace.bzl", rocm_device_libs = "repo")
load("//third_party/shardy:workspace.bzl", shardy = "repo")
load("//third_party/slinky:workspace.bzl", slinky = "repo")
load("//third_party/spdlog:workspace.bzl", spdlog = "repo")
load("//third_party/sqlite:workspace.bzl", sqlite = "repo")
load("//third_party/stablehlo:workspace.bzl", stablehlo = "repo")
load("//third_party/tensorrt:workspace.bzl", tensorrt = "repo")
load("//third_party/transformer_engine:workspace.bzl", transformer_engine = "repo")
load("//third_party/triton:workspace.bzl", triton = "repo")
load("//third_party/uv:workspace.bzl", uv = "repo")
load("//third_party/xnnpack:workspace.bzl", xnnpack = "repo")

def _third_party_ext_impl(mctx):  # @unused
    # go/keep-sorted start
    FP16()
    benchmark()
    brotli()  # Needed for Riegeli
    compute_library()
    cpuinfo()
    cudnn_frontend()
    cutlass()
    dlpack()
    ducc()
    eigen3()
    farmhash()
    fmt()
    fxdiv()
    gemmlowp()
    gloo()
    highwayhash()
    hwloc()
    implib_so()
    kleidiai()
    llvm(name = "llvm-raw")
    llvm_openmp()
    ml_dtypes()
    mpitrampoline()
    nanobind()
    nasm()
    nccl()
    net_zstd()  # Needed for Riegeli
    nvshmem()
    nvtx()
    oneccl()
    oneccl_v1()
    onednn()
    pthreadpool()
    raft()
    rapids_logger()
    riegeli()
    rmm()
    robin_map()
    rocm_device_libs()
    shardy()
    slinky()
    spdlog()
    sqlite()
    stablehlo()
    tensorrt()
    transformer_engine()
    triton()
    uv()
    xnnpack()
    # go/keep-sorted end

third_party_ext = module_extension(
    implementation = _third_party_ext_impl,
)