PyPI - nvmath-python - Versions diffs - 1.0.0__cp314-cp314t-win_amd64.whl - Mend

nvmath-python 1.0.0__cp314-cp314t-win_amd64.whl

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.

Files changed (332) hide show

nvmath/__init__.pxd +0 -0
nvmath/__init__.py +45 -0
nvmath/_internal/__init__.py +0 -0
nvmath/_internal/attribute_ifc_factory.py +330 -0
nvmath/_internal/layout.py +70 -0
nvmath/_internal/templates.py +130 -0
nvmath/_internal/threadsafe.py +106 -0
nvmath/_internal/utils.py +43 -0
nvmath/_internal/workspace.py +490 -0
nvmath/_utils.py +147 -0
nvmath/bindings/__init__.py +60 -0
nvmath/bindings/_internal/__init__.pxd +0 -0
nvmath/bindings/_internal/__init__.py +0 -0
nvmath/bindings/_internal/common_types.pxd +31 -0
nvmath/bindings/_internal/cublas.cp314t-win_amd64.pyd +0 -0
nvmath/bindings/_internal/cublas.pxd +530 -0
nvmath/bindings/_internal/cublasLt.cp314t-win_amd64.pyd +0 -0
nvmath/bindings/_internal/cublasLt.pxd +59 -0
nvmath/bindings/_internal/cublasMp.pxd +52 -0
nvmath/bindings/_internal/cudss.cp314t-win_amd64.pyd +0 -0
nvmath/bindings/_internal/cudss.pxd +54 -0
nvmath/bindings/_internal/cufft.cp314t-win_amd64.pyd +0 -0
nvmath/bindings/_internal/cufft.pxd +70 -0
nvmath/bindings/_internal/cufftMp.pxd +77 -0
nvmath/bindings/_internal/curand.cp314t-win_amd64.pyd +0 -0
nvmath/bindings/_internal/curand.pxd +42 -0
nvmath/bindings/_internal/cusolver.cp314t-win_amd64.pyd +0 -0
nvmath/bindings/_internal/cusolver.pxd +15 -0
nvmath/bindings/_internal/cusolverDn.cp314t-win_amd64.pyd +0 -0
nvmath/bindings/_internal/cusolverDn.pxd +406 -0
nvmath/bindings/_internal/cusolverMp.pxd +71 -0
nvmath/bindings/_internal/cusolverSp.cp314t-win_amd64.pyd +0 -0
nvmath/bindings/_internal/cusolverSp.pxd +75 -0
nvmath/bindings/_internal/cusparse.cp314t-win_amd64.pyd +0 -0
nvmath/bindings/_internal/cusparse.pxd +471 -0
nvmath/bindings/_internal/cusparseLt.cp314t-win_amd64.pyd +0 -0
nvmath/bindings/_internal/cusparseLt.pxd +48 -0
nvmath/bindings/_internal/cutensor.cp314t-win_amd64.pyd +0 -0
nvmath/bindings/_internal/cutensor.pxd +58 -0
nvmath/bindings/_internal/mathdx.cp314t-win_amd64.pyd +0 -0
nvmath/bindings/_internal/mathdx.pxd +116 -0
nvmath/bindings/_internal/nvshmem.pxd +29 -0
nvmath/bindings/_internal/utils.cp314t-win_amd64.pyd +0 -0
nvmath/bindings/_internal/utils.pxd +174 -0
nvmath/bindings/_internal/utils.pyi +10 -0
nvmath/bindings/cublas.cp314t-win_amd64.pyd +0 -0
nvmath/bindings/cublas.pxd +558 -0
nvmath/bindings/cublas.pyi +812 -0
nvmath/bindings/cublasLt.cp314t-win_amd64.pyd +0 -0
nvmath/bindings/cublasLt.pxd +109 -0
nvmath/bindings/cublasLt.pyi +1461 -0
nvmath/bindings/cublasMp.pxd +85 -0
nvmath/bindings/cublasMp.pyi +267 -0
nvmath/bindings/cudss.cp314t-win_amd64.pyd +0 -0
nvmath/bindings/cudss.pxd +98 -0
nvmath/bindings/cudss.pyi +443 -0
nvmath/bindings/cufft.cp314t-win_amd64.pyd +0 -0
nvmath/bindings/cufft.pxd +118 -0
nvmath/bindings/cufft.pyi +301 -0
nvmath/bindings/cufftMp.pxd +124 -0
nvmath/bindings/cufftMp.pyi +326 -0
nvmath/bindings/curand.cp314t-win_amd64.pyd +0 -0
nvmath/bindings/curand.pxd +71 -0
nvmath/bindings/curand.pyi +189 -0
nvmath/bindings/cusolver.cp314t-win_amd64.pyd +0 -0
nvmath/bindings/cusolver.pxd +62 -0
nvmath/bindings/cusolver.pyi +320 -0
nvmath/bindings/cusolverDn.cp314t-win_amd64.pyd +0 -0
nvmath/bindings/cusolverDn.pxd +430 -0
nvmath/bindings/cusolverDn.pyi +422 -0
nvmath/bindings/cusolverMp.pxd +98 -0
nvmath/bindings/cusolverMp.pyi +114 -0
nvmath/bindings/cusolverSp.cp314t-win_amd64.pyd +0 -0
nvmath/bindings/cusolverSp.pxd +95 -0
nvmath/bindings/cusolverSp.pyi +70 -0
nvmath/bindings/cusparse.cp314t-win_amd64.pyd +0 -0
nvmath/bindings/cusparse.pxd +546 -0
nvmath/bindings/cusparse.pyi +1017 -0
nvmath/bindings/cusparseLt.cp314t-win_amd64.pyd +0 -0
nvmath/bindings/cusparseLt.pxd +99 -0
nvmath/bindings/cusparseLt.pyi +252 -0
nvmath/bindings/cutensor.cp314t-win_amd64.pyd +0 -0
nvmath/bindings/cutensor.pxd +98 -0
nvmath/bindings/cutensor.pyi +324 -0
nvmath/bindings/cycublas.cp314t-win_amd64.pyd +0 -0
nvmath/bindings/cycublas.pxd +664 -0
nvmath/bindings/cycublasLt.cp314t-win_amd64.pyd +0 -0
nvmath/bindings/cycublasLt.pxd +1045 -0
nvmath/bindings/cycublasMp.pxd +171 -0
nvmath/bindings/cycudss.cp314t-win_amd64.pyd +0 -0
nvmath/bindings/cycudss.pxd +277 -0
nvmath/bindings/cycufft.cp314t-win_amd64.pyd +0 -0
nvmath/bindings/cycufft.pxd +333 -0
nvmath/bindings/cycufftMp.pxd +342 -0
nvmath/bindings/cycurand.cp314t-win_amd64.pyd +0 -0
nvmath/bindings/cycurand.pxd +141 -0
nvmath/bindings/cycusolver.cp314t-win_amd64.pyd +0 -0
nvmath/bindings/cycusolver.pxd +137 -0
nvmath/bindings/cycusolverDn.cp314t-win_amd64.pyd +0 -0
nvmath/bindings/cycusolverDn.pxd +443 -0
nvmath/bindings/cycusolverMp.pxd +107 -0
nvmath/bindings/cycusolverSp.cp314t-win_amd64.pyd +0 -0
nvmath/bindings/cycusolverSp.pxd +93 -0
nvmath/bindings/cycusparse.cp314t-win_amd64.pyd +0 -0
nvmath/bindings/cycusparse.pxd +679 -0
nvmath/bindings/cycusparseLt.cp314t-win_amd64.pyd +0 -0
nvmath/bindings/cycusparseLt.pxd +135 -0
nvmath/bindings/cycutensor.cp314t-win_amd64.pyd +0 -0
nvmath/bindings/cycutensor.pxd +189 -0
nvmath/bindings/cymathdx.cp314t-win_amd64.pyd +0 -0
nvmath/bindings/cymathdx.pxd +552 -0
nvmath/bindings/cynvshmem.pxd +118 -0
nvmath/bindings/mathdx.cp314t-win_amd64.pyd +0 -0
nvmath/bindings/mathdx.pxd +182 -0
nvmath/bindings/mathdx.pyi +1562 -0
nvmath/bindings/nvpl/__init__.pxd +0 -0
nvmath/bindings/nvpl/__init__.py +13 -0
nvmath/bindings/nvpl/_internal/__init__.pxd +0 -0
nvmath/bindings/nvpl/_internal/__init__.py +0 -0
nvmath/bindings/nvpl/_internal/blas.cp314t-win_amd64.pyd +0 -0
nvmath/bindings/nvpl/_internal/blas.pxd +237 -0
nvmath/bindings/nvpl/_internal/fft.cp314t-win_amd64.pyd +0 -0
nvmath/bindings/nvpl/_internal/fft.pxd +36 -0
nvmath/bindings/nvpl/blas.cp314t-win_amd64.pyd +0 -0
nvmath/bindings/nvpl/blas.pxd +131 -0
nvmath/bindings/nvpl/blas.pyi +168 -0
nvmath/bindings/nvpl/cyblas.cp314t-win_amd64.pyd +0 -0
nvmath/bindings/nvpl/cyblas.pxd +280 -0
nvmath/bindings/nvpl/cyfft.cp314t-win_amd64.pyd +0 -0
nvmath/bindings/nvpl/cyfft.pxd +93 -0
nvmath/bindings/nvpl/fft.cp314t-win_amd64.pyd +0 -0
nvmath/bindings/nvpl/fft.pxd +100 -0
nvmath/bindings/nvpl/fft.pyi +168 -0
nvmath/bindings/nvshmem.pxd +54 -0
nvmath/bindings/nvshmem.pyi +191 -0
nvmath/device/__init__.py +38 -0
nvmath/device/_deprecated.py +33 -0
nvmath/device/common.py +315 -0
nvmath/device/common_backend.py +131 -0
nvmath/device/common_cuda.py +201 -0
nvmath/device/common_numba.py +300 -0
nvmath/device/common_numba_cuda_mlir.py +202 -0
nvmath/device/common_opaque_tensor.py +201 -0
nvmath/device/cublasdx.py +1606 -0
nvmath/device/cublasdx_backend.py +860 -0
nvmath/device/cublasdx_numba.py +1534 -0
nvmath/device/cublasdx_numba_cuda_mlir.py +208 -0
nvmath/device/cufftdx.py +373 -0
nvmath/device/cufftdx_backend.py +220 -0
nvmath/device/cufftdx_numba.py +140 -0
nvmath/device/cufftdx_numba_cuda_mlir.py +79 -0
nvmath/device/curand_kernel.py +9147 -0
nvmath/device/cusolverdx.py +2708 -0
nvmath/device/cusolverdx_backend.py +440 -0
nvmath/device/cusolverdx_numba.py +567 -0
nvmath/device/cusolverdx_numba_cuda_mlir.py +604 -0
nvmath/device/cusolverdx_overload_backend.py +1029 -0
nvmath/device/llvm_array.py +29 -0
nvmath/device/random.py +441 -0
nvmath/device/random_helpers.py +23 -0
nvmath/device/random_states.py +187 -0
nvmath/device/types.py +138 -0
nvmath/device/vector_types_numba.py +259 -0
nvmath/distributed/__init__.py +200 -0
nvmath/distributed/_internal/__init__.py +0 -0
nvmath/distributed/_internal/nccl.py +86 -0
nvmath/distributed/_internal/nvshmem.py +307 -0
nvmath/distributed/_internal/symmetric_memory.py +35 -0
nvmath/distributed/_internal/tensor_ifc.py +70 -0
nvmath/distributed/_internal/tensor_ifc_cupy.py +68 -0
nvmath/distributed/_internal/tensor_ifc_host_device.py +172 -0
nvmath/distributed/_internal/tensor_ifc_numpy.py +46 -0
nvmath/distributed/_internal/tensor_ifc_torch.py +162 -0
nvmath/distributed/_internal/tensor_wrapper.py +81 -0
nvmath/distributed/_utils.py +167 -0
nvmath/distributed/distribution/__init__.py +30 -0
nvmath/distributed/distribution/_configuration.py +39 -0
nvmath/distributed/distribution/distributions.py +1024 -0
nvmath/distributed/distribution/redistribute.py +1284 -0
nvmath/distributed/fft/__init__.py +7 -0
nvmath/distributed/fft/_configuration.py +82 -0
nvmath/distributed/fft/fft.py +2742 -0
nvmath/distributed/linalg/__init__.py +22 -0
nvmath/distributed/linalg/_internal/__init__.py +3 -0
nvmath/distributed/linalg/_internal/epilog_protocol.py +586 -0
nvmath/distributed/linalg/_internal/matmul_desc_ifc.py +28 -0
nvmath/distributed/linalg/advanced/__init__.py +8 -0
nvmath/distributed/linalg/advanced/_configuration.py +171 -0
nvmath/distributed/linalg/advanced/matmulmod.py +3573 -0
nvmath/distributed/linalg/generic/__init__.py +8 -0
nvmath/distributed/linalg/generic/_caching.py +66 -0
nvmath/distributed/linalg/generic/_configuration.py +61 -0
nvmath/distributed/linalg/generic/_factorization.py +172 -0
nvmath/distributed/linalg/generic/_initialization.py +966 -0
nvmath/distributed/linalg/generic/_problem_spec.py +511 -0
nvmath/distributed/linalg/generic/solvermod.py +1368 -0
nvmath/distributed/process_group.py +408 -0
nvmath/fft/__init__.py +7 -0
nvmath/fft/_configuration.py +189 -0
nvmath/fft/_exec_utils.py +82 -0
nvmath/fft/_helpers.py +237 -0
nvmath/fft/fft.py +3122 -0
nvmath/internal/__init__.pxd +3 -0
nvmath/internal/__init__.py +10 -0
nvmath/internal/_bindings.cp314t-win_amd64.pyd +0 -0
nvmath/internal/_bindings.pxd +18 -0
nvmath/internal/_device_utils.py +45 -0
nvmath/internal/_layout/__init__.pxd +3 -0
nvmath/internal/_layout/__init__.py +7 -0
nvmath/internal/_layout/_layout.cp314t-win_amd64.pyd +0 -0
nvmath/internal/_layout/_layout.pxd +1303 -0
nvmath/internal/_layout/_layout.pyi +1145 -0
nvmath/internal/enum_utils.py +142 -0
nvmath/internal/formatters.py +87 -0
nvmath/internal/mem_limit.py +51 -0
nvmath/internal/memory.cp314t-win_amd64.pyd +0 -0
nvmath/internal/memory.pxd +13 -0
nvmath/internal/memory.pyi +50 -0
nvmath/internal/ndbuffer/__init__.pxd +3 -0
nvmath/internal/ndbuffer/__init__.py +9 -0
nvmath/internal/ndbuffer/_copy_kernel.cp314t-win_amd64.pyd +0 -0
nvmath/internal/ndbuffer/_copy_kernel.pxd +10 -0
nvmath/internal/ndbuffer/_jit.cp314t-win_amd64.pyd +0 -0
nvmath/internal/ndbuffer/_jit.pxd +7 -0
nvmath/internal/ndbuffer/_ndbuffer.cp314t-win_amd64.pyd +0 -0
nvmath/internal/ndbuffer/_ndbuffer.pxd +40 -0
nvmath/internal/ndbuffer/_ndbuffer.pyi +463 -0
nvmath/internal/ndbuffer/copy_kernel/args.h +34 -0
nvmath/internal/ndbuffer/copy_kernel/copy_kernel_impl/array_view.h +52 -0
nvmath/internal/ndbuffer/copy_kernel/copy_kernel_impl/elementwise.h +68 -0
nvmath/internal/ndbuffer/copy_kernel/copy_kernel_impl/grid_indexer.h +69 -0
nvmath/internal/ndbuffer/copy_kernel/copy_kernel_impl/transposed.h +242 -0
nvmath/internal/ndbuffer/copy_kernel/copy_kernel_impl/type_utils.h +39 -0
nvmath/internal/ndbuffer/copy_kernel/copy_kernel_impl/utils.h +132 -0
nvmath/internal/ndbuffer/copy_kernel/copy_kernel_impl/vec.h +159 -0
nvmath/internal/ndbuffer/copy_kernel/elementwise.h +53 -0
nvmath/internal/ndbuffer/copy_kernel/transposed.h +58 -0
nvmath/internal/package_ifc.py +168 -0
nvmath/internal/package_ifc_cuda.py +57 -0
nvmath/internal/package_ifc_cupy.py +67 -0
nvmath/internal/package_ifc_torch.py +69 -0
nvmath/internal/package_wrapper.py +14 -0
nvmath/internal/tensor_ifc.py +179 -0
nvmath/internal/tensor_ifc_cupy.py +234 -0
nvmath/internal/tensor_ifc_ndbuffer.py +147 -0
nvmath/internal/tensor_ifc_numpy.py +184 -0
nvmath/internal/tensor_ifc_torch.py +178 -0
nvmath/internal/tensor_wrapper.py +160 -0
nvmath/internal/typemaps.py +113 -0
nvmath/internal/utils.py +805 -0
nvmath/linalg/__init__.py +56 -0
nvmath/linalg/_internal/__init__.py +3 -0
nvmath/linalg/_internal/algo_cap_ifc.py +82 -0
nvmath/linalg/_internal/algo_config_ifc.py +43 -0
nvmath/linalg/_internal/batch.py +234 -0
nvmath/linalg/_internal/enum_to_tuples.py +64 -0
nvmath/linalg/_internal/epilog_protocol.py +766 -0
nvmath/linalg/_internal/layout.py +624 -0
nvmath/linalg/_internal/matmul_desc_ifc.py +28 -0
nvmath/linalg/_internal/matmul_pref_ifc.py +27 -0
nvmath/linalg/_internal/matrix_layout_ifc.py +26 -0
nvmath/linalg/_internal/solver_utils.py +432 -0
nvmath/linalg/_internal/typemaps.py +144 -0
nvmath/linalg/_internal/utils.py +157 -0
nvmath/linalg/advanced/__init__.py +8 -0
nvmath/linalg/advanced/_algorithmmod.py +170 -0
nvmath/linalg/advanced/_configuration.py +351 -0
nvmath/linalg/advanced/helpers/__init__.py +5 -0
nvmath/linalg/advanced/helpers/matmul.py +1316 -0
nvmath/linalg/advanced/matmulmod.py +3734 -0
nvmath/linalg/generic/__init__.py +53 -0
nvmath/linalg/generic/_configuration/__init__.py +39 -0
nvmath/linalg/generic/_configuration/layout.py +263 -0
nvmath/linalg/generic/_configuration/match.py +734 -0
nvmath/linalg/generic/_configuration/qualifiers.py +493 -0
nvmath/linalg/generic/_configuration/solver_configuration.py +59 -0
nvmath/linalg/generic/_configuration/wrap.py +217 -0
nvmath/linalg/generic/_dtype.py +15 -0
nvmath/linalg/generic/matmulmod.py +2094 -0
nvmath/linalg/generic/solvermod.py +1301 -0
nvmath/memory.py +279 -0
nvmath/sparse/__init__.py +38 -0
nvmath/sparse/_internal/__init__.py +21 -0
nvmath/sparse/_internal/common_utils.py +147 -0
nvmath/sparse/_internal/cudss_config_ifc.py +702 -0
nvmath/sparse/_internal/cudss_data_ifc.py +399 -0
nvmath/sparse/_internal/cudss_utils.py +506 -0
nvmath/sparse/_internal/cusparse_utils.py +382 -0
nvmath/sparse/_internal/sparse_bsc_ifc.py +303 -0
nvmath/sparse/_internal/sparse_bsr_ifc.py +305 -0
nvmath/sparse/_internal/sparse_coo_ifc.py +256 -0
nvmath/sparse/_internal/sparse_csc_ifc.py +268 -0
nvmath/sparse/_internal/sparse_csr_ifc.py +288 -0
nvmath/sparse/_internal/sparse_dia_ifc.py +242 -0
nvmath/sparse/_internal/sparse_format_helpers.py +601 -0
nvmath/sparse/_internal/sparse_tensor_ifc.py +133 -0
nvmath/sparse/_internal/sparse_ust_ifc.py +141 -0
nvmath/sparse/_internal/utils.py +56 -0
nvmath/sparse/advanced/__init__.py +7 -0
nvmath/sparse/advanced/_configuration.py +227 -0
nvmath/sparse/advanced/direct_solver.py +2069 -0
nvmath/sparse/generic/__init__.py +7 -0
nvmath/sparse/generic/_configuration.py +129 -0
nvmath/sparse/generic/_helpers.py +137 -0
nvmath/sparse/generic/_thunks.py +21 -0
nvmath/sparse/generic/matmulmod.py +2353 -0
nvmath/sparse/ust/__init__.py +7 -0
nvmath/sparse/ust/_converters.py +422 -0
nvmath/sparse/ust/_cpp.py +28 -0
nvmath/sparse/ust/_drawer.py +565 -0
nvmath/sparse/ust/_emitter.py +1033 -0
nvmath/sparse/ust/_jit.py +188 -0
nvmath/sparse/ust/_kernel.py +282 -0
nvmath/sparse/ust/_utils.py +149 -0
nvmath/sparse/ust/interfaces/__init__.py +0 -0
nvmath/sparse/ust/interfaces/torch_interface.py +476 -0
nvmath/sparse/ust/tensor.py +1016 -0
nvmath/sparse/ust/tensor_format.py +957 -0
nvmath/tensor/__init__.py +6 -0
nvmath/tensor/_configuration.py +120 -0
nvmath/tensor/_internal/__init__.py +3 -0
nvmath/tensor/_internal/cutensor_config_ifc.py +279 -0
nvmath/tensor/_internal/cutensor_utils.py +230 -0
nvmath/tensor/_internal/data.py +43 -0
nvmath/tensor/_internal/einsum_parser.py +444 -0
nvmath/tensor/_internal/typemaps.py +96 -0
nvmath/tensor/contract.py +1900 -0
nvmath_python-1.0.0.dist-info/METADATA +134 -0
nvmath_python-1.0.0.dist-info/RECORD +332 -0
nvmath_python-1.0.0.dist-info/WHEEL +5 -0
nvmath_python-1.0.0.dist-info/licenses/LICENSE +177 -0
nvmath_python-1.0.0.dist-info/top_level.txt +2 -0

