nvmath-python 1.0.0__cp314-cp314t-win_amd64.whl

nvmath/bindings/__init__.py

@@ -0,0 +1,60 @@
+# 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 cusolverSp
+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
+
+try:
+    # cusolverMp is Linux-only.
+    from nvmath.bindings import cusolverMp
+except ImportError:
+    cusolverMp = None
+
+__all__ = [
+    "cublas",
+    "cublasLt",
+    "cublasMp",
+    "cudss",
+    "cufft",
+    "cufftMp",
+    "curand",
+    "cusolver",
+    "cusolverDn",
+    "cusolverSp",
+    "cusolverMp",
+    "cusparse",
+    "cusparseLt",
+    "cutensor",
+    "nvpl",
+    "nvshmem",
+]

nvmath/bindings/_internal/common_types.pxd

@@ -0,0 +1,31 @@
+# Copyright (c) 2026, NVIDIA CORPORATION & AFFILIATES. ALL RIGHTS RESERVED.
+#
+# SPDX-License-Identifier: Apache-2.0
+
+"""Shared CUDA-level Cython declarations used by generated bindings."""
+
+
+cdef extern from *:
+    """
+    #include <driver_types.h>
+    #include <library_types.h>
+    #include <cuComplex.h>
+    """
+    ctypedef void* cudaStream_t 'cudaStream_t'
+    ctypedef int cudaDataType_t 'cudaDataType_t'
+    ctypedef int cudaDataType 'cudaDataType'
+    ctypedef int libraryPropertyType_t 'libraryPropertyType_t'
+    ctypedef int libraryPropertyType 'libraryPropertyType'
+
+    ctypedef int cudaEmulationStrategy_t 'cudaEmulationStrategy_t'
+    ctypedef int cudaEmulationMantissaControl_t 'cudaEmulationMantissaControl_t'
+    ctypedef int cudaEmulationMantissaControl 'cudaEmulationMantissaControl'
+    ctypedef int cudaEmulationSpecialValuesSupport_t 'cudaEmulationSpecialValuesSupport_t'
+    ctypedef int cudaEmulationSpecialValuesSupport 'cudaEmulationSpecialValuesSupport'
+
+    ctypedef struct cuComplex:
+        float x
+        float y
+    ctypedef struct cuDoubleComplex:
+        double x
+        double y