# Copyright (c) 2024-2026, NVIDIA CORPORATION & AFFILIATES. ALL RIGHTS RESERVED.
#
# SPDX-License-Identifier: Apache-2.0

# type: ignore

from nvmath.bindings import cublas
from nvmath.bindings import cublasLt
from nvmath.bindings import cudss
from nvmath.bindings import cufft
from nvmath.bindings import curand
from nvmath.bindings import cusolver
from nvmath.bindings import cusolverDn
from nvmath.bindings import cusparse
from nvmath.bindings import cusparseLt
from nvmath.bindings import cutensor

try:
    # cufftMp is Linux-only.
    from nvmath.bindings import cufftMp
except ImportError:
    cufftMp = None

try:
    # nvshmem is Linux-only.
    from nvmath.bindings import nvshmem
except ImportError:
    nvshmem = None

try:
    # cublasMp is Linux-only.
    from nvmath.bindings import cublasMp
except ImportError:
    cublasMp = None

__all__ = [
    "cublas",
    "cublasLt",
    "cublasMp",
    "cudss",
    "cufft",
    "cufftMp",
    "curand",
    "cusolver",
    "cusolverDn",
    "cusparse",
    "cusparseLt",
    "cutensor",
    "nvpl",
    "nvshmem",
]