nvmath/bindings/cufftMp.pyi ADDED Viewed

@@ -0,0 +1,326 @@
+# Copyright (c) 2026, NVIDIA CORPORATION & AFFILIATES. ALL RIGHTS RESERVED.
+#
+# SPDX-License-Identifier: Apache-2.0
+import _cython_3_2_5
+import enum
+from typing import Any, Callable, ClassVar
+__pyx_capi__: dict
+__test__: dict
+attach_comm: _cython_3_2_5.cython_function_or_method
+attach_reshape_comm: _cython_3_2_5.cython_function_or_method
+check_status: _cython_3_2_5.cython_function_or_method
+create: _cython_3_2_5.cython_function_or_method
+create_reshape: _cython_3_2_5.cython_function_or_method
+destroy: _cython_3_2_5.cython_function_or_method
+destroy_reshape: _cython_3_2_5.cython_function_or_method
+estimate1d: _cython_3_2_5.cython_function_or_method
+estimate2d: _cython_3_2_5.cython_function_or_method
+estimate3d: _cython_3_2_5.cython_function_or_method
+estimate_many: _cython_3_2_5.cython_function_or_method
+exec_c2c: _cython_3_2_5.cython_function_or_method
+exec_c2r: _cython_3_2_5.cython_function_or_method
+exec_d2z: _cython_3_2_5.cython_function_or_method
+exec_r2c: _cython_3_2_5.cython_function_or_method
+exec_reshape_async: _cython_3_2_5.cython_function_or_method
+exec_z2d: _cython_3_2_5.cython_function_or_method
+exec_z2z: _cython_3_2_5.cython_function_or_method
+get_plan_property_int64: _cython_3_2_5.cython_function_or_method
+get_property: _cython_3_2_5.cython_function_or_method
+get_reshape_size: _cython_3_2_5.cython_function_or_method
+get_size: _cython_3_2_5.cython_function_or_method
+get_size1d: _cython_3_2_5.cython_function_or_method
+get_size2d: _cython_3_2_5.cython_function_or_method
+get_size3d: _cython_3_2_5.cython_function_or_method
+get_size_many: _cython_3_2_5.cython_function_or_method
+get_size_many64: _cython_3_2_5.cython_function_or_method
+get_version: _cython_3_2_5.cython_function_or_method
+make_plan1d: _cython_3_2_5.cython_function_or_method
+make_plan2d: _cython_3_2_5.cython_function_or_method
+make_plan3d: _cython_3_2_5.cython_function_or_method
+make_plan_many: _cython_3_2_5.cython_function_or_method
+make_plan_many64: _cython_3_2_5.cython_function_or_method
+make_reshape: _cython_3_2_5.cython_function_or_method
+make_reshape_11_2: _cython_3_2_5.cython_function_or_method
+make_reshape_11_4: _cython_3_2_5.cython_function_or_method
+plan1d: _cython_3_2_5.cython_function_or_method
+plan2d: _cython_3_2_5.cython_function_or_method
+plan3d: _cython_3_2_5.cython_function_or_method
+plan_many: _cython_3_2_5.cython_function_or_method
+reset_plan_property: _cython_3_2_5.cython_function_or_method
+set_auto_allocation: _cython_3_2_5.cython_function_or_method
+set_descriptor_data: _cython_3_2_5.cython_function_or_method
+set_plan_property_int64: _cython_3_2_5.cython_function_or_method
+set_stream: _cython_3_2_5.cython_function_or_method
+set_work_area: _cython_3_2_5.cython_function_or_method
+xt_clear_callback: _cython_3_2_5.cython_function_or_method
+xt_exec: _cython_3_2_5.cython_function_or_method
+xt_exec_descriptor: _cython_3_2_5.cython_function_or_method
+xt_exec_descriptor_c2c: _cython_3_2_5.cython_function_or_method
+xt_exec_descriptor_c2r: _cython_3_2_5.cython_function_or_method
+xt_exec_descriptor_d2z: _cython_3_2_5.cython_function_or_method
+xt_exec_descriptor_r2c: _cython_3_2_5.cython_function_or_method
+xt_exec_descriptor_z2d: _cython_3_2_5.cython_function_or_method
+xt_exec_descriptor_z2z: _cython_3_2_5.cython_function_or_method
+xt_free: _cython_3_2_5.cython_function_or_method
+xt_get_size_many: _cython_3_2_5.cython_function_or_method
+xt_make_plan_many: _cython_3_2_5.cython_function_or_method
+xt_malloc: _cython_3_2_5.cython_function_or_method
+xt_memcpy: _cython_3_2_5.cython_function_or_method
+xt_query_plan: _cython_3_2_5.cython_function_or_method
+xt_set_callback_shared_size: _cython_3_2_5.cython_function_or_method
+xt_set_distribution: _cython_3_2_5.cython_function_or_method
+xt_set_gpus: _cython_3_2_5.cython_function_or_method
+xt_set_subformat_default: _cython_3_2_5.cython_function_or_method
+xt_set_work_area: _cython_3_2_5.cython_function_or_method
+xt_set_work_area_policy: _cython_3_2_5.cython_function_or_method
+class Compatibility(enum.IntEnum):
+    """
+    See `cufftCompatibility`.
+    """
+    __new__: ClassVar[Callable] = ...
+    FFTW_PADDING: ClassVar[Compatibility] = ...
+    _generate_next_value_: ClassVar[Callable] = ...
+    _hashable_values_: ClassVar[list] = ...
+    _member_map_: ClassVar[dict] = ...
+    _member_names_: ClassVar[list] = ...
+    _member_type_: ClassVar[type[int]] = ...
+    _unhashable_values_: ClassVar[list] = ...
+    _unhashable_values_map_: ClassVar[dict] = ...
+    _use_args_: ClassVar[bool] = ...
+    _value2member_map_: ClassVar[dict] = ...
+    def __format__(self, *args, **kwargs) -> str:
+        """Convert to a string according to format_spec."""
+class LibFormat(enum.IntEnum):
+    """
+    See `libFormat_t`.
+    """
+    __new__: ClassVar[Callable] = ...
+    CUFFT: ClassVar[LibFormat] = ...
+    UNDEFINED: ClassVar[LibFormat] = ...
+    _generate_next_value_: ClassVar[Callable] = ...
+    _hashable_values_: ClassVar[list] = ...
+    _member_map_: ClassVar[dict] = ...
+    _member_names_: ClassVar[list] = ...
+    _member_type_: ClassVar[type[int]] = ...
+    _unhashable_values_: ClassVar[list] = ...
+    _unhashable_values_map_: ClassVar[dict] = ...
+    _use_args_: ClassVar[bool] = ...
+    _value2member_map_: ClassVar[dict] = ...
+    def __format__(self, *args, **kwargs) -> str:
+        """Convert to a string according to format_spec."""
+class MpCommType(enum.IntEnum):
+    """
+    See `cufftMpCommType`.
+    """
+    __new__: ClassVar[Callable] = ...
+    COMM_MPI: ClassVar[MpCommType] = ...
+    COMM_NONE: ClassVar[MpCommType] = ...
+    _generate_next_value_: ClassVar[Callable] = ...
+    _hashable_values_: ClassVar[list] = ...
+    _member_map_: ClassVar[dict] = ...
+    _member_names_: ClassVar[list] = ...
+    _member_type_: ClassVar[type[int]] = ...
+    _unhashable_values_: ClassVar[list] = ...
+    _unhashable_values_map_: ClassVar[dict] = ...
+    _use_args_: ClassVar[bool] = ...
+    _value2member_map_: ClassVar[dict] = ...
+    def __format__(self, *args, **kwargs) -> str:
+        """Convert to a string according to format_spec."""
+class Property(enum.IntEnum):
+    """
+    See `cufftProperty`.
+    """
+    __new__: ClassVar[Callable] = ...
+    MAX_NUM_HOST_THREADS: ClassVar[Property] = ...
+    PATIENT_JIT: ClassVar[Property] = ...
+    _generate_next_value_: ClassVar[Callable] = ...
+    _hashable_values_: ClassVar[list] = ...
+    _member_map_: ClassVar[dict] = ...
+    _member_names_: ClassVar[list] = ...
+    _member_type_: ClassVar[type[int]] = ...
+    _unhashable_values_: ClassVar[list] = ...
+    _unhashable_values_map_: ClassVar[dict] = ...
+    _use_args_: ClassVar[bool] = ...
+    _value2member_map_: ClassVar[dict] = ...
+    def __format__(self, *args, **kwargs) -> str:
+        """Convert to a string according to format_spec."""
+class Result(enum.IntEnum):
+    """
+    See `cufftResult`.
+    """
+    __new__: ClassVar[Callable] = ...
+    ALLOC_FAILED: ClassVar[Result] = ...
+    EXEC_FAILED: ClassVar[Result] = ...
+    INCOMPLETE_PARAMETER_LIST: ClassVar[Result] = ...
+    INTERNAL_ERROR: ClassVar[Result] = ...
+    INVALID_DEVICE: ClassVar[Result] = ...
+    INVALID_PLAN: ClassVar[Result] = ...
+    INVALID_SIZE: ClassVar[Result] = ...
+    INVALID_TYPE: ClassVar[Result] = ...
+    INVALID_VALUE: ClassVar[Result] = ...
+    LICENSE_ERROR: ClassVar[Result] = ...
+    NOT_IMPLEMENTED: ClassVar[Result] = ...
+    NOT_SUPPORTED: ClassVar[Result] = ...
+    NO_WORKSPACE: ClassVar[Result] = ...
+    PARSE_ERROR: ClassVar[Result] = ...
+    SETUP_FAILED: ClassVar[Result] = ...
+    SUCCESS: ClassVar[Result] = ...
+    UNALIGNED_DATA: ClassVar[Result] = ...
+    _generate_next_value_: ClassVar[Callable] = ...
+    _hashable_values_: ClassVar[list] = ...
+    _member_map_: ClassVar[dict] = ...
+    _member_names_: ClassVar[list] = ...
+    _member_type_: ClassVar[type[int]] = ...
+    _unhashable_values_: ClassVar[list] = ...
+    _unhashable_values_map_: ClassVar[dict] = ...
+    _use_args_: ClassVar[bool] = ...
+    _value2member_map_: ClassVar[dict] = ...
+    def __format__(self, *args, **kwargs) -> str:
+        """Convert to a string according to format_spec."""
+class Type(enum.IntEnum):
+    """
+    See `cufftType`.
+    """
+    __new__: ClassVar[Callable] = ...
+    C2C: ClassVar[Type] = ...
+    C2R: ClassVar[Type] = ...
+    D2Z: ClassVar[Type] = ...
+    R2C: ClassVar[Type] = ...
+    Z2D: ClassVar[Type] = ...
+    Z2Z: ClassVar[Type] = ...
+    _generate_next_value_: ClassVar[Callable] = ...
+    _hashable_values_: ClassVar[list] = ...
+    _member_map_: ClassVar[dict] = ...
+    _member_names_: ClassVar[list] = ...
+    _member_type_: ClassVar[type[int]] = ...
+    _unhashable_values_: ClassVar[list] = ...
+    _unhashable_values_map_: ClassVar[dict] = ...
+    _use_args_: ClassVar[bool] = ...
+    _value2member_map_: ClassVar[dict] = ...
+    def __format__(self, *args, **kwargs) -> str:
+        """Convert to a string according to format_spec."""
+class XtCallbackType(enum.IntEnum):
+    """
+    See `cufftXtCallbackType`.
+    """
+    __new__: ClassVar[Callable] = ...
+    LD_COMPLEX: ClassVar[XtCallbackType] = ...
+    LD_COMPLEX_DOUBLE: ClassVar[XtCallbackType] = ...
+    LD_REAL: ClassVar[XtCallbackType] = ...
+    LD_REAL_DOUBLE: ClassVar[XtCallbackType] = ...
+    ST_COMPLEX: ClassVar[XtCallbackType] = ...
+    ST_COMPLEX_DOUBLE: ClassVar[XtCallbackType] = ...
+    ST_REAL: ClassVar[XtCallbackType] = ...
+    ST_REAL_DOUBLE: ClassVar[XtCallbackType] = ...
+    UNDEFINED: ClassVar[XtCallbackType] = ...
+    _generate_next_value_: ClassVar[Callable] = ...
+    _hashable_values_: ClassVar[list] = ...
+    _member_map_: ClassVar[dict] = ...
+    _member_names_: ClassVar[list] = ...
+    _member_type_: ClassVar[type[int]] = ...
+    _unhashable_values_: ClassVar[list] = ...
+    _unhashable_values_map_: ClassVar[dict] = ...
+    _use_args_: ClassVar[bool] = ...
+    _value2member_map_: ClassVar[dict] = ...
+    def __format__(self, *args, **kwargs) -> str:
+        """Convert to a string according to format_spec."""
+class XtCopyType(enum.IntEnum):
+    """
+    See `cufftXtCopyType`.
+    """
+    __new__: ClassVar[Callable] = ...
+    DEVICE_TO_DEVICE: ClassVar[XtCopyType] = ...
+    DEVICE_TO_HOST: ClassVar[XtCopyType] = ...
+    HOST_TO_DEVICE: ClassVar[XtCopyType] = ...
+    UNDEFINED: ClassVar[XtCopyType] = ...
+    _generate_next_value_: ClassVar[Callable] = ...
+    _hashable_values_: ClassVar[list] = ...
+    _member_map_: ClassVar[dict] = ...
+    _member_names_: ClassVar[list] = ...
+    _member_type_: ClassVar[type[int]] = ...
+    _unhashable_values_: ClassVar[list] = ...
+    _unhashable_values_map_: ClassVar[dict] = ...
+    _use_args_: ClassVar[bool] = ...
+    _value2member_map_: ClassVar[dict] = ...
+    def __format__(self, *args, **kwargs) -> str:
+        """Convert to a string according to format_spec."""
+class XtQueryType(enum.IntEnum):
+    """
+    See `cufftXtQueryType`.
+    """
+    __new__: ClassVar[Callable] = ...
+    QUERY_1D_FACTORS: ClassVar[XtQueryType] = ...
+    QUERY_UNDEFINED: ClassVar[XtQueryType] = ...
+    _generate_next_value_: ClassVar[Callable] = ...
+    _hashable_values_: ClassVar[list] = ...
+    _member_map_: ClassVar[dict] = ...
+    _member_names_: ClassVar[list] = ...
+    _member_type_: ClassVar[type[int]] = ...
+    _unhashable_values_: ClassVar[list] = ...
+    _unhashable_values_map_: ClassVar[dict] = ...
+    _use_args_: ClassVar[bool] = ...
+    _value2member_map_: ClassVar[dict] = ...
+    def __format__(self, *args, **kwargs) -> str:
+        """Convert to a string according to format_spec."""
+class XtSubFormat(enum.IntEnum):
+    """
+    See `cufftXtSubFormat`.
+    """
+    __new__: ClassVar[Callable] = ...
+    FORMAT_1D_INPUT_SHUFFLED: ClassVar[XtSubFormat] = ...
+    FORMAT_DISTRIBUTED_INPUT: ClassVar[XtSubFormat] = ...
+    FORMAT_DISTRIBUTED_OUTPUT: ClassVar[XtSubFormat] = ...
+    FORMAT_INPLACE: ClassVar[XtSubFormat] = ...
+    FORMAT_INPLACE_SHUFFLED: ClassVar[XtSubFormat] = ...
+    FORMAT_INPUT: ClassVar[XtSubFormat] = ...
+    FORMAT_OUTPUT: ClassVar[XtSubFormat] = ...
+    FORMAT_UNDEFINED: ClassVar[XtSubFormat] = ...
+    _generate_next_value_: ClassVar[Callable] = ...
+    _hashable_values_: ClassVar[list] = ...
+    _member_map_: ClassVar[dict] = ...
+    _member_names_: ClassVar[list] = ...
+    _member_type_: ClassVar[type[int]] = ...
+    _unhashable_values_: ClassVar[list] = ...
+    _unhashable_values_map_: ClassVar[dict] = ...
+    _use_args_: ClassVar[bool] = ...
+    _value2member_map_: ClassVar[dict] = ...
+    def __format__(self, *args, **kwargs) -> str:
+        """Convert to a string according to format_spec."""
+class XtWorkAreaPolicy(enum.IntEnum):
+    """
+    See `cufftXtWorkAreaPolicy`.
+    """
+    __new__: ClassVar[Callable] = ...
+    MINIMAL: ClassVar[XtWorkAreaPolicy] = ...
+    PERFORMANCE: ClassVar[XtWorkAreaPolicy] = ...
+    USER: ClassVar[XtWorkAreaPolicy] = ...
+    _generate_next_value_: ClassVar[Callable] = ...
+    _hashable_values_: ClassVar[list] = ...
+    _member_map_: ClassVar[dict] = ...
+    _member_names_: ClassVar[list] = ...
+    _member_type_: ClassVar[type[int]] = ...
+    _unhashable_values_: ClassVar[list] = ...
+    _unhashable_values_map_: ClassVar[dict] = ...
+    _use_args_: ClassVar[bool] = ...
+    _value2member_map_: ClassVar[dict] = ...
+    def __format__(self, *args, **kwargs) -> str:
+        """Convert to a string according to format_spec."""
+class cuFFTMpError(Exception):
+    def __init__(self, status) -> Any:
+        """cuFFTMpError.__init__(self, status)"""
+    def __reduce__(self) -> Any:
+        """cuFFTMpError.__reduce__(self)"""

nvmath/bindings/curand.cp314t-win_amd64.pyd ADDED Viewed

Binary file

nvmath/bindings/curand.pxd ADDED Viewed

@@ -0,0 +1,71 @@
+# Copyright (c) 2024-2026, NVIDIA CORPORATION & AFFILIATES. ALL RIGHTS RESERVED.
+#
+# SPDX-License-Identifier: Apache-2.0
+#
+# This code was automatically generated across versions from 12.0.1 to 13.2.0, generator version 0.3.1.dev1301+g7215ac36e. Do not modify it directly.
+cimport cython
+from libc.stdint cimport intptr_t
+from .cycurand cimport *
+###############################################################################
+# Types
+###############################################################################
+ctypedef curandGenerator_t Generator
+ctypedef curandDistribution_t Distribution
+ctypedef curandDistributionShift_t DistributionShift
+ctypedef curandDistributionM2Shift_t DistributionM2Shift
+ctypedef curandHistogramM2_t HistogramM2
+ctypedef curandHistogramM2K_t HistogramM2K
+ctypedef curandHistogramM2V_t HistogramM2V
+ctypedef curandDiscreteDistribution_t DiscreteDistribution
+ctypedef cudaStream_t Stream
+ctypedef cudaDataType DataType
+ctypedef libraryPropertyType_t LibraryPropertyType
+###############################################################################
+# Enum
+###############################################################################
+ctypedef curandStatus_t _Status
+ctypedef curandRngType_t _RngType
+ctypedef curandOrdering_t _Ordering
+ctypedef curandDirectionVectorSet_t _DirectionVectorSet
+ctypedef curandMethod_t _Method
+###############################################################################
+# Functions
+###############################################################################
+cpdef intptr_t create_generator(int rng_type) except? 0
+cpdef intptr_t create_generator_host(int rng_type) except? 0
+cpdef destroy_generator(intptr_t generator)
+cpdef int get_version() except? -1
+cpdef int get_property(int type) except? -1
+cpdef set_stream(intptr_t generator, intptr_t stream)
+cpdef set_pseudo_random_generator_seed(intptr_t generator, unsigned long long seed)
+cpdef set_generator_offset(intptr_t generator, unsigned long long offset)
+cpdef set_generator_ordering(intptr_t generator, int order)
+cpdef set_quasi_random_generator_dimensions(intptr_t generator, unsigned int num_dimensions)
+cpdef generate(intptr_t generator, intptr_t output_ptr, size_t num)
+cpdef generate_long_long(intptr_t generator, intptr_t output_ptr, size_t num)
+cpdef generate_uniform(intptr_t generator, intptr_t output_ptr, size_t num)
+cpdef generate_uniform_double(intptr_t generator, intptr_t output_ptr, size_t num)
+cpdef generate_normal(intptr_t generator, intptr_t output_ptr, size_t n, float mean, float stddev)
+cpdef generate_normal_double(intptr_t generator, intptr_t output_ptr, size_t n, double mean, double stddev)
+cpdef generate_log_normal(intptr_t generator, intptr_t output_ptr, size_t n, float mean, float stddev)
+cpdef generate_log_normal_double(intptr_t generator, intptr_t output_ptr, size_t n, double mean, double stddev)
+cpdef create_poisson_distribution(double lambda_, intptr_t discrete_distribution)
+cpdef destroy_distribution(intptr_t discrete_distribution)
+cpdef generate_poisson(intptr_t generator, intptr_t output_ptr, size_t n, double lambda_)
+cpdef generate_poisson_method(intptr_t generator, intptr_t output_ptr, size_t n, double lambda_, int method)
+cpdef generate_binomial(intptr_t generator, intptr_t output_ptr, size_t num, unsigned int n, double p)
+cpdef generate_binomial_method(intptr_t generator, intptr_t output_ptr, size_t num, unsigned int n, double p, int method)
+cpdef generate_seeds(intptr_t generator)

nvmath/bindings/curand.pyi ADDED Viewed

@@ -0,0 +1,189 @@
+# Copyright (c) 2026, NVIDIA CORPORATION & AFFILIATES. ALL RIGHTS RESERVED.
+#
+# SPDX-License-Identifier: Apache-2.0
+import _cython_3_2_5
+import enum
+from typing import Any, Callable, ClassVar
+__pyx_capi__: dict
+__test__: dict
+check_status: _cython_3_2_5.cython_function_or_method
+create_generator: _cython_3_2_5.cython_function_or_method
+create_generator_host: _cython_3_2_5.cython_function_or_method
+create_poisson_distribution: _cython_3_2_5.cython_function_or_method
+destroy_distribution: _cython_3_2_5.cython_function_or_method
+destroy_generator: _cython_3_2_5.cython_function_or_method
+generate: _cython_3_2_5.cython_function_or_method
+generate_binomial: _cython_3_2_5.cython_function_or_method
+generate_binomial_method: _cython_3_2_5.cython_function_or_method
+generate_log_normal: _cython_3_2_5.cython_function_or_method
+generate_log_normal_double: _cython_3_2_5.cython_function_or_method
+generate_long_long: _cython_3_2_5.cython_function_or_method
+generate_normal: _cython_3_2_5.cython_function_or_method
+generate_normal_double: _cython_3_2_5.cython_function_or_method
+generate_poisson: _cython_3_2_5.cython_function_or_method
+generate_poisson_method: _cython_3_2_5.cython_function_or_method
+generate_seeds: _cython_3_2_5.cython_function_or_method
+generate_uniform: _cython_3_2_5.cython_function_or_method
+generate_uniform_double: _cython_3_2_5.cython_function_or_method
+get_direction_vectors32: _cython_3_2_5.cython_function_or_method
+get_direction_vectors64: _cython_3_2_5.cython_function_or_method
+get_property: _cython_3_2_5.cython_function_or_method
+get_scramble_constants32: _cython_3_2_5.cython_function_or_method
+get_scramble_constants64: _cython_3_2_5.cython_function_or_method
+get_version: _cython_3_2_5.cython_function_or_method
+set_generator_offset: _cython_3_2_5.cython_function_or_method
+set_generator_ordering: _cython_3_2_5.cython_function_or_method
+set_pseudo_random_generator_seed: _cython_3_2_5.cython_function_or_method
+set_quasi_random_generator_dimensions: _cython_3_2_5.cython_function_or_method
+set_stream: _cython_3_2_5.cython_function_or_method
+class DirectionVectorSet(enum.IntEnum):
+    """
+    CURAND choice of direction vector set
+    See `curandDirectionVectorSet`.
+    """
+    __new__: ClassVar[Callable] = ...
+    DIRECTION_VECTORS_32_JOEKUO6: ClassVar[DirectionVectorSet] = ...
+    DIRECTION_VECTORS_64_JOEKUO6: ClassVar[DirectionVectorSet] = ...
+    SCRAMBLED_DIRECTION_VECTORS_32_JOEKUO6: ClassVar[DirectionVectorSet] = ...
+    SCRAMBLED_DIRECTION_VECTORS_64_JOEKUO6: ClassVar[DirectionVectorSet] = ...
+    _generate_next_value_: ClassVar[Callable] = ...
+    _hashable_values_: ClassVar[list] = ...
+    _member_map_: ClassVar[dict] = ...
+    _member_names_: ClassVar[list] = ...
+    _member_type_: ClassVar[type[int]] = ...
+    _unhashable_values_: ClassVar[list] = ...
+    _unhashable_values_map_: ClassVar[dict] = ...
+    _use_args_: ClassVar[bool] = ...
+    _value2member_map_: ClassVar[dict] = ...
+    def __format__(self, *args, **kwargs) -> str:
+        """Convert to a string according to format_spec."""
+class Method(enum.IntEnum):
+    """
+    See `curandMethod`.
+    """
+    __new__: ClassVar[Callable] = ...
+    METHOD_3RD: ClassVar[Method] = ...
+    METHOD_BINARY_SEARCH: ClassVar[Method] = ...
+    METHOD_CHOOSE_BEST: ClassVar[Method] = ...
+    METHOD_DEFINITION: ClassVar[Method] = ...
+    METHOD_DEVICE_API: ClassVar[Method] = ...
+    METHOD_DISCRETE_GAUSS: ClassVar[Method] = ...
+    METHOD_FAST_REJECTION: ClassVar[Method] = ...
+    METHOD_HITR: ClassVar[Method] = ...
+    METHOD_ITR: ClassVar[Method] = ...
+    METHOD_KNUTH: ClassVar[Method] = ...
+    METHOD_M1: ClassVar[Method] = ...
+    METHOD_M2: ClassVar[Method] = ...
+    METHOD_POISSON: ClassVar[Method] = ...
+    METHOD_REJECTION: ClassVar[Method] = ...
+    _generate_next_value_: ClassVar[Callable] = ...
+    _hashable_values_: ClassVar[list] = ...
+    _member_map_: ClassVar[dict] = ...
+    _member_names_: ClassVar[list] = ...
+    _member_type_: ClassVar[type[int]] = ...
+    _unhashable_values_: ClassVar[list] = ...
+    _unhashable_values_map_: ClassVar[dict] = ...
+    _use_args_: ClassVar[bool] = ...
+    _value2member_map_: ClassVar[dict] = ...
+    def __format__(self, *args, **kwargs) -> str:
+        """Convert to a string according to format_spec."""
+class Ordering(enum.IntEnum):
+    """
+    CURAND ordering of results in memory
+    See `curandOrdering`.
+    """
+    __new__: ClassVar[Callable] = ...
+    PSEUDO_BEST: ClassVar[Ordering] = ...
+    PSEUDO_DEFAULT: ClassVar[Ordering] = ...
+    PSEUDO_DYNAMIC: ClassVar[Ordering] = ...
+    PSEUDO_LEGACY: ClassVar[Ordering] = ...
+    PSEUDO_SEEDED: ClassVar[Ordering] = ...
+    QUASI_DEFAULT: ClassVar[Ordering] = ...
+    _generate_next_value_: ClassVar[Callable] = ...
+    _hashable_values_: ClassVar[list] = ...
+    _member_map_: ClassVar[dict] = ...
+    _member_names_: ClassVar[list] = ...
+    _member_type_: ClassVar[type[int]] = ...
+    _unhashable_values_: ClassVar[list] = ...
+    _unhashable_values_map_: ClassVar[dict] = ...
+    _use_args_: ClassVar[bool] = ...
+    _value2member_map_: ClassVar[dict] = ...
+    def __format__(self, *args, **kwargs) -> str:
+        """Convert to a string according to format_spec."""
+class RngType(enum.IntEnum):
+    """
+    CURAND generator types
+    See `curandRngType`.
+    """
+    __new__: ClassVar[Callable] = ...
+    PSEUDO_DEFAULT: ClassVar[RngType] = ...
+    PSEUDO_MRG32K3A: ClassVar[RngType] = ...
+    PSEUDO_MT19937: ClassVar[RngType] = ...
+    PSEUDO_MTGP32: ClassVar[RngType] = ...
+    PSEUDO_PHILOX4_32_10: ClassVar[RngType] = ...
+    PSEUDO_XORWOW: ClassVar[RngType] = ...
+    QUASI_DEFAULT: ClassVar[RngType] = ...
+    QUASI_SCRAMBLED_SOBOL32: ClassVar[RngType] = ...
+    QUASI_SCRAMBLED_SOBOL64: ClassVar[RngType] = ...
+    QUASI_SOBOL32: ClassVar[RngType] = ...
+    QUASI_SOBOL64: ClassVar[RngType] = ...
+    TEST: ClassVar[RngType] = ...
+    _generate_next_value_: ClassVar[Callable] = ...
+    _hashable_values_: ClassVar[list] = ...
+    _member_map_: ClassVar[dict] = ...
+    _member_names_: ClassVar[list] = ...
+    _member_type_: ClassVar[type[int]] = ...
+    _unhashable_values_: ClassVar[list] = ...
+    _unhashable_values_map_: ClassVar[dict] = ...
+    _use_args_: ClassVar[bool] = ...
+    _value2member_map_: ClassVar[dict] = ...
+    def __format__(self, *args, **kwargs) -> str:
+        """Convert to a string according to format_spec."""
+class Status(enum.IntEnum):
+    """
+    CURAND function call status types
+    See `curandStatus`.
+    """
+    __new__: ClassVar[Callable] = ...
+    ALLOCATION_FAILED: ClassVar[Status] = ...
+    ARCH_MISMATCH: ClassVar[Status] = ...
+    DOUBLE_PRECISION_REQUIRED: ClassVar[Status] = ...
+    INITIALIZATION_FAILED: ClassVar[Status] = ...
+    INTERNAL_ERROR: ClassVar[Status] = ...
+    LAUNCH_FAILURE: ClassVar[Status] = ...
+    LENGTH_NOT_MULTIPLE: ClassVar[Status] = ...
+    NOT_INITIALIZED: ClassVar[Status] = ...
+    OUT_OF_RANGE: ClassVar[Status] = ...
+    PREEXISTING_FAILURE: ClassVar[Status] = ...
+    SUCCESS: ClassVar[Status] = ...
+    TYPE_ERROR: ClassVar[Status] = ...
+    VERSION_MISMATCH: ClassVar[Status] = ...
+    _generate_next_value_: ClassVar[Callable] = ...
+    _hashable_values_: ClassVar[list] = ...
+    _member_map_: ClassVar[dict] = ...
+    _member_names_: ClassVar[list] = ...
+    _member_type_: ClassVar[type[int]] = ...
+    _unhashable_values_: ClassVar[list] = ...
+    _unhashable_values_map_: ClassVar[dict] = ...
+    _use_args_: ClassVar[bool] = ...
+    _value2member_map_: ClassVar[dict] = ...
+    def __format__(self, *args, **kwargs) -> str:
+        """Convert to a string according to format_spec."""
+class cuRANDError(Exception):
+    """cuRANDError(status)"""
+    def __init__(self, status) -> Any:
+        """Initialize self.  See help(type(self)) for accurate signature."""
+    def __reduce__(self) -> Any:
+        """cuRANDError.__reduce__(self)"""

nvmath/bindings/cusolver.cp314t-win_amd64.pyd ADDED Viewed

Binary file

nvmath/bindings/cusolver.pxd ADDED Viewed

@@ -0,0 +1,62 @@
+# Copyright (c) 2024-2026, NVIDIA CORPORATION & AFFILIATES. ALL RIGHTS RESERVED.
+#
+# SPDX-License-Identifier: Apache-2.0
+#
+# This code was automatically generated across versions from 12.0.1 to 13.2.1, generator version 0.3.1.dev1380+g2c74a7741. Do not modify it directly.
+cimport cython
+from libc.stdint cimport intptr_t
+from .cycusolver cimport *
+###############################################################################
+# Types
+###############################################################################
+ctypedef cudaStream_t Stream
+ctypedef cudaDataType DataType
+ctypedef libraryPropertyType_t LibraryPropertyType
+###############################################################################
+# Enum
+###############################################################################
+ctypedef cusolverStatus_t _Status
+ctypedef cusolverEigType_t _EigType
+ctypedef cusolverEigMode_t _EigMode
+ctypedef cusolverEigRange_t _EigRange
+ctypedef cusolverNorm_t _Norm
+ctypedef cusolverIRSRefinement_t _IRSRefinement
+ctypedef cusolverPrecType_t _PrecType
+ctypedef cusolverAlgMode_t _AlgMode
+ctypedef cusolverStorevMode_t _StorevMode
+ctypedef cusolverDirectMode_t _DirectMode
+ctypedef cusolverDeterministicMode_t _DeterministicMode
+ctypedef cusolverMathMode_t _MathMode
+ctypedef cusolverEigComp_t _EigComp
+###############################################################################
+# Functions
+###############################################################################
+cpdef int get_property(int type) except? -1
+cpdef int get_version() except? -1
+###############################################################################
+# Error handling
+###############################################################################
+cdef class cuSOLVERError(Exception): pass
+@cython.profile(False)
+cpdef inline check_status(int status):
+    if status != 0:
+        raise cuSOLVERError(status)