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.
- 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
|
@@ -0,0 +1,422 @@
|
|
|
1
|
+
# Copyright (c) 2026, NVIDIA CORPORATION & AFFILIATES. ALL RIGHTS RESERVED.
|
|
2
|
+
#
|
|
3
|
+
# SPDX-License-Identifier: Apache-2.0
|
|
4
|
+
|
|
5
|
+
import _cython_3_2_5
|
|
6
|
+
import enum
|
|
7
|
+
from typing import Callable, ClassVar
|
|
8
|
+
|
|
9
|
+
__pyx_capi__: dict
|
|
10
|
+
__test__: dict
|
|
11
|
+
cc_gels: _cython_3_2_5.cython_function_or_method
|
|
12
|
+
cc_gels_buffer_size: _cython_3_2_5.cython_function_or_method
|
|
13
|
+
cc_gesv: _cython_3_2_5.cython_function_or_method
|
|
14
|
+
cc_gesv_buffer_size: _cython_3_2_5.cython_function_or_method
|
|
15
|
+
ce_gels: _cython_3_2_5.cython_function_or_method
|
|
16
|
+
ce_gels_buffer_size: _cython_3_2_5.cython_function_or_method
|
|
17
|
+
ce_gesv: _cython_3_2_5.cython_function_or_method
|
|
18
|
+
ce_gesv_buffer_size: _cython_3_2_5.cython_function_or_method
|
|
19
|
+
cgebrd: _cython_3_2_5.cython_function_or_method
|
|
20
|
+
cgebrd_buffer_size: _cython_3_2_5.cython_function_or_method
|
|
21
|
+
cgeqrf: _cython_3_2_5.cython_function_or_method
|
|
22
|
+
cgeqrf_buffer_size: _cython_3_2_5.cython_function_or_method
|
|
23
|
+
cgesvd: _cython_3_2_5.cython_function_or_method
|
|
24
|
+
cgesvd_buffer_size: _cython_3_2_5.cython_function_or_method
|
|
25
|
+
cgesvda_strided_batched: _cython_3_2_5.cython_function_or_method
|
|
26
|
+
cgesvda_strided_batched_buffer_size: _cython_3_2_5.cython_function_or_method
|
|
27
|
+
cgesvdj: _cython_3_2_5.cython_function_or_method
|
|
28
|
+
cgesvdj_batched: _cython_3_2_5.cython_function_or_method
|
|
29
|
+
cgesvdj_batched_buffer_size: _cython_3_2_5.cython_function_or_method
|
|
30
|
+
cgesvdj_buffer_size: _cython_3_2_5.cython_function_or_method
|
|
31
|
+
cgetrf: _cython_3_2_5.cython_function_or_method
|
|
32
|
+
cgetrf_buffer_size: _cython_3_2_5.cython_function_or_method
|
|
33
|
+
cgetrs: _cython_3_2_5.cython_function_or_method
|
|
34
|
+
cheevd: _cython_3_2_5.cython_function_or_method
|
|
35
|
+
cheevd_buffer_size: _cython_3_2_5.cython_function_or_method
|
|
36
|
+
cheevdx: _cython_3_2_5.cython_function_or_method
|
|
37
|
+
cheevdx_buffer_size: _cython_3_2_5.cython_function_or_method
|
|
38
|
+
cheevj: _cython_3_2_5.cython_function_or_method
|
|
39
|
+
cheevj_batched: _cython_3_2_5.cython_function_or_method
|
|
40
|
+
cheevj_batched_buffer_size: _cython_3_2_5.cython_function_or_method
|
|
41
|
+
cheevj_buffer_size: _cython_3_2_5.cython_function_or_method
|
|
42
|
+
chegvd: _cython_3_2_5.cython_function_or_method
|
|
43
|
+
chegvd_buffer_size: _cython_3_2_5.cython_function_or_method
|
|
44
|
+
chegvdx: _cython_3_2_5.cython_function_or_method
|
|
45
|
+
chegvdx_buffer_size: _cython_3_2_5.cython_function_or_method
|
|
46
|
+
chegvj: _cython_3_2_5.cython_function_or_method
|
|
47
|
+
chegvj_buffer_size: _cython_3_2_5.cython_function_or_method
|
|
48
|
+
chetrd: _cython_3_2_5.cython_function_or_method
|
|
49
|
+
chetrd_buffer_size: _cython_3_2_5.cython_function_or_method
|
|
50
|
+
ck_gels: _cython_3_2_5.cython_function_or_method
|
|
51
|
+
ck_gels_buffer_size: _cython_3_2_5.cython_function_or_method
|
|
52
|
+
ck_gesv: _cython_3_2_5.cython_function_or_method
|
|
53
|
+
ck_gesv_buffer_size: _cython_3_2_5.cython_function_or_method
|
|
54
|
+
claswp: _cython_3_2_5.cython_function_or_method
|
|
55
|
+
clauum: _cython_3_2_5.cython_function_or_method
|
|
56
|
+
clauum_buffer_size: _cython_3_2_5.cython_function_or_method
|
|
57
|
+
cpotrf: _cython_3_2_5.cython_function_or_method
|
|
58
|
+
cpotrf_batched: _cython_3_2_5.cython_function_or_method
|
|
59
|
+
cpotrf_buffer_size: _cython_3_2_5.cython_function_or_method
|
|
60
|
+
cpotri: _cython_3_2_5.cython_function_or_method
|
|
61
|
+
cpotri_buffer_size: _cython_3_2_5.cython_function_or_method
|
|
62
|
+
cpotrs: _cython_3_2_5.cython_function_or_method
|
|
63
|
+
cpotrs_batched: _cython_3_2_5.cython_function_or_method
|
|
64
|
+
create: _cython_3_2_5.cython_function_or_method
|
|
65
|
+
create_gesvdj_info: _cython_3_2_5.cython_function_or_method
|
|
66
|
+
create_params: _cython_3_2_5.cython_function_or_method
|
|
67
|
+
create_syevj_info: _cython_3_2_5.cython_function_or_method
|
|
68
|
+
csytrf: _cython_3_2_5.cython_function_or_method
|
|
69
|
+
csytrf_buffer_size: _cython_3_2_5.cython_function_or_method
|
|
70
|
+
csytri: _cython_3_2_5.cython_function_or_method
|
|
71
|
+
csytri_buffer_size: _cython_3_2_5.cython_function_or_method
|
|
72
|
+
cungbr: _cython_3_2_5.cython_function_or_method
|
|
73
|
+
cungbr_buffer_size: _cython_3_2_5.cython_function_or_method
|
|
74
|
+
cungqr: _cython_3_2_5.cython_function_or_method
|
|
75
|
+
cungqr_buffer_size: _cython_3_2_5.cython_function_or_method
|
|
76
|
+
cungtr: _cython_3_2_5.cython_function_or_method
|
|
77
|
+
cungtr_buffer_size: _cython_3_2_5.cython_function_or_method
|
|
78
|
+
cunmqr: _cython_3_2_5.cython_function_or_method
|
|
79
|
+
cunmqr_buffer_size: _cython_3_2_5.cython_function_or_method
|
|
80
|
+
cunmtr: _cython_3_2_5.cython_function_or_method
|
|
81
|
+
cunmtr_buffer_size: _cython_3_2_5.cython_function_or_method
|
|
82
|
+
cy_gels: _cython_3_2_5.cython_function_or_method
|
|
83
|
+
cy_gels_buffer_size: _cython_3_2_5.cython_function_or_method
|
|
84
|
+
cy_gesv: _cython_3_2_5.cython_function_or_method
|
|
85
|
+
cy_gesv_buffer_size: _cython_3_2_5.cython_function_or_method
|
|
86
|
+
db_gels: _cython_3_2_5.cython_function_or_method
|
|
87
|
+
db_gels_buffer_size: _cython_3_2_5.cython_function_or_method
|
|
88
|
+
db_gesv: _cython_3_2_5.cython_function_or_method
|
|
89
|
+
db_gesv_buffer_size: _cython_3_2_5.cython_function_or_method
|
|
90
|
+
dd_gels: _cython_3_2_5.cython_function_or_method
|
|
91
|
+
dd_gels_buffer_size: _cython_3_2_5.cython_function_or_method
|
|
92
|
+
dd_gesv: _cython_3_2_5.cython_function_or_method
|
|
93
|
+
dd_gesv_buffer_size: _cython_3_2_5.cython_function_or_method
|
|
94
|
+
destroy: _cython_3_2_5.cython_function_or_method
|
|
95
|
+
destroy_gesvdj_info: _cython_3_2_5.cython_function_or_method
|
|
96
|
+
destroy_params: _cython_3_2_5.cython_function_or_method
|
|
97
|
+
destroy_syevj_info: _cython_3_2_5.cython_function_or_method
|
|
98
|
+
dgebrd: _cython_3_2_5.cython_function_or_method
|
|
99
|
+
dgebrd_buffer_size: _cython_3_2_5.cython_function_or_method
|
|
100
|
+
dgeqrf: _cython_3_2_5.cython_function_or_method
|
|
101
|
+
dgeqrf_buffer_size: _cython_3_2_5.cython_function_or_method
|
|
102
|
+
dgesvd: _cython_3_2_5.cython_function_or_method
|
|
103
|
+
dgesvd_buffer_size: _cython_3_2_5.cython_function_or_method
|
|
104
|
+
dgesvda_strided_batched: _cython_3_2_5.cython_function_or_method
|
|
105
|
+
dgesvda_strided_batched_buffer_size: _cython_3_2_5.cython_function_or_method
|
|
106
|
+
dgesvdj: _cython_3_2_5.cython_function_or_method
|
|
107
|
+
dgesvdj_batched: _cython_3_2_5.cython_function_or_method
|
|
108
|
+
dgesvdj_batched_buffer_size: _cython_3_2_5.cython_function_or_method
|
|
109
|
+
dgesvdj_buffer_size: _cython_3_2_5.cython_function_or_method
|
|
110
|
+
dgetrf: _cython_3_2_5.cython_function_or_method
|
|
111
|
+
dgetrf_buffer_size: _cython_3_2_5.cython_function_or_method
|
|
112
|
+
dgetrs: _cython_3_2_5.cython_function_or_method
|
|
113
|
+
dh_gels: _cython_3_2_5.cython_function_or_method
|
|
114
|
+
dh_gels_buffer_size: _cython_3_2_5.cython_function_or_method
|
|
115
|
+
dh_gesv: _cython_3_2_5.cython_function_or_method
|
|
116
|
+
dh_gesv_buffer_size: _cython_3_2_5.cython_function_or_method
|
|
117
|
+
dlaswp: _cython_3_2_5.cython_function_or_method
|
|
118
|
+
dlauum: _cython_3_2_5.cython_function_or_method
|
|
119
|
+
dlauum_buffer_size: _cython_3_2_5.cython_function_or_method
|
|
120
|
+
dorgbr: _cython_3_2_5.cython_function_or_method
|
|
121
|
+
dorgbr_buffer_size: _cython_3_2_5.cython_function_or_method
|
|
122
|
+
dorgqr: _cython_3_2_5.cython_function_or_method
|
|
123
|
+
dorgqr_buffer_size: _cython_3_2_5.cython_function_or_method
|
|
124
|
+
dorgtr: _cython_3_2_5.cython_function_or_method
|
|
125
|
+
dorgtr_buffer_size: _cython_3_2_5.cython_function_or_method
|
|
126
|
+
dormqr: _cython_3_2_5.cython_function_or_method
|
|
127
|
+
dormqr_buffer_size: _cython_3_2_5.cython_function_or_method
|
|
128
|
+
dormtr: _cython_3_2_5.cython_function_or_method
|
|
129
|
+
dormtr_buffer_size: _cython_3_2_5.cython_function_or_method
|
|
130
|
+
dpotrf: _cython_3_2_5.cython_function_or_method
|
|
131
|
+
dpotrf_batched: _cython_3_2_5.cython_function_or_method
|
|
132
|
+
dpotrf_buffer_size: _cython_3_2_5.cython_function_or_method
|
|
133
|
+
dpotri: _cython_3_2_5.cython_function_or_method
|
|
134
|
+
dpotri_buffer_size: _cython_3_2_5.cython_function_or_method
|
|
135
|
+
dpotrs: _cython_3_2_5.cython_function_or_method
|
|
136
|
+
dpotrs_batched: _cython_3_2_5.cython_function_or_method
|
|
137
|
+
ds_gels: _cython_3_2_5.cython_function_or_method
|
|
138
|
+
ds_gels_buffer_size: _cython_3_2_5.cython_function_or_method
|
|
139
|
+
ds_gesv: _cython_3_2_5.cython_function_or_method
|
|
140
|
+
ds_gesv_buffer_size: _cython_3_2_5.cython_function_or_method
|
|
141
|
+
dsyevd: _cython_3_2_5.cython_function_or_method
|
|
142
|
+
dsyevd_buffer_size: _cython_3_2_5.cython_function_or_method
|
|
143
|
+
dsyevdx: _cython_3_2_5.cython_function_or_method
|
|
144
|
+
dsyevdx_buffer_size: _cython_3_2_5.cython_function_or_method
|
|
145
|
+
dsyevj: _cython_3_2_5.cython_function_or_method
|
|
146
|
+
dsyevj_batched: _cython_3_2_5.cython_function_or_method
|
|
147
|
+
dsyevj_batched_buffer_size: _cython_3_2_5.cython_function_or_method
|
|
148
|
+
dsyevj_buffer_size: _cython_3_2_5.cython_function_or_method
|
|
149
|
+
dsygvd: _cython_3_2_5.cython_function_or_method
|
|
150
|
+
dsygvd_buffer_size: _cython_3_2_5.cython_function_or_method
|
|
151
|
+
dsygvdx: _cython_3_2_5.cython_function_or_method
|
|
152
|
+
dsygvdx_buffer_size: _cython_3_2_5.cython_function_or_method
|
|
153
|
+
dsygvj: _cython_3_2_5.cython_function_or_method
|
|
154
|
+
dsygvj_buffer_size: _cython_3_2_5.cython_function_or_method
|
|
155
|
+
dsytrd: _cython_3_2_5.cython_function_or_method
|
|
156
|
+
dsytrd_buffer_size: _cython_3_2_5.cython_function_or_method
|
|
157
|
+
dsytrf: _cython_3_2_5.cython_function_or_method
|
|
158
|
+
dsytrf_buffer_size: _cython_3_2_5.cython_function_or_method
|
|
159
|
+
dsytri: _cython_3_2_5.cython_function_or_method
|
|
160
|
+
dsytri_buffer_size: _cython_3_2_5.cython_function_or_method
|
|
161
|
+
dx_gels: _cython_3_2_5.cython_function_or_method
|
|
162
|
+
dx_gels_buffer_size: _cython_3_2_5.cython_function_or_method
|
|
163
|
+
dx_gesv: _cython_3_2_5.cython_function_or_method
|
|
164
|
+
dx_gesv_buffer_size: _cython_3_2_5.cython_function_or_method
|
|
165
|
+
get_deterministic_mode: _cython_3_2_5.cython_function_or_method
|
|
166
|
+
get_emulation_special_values_support: _cython_3_2_5.cython_function_or_method
|
|
167
|
+
get_emulation_strategy: _cython_3_2_5.cython_function_or_method
|
|
168
|
+
get_fixed_point_emulation_mantissa_bit_offset: _cython_3_2_5.cython_function_or_method
|
|
169
|
+
get_fixed_point_emulation_mantissa_control: _cython_3_2_5.cython_function_or_method
|
|
170
|
+
get_fixed_point_emulation_max_mantissa_bit_count: _cython_3_2_5.cython_function_or_method
|
|
171
|
+
get_math_mode: _cython_3_2_5.cython_function_or_method
|
|
172
|
+
get_stream: _cython_3_2_5.cython_function_or_method
|
|
173
|
+
irs_infos_create: _cython_3_2_5.cython_function_or_method
|
|
174
|
+
irs_infos_destroy: _cython_3_2_5.cython_function_or_method
|
|
175
|
+
irs_infos_get_max_iters: _cython_3_2_5.cython_function_or_method
|
|
176
|
+
irs_infos_get_niters: _cython_3_2_5.cython_function_or_method
|
|
177
|
+
irs_infos_get_outer_niters: _cython_3_2_5.cython_function_or_method
|
|
178
|
+
irs_infos_get_residual_history: _cython_3_2_5.cython_function_or_method
|
|
179
|
+
irs_infos_request_residual: _cython_3_2_5.cython_function_or_method
|
|
180
|
+
irs_params_create: _cython_3_2_5.cython_function_or_method
|
|
181
|
+
irs_params_destroy: _cython_3_2_5.cython_function_or_method
|
|
182
|
+
irs_params_disable_fallback: _cython_3_2_5.cython_function_or_method
|
|
183
|
+
irs_params_enable_fallback: _cython_3_2_5.cython_function_or_method
|
|
184
|
+
irs_params_get_max_iters: _cython_3_2_5.cython_function_or_method
|
|
185
|
+
irs_params_set_max_iters: _cython_3_2_5.cython_function_or_method
|
|
186
|
+
irs_params_set_max_iters_inner: _cython_3_2_5.cython_function_or_method
|
|
187
|
+
irs_params_set_refinement_solver: _cython_3_2_5.cython_function_or_method
|
|
188
|
+
irs_params_set_solver_lowest_precision: _cython_3_2_5.cython_function_or_method
|
|
189
|
+
irs_params_set_solver_main_precision: _cython_3_2_5.cython_function_or_method
|
|
190
|
+
irs_params_set_solver_precisions: _cython_3_2_5.cython_function_or_method
|
|
191
|
+
irs_params_set_tol: _cython_3_2_5.cython_function_or_method
|
|
192
|
+
irs_params_set_tol_inner: _cython_3_2_5.cython_function_or_method
|
|
193
|
+
irs_xgels: _cython_3_2_5.cython_function_or_method
|
|
194
|
+
irs_xgels_buffer_size: _cython_3_2_5.cython_function_or_method
|
|
195
|
+
irs_xgesv: _cython_3_2_5.cython_function_or_method
|
|
196
|
+
irs_xgesv_buffer_size: _cython_3_2_5.cython_function_or_method
|
|
197
|
+
logger_force_disable: _cython_3_2_5.cython_function_or_method
|
|
198
|
+
logger_open_file: _cython_3_2_5.cython_function_or_method
|
|
199
|
+
logger_set_level: _cython_3_2_5.cython_function_or_method
|
|
200
|
+
logger_set_mask: _cython_3_2_5.cython_function_or_method
|
|
201
|
+
sb_gels: _cython_3_2_5.cython_function_or_method
|
|
202
|
+
sb_gels_buffer_size: _cython_3_2_5.cython_function_or_method
|
|
203
|
+
sb_gesv: _cython_3_2_5.cython_function_or_method
|
|
204
|
+
sb_gesv_buffer_size: _cython_3_2_5.cython_function_or_method
|
|
205
|
+
set_adv_options: _cython_3_2_5.cython_function_or_method
|
|
206
|
+
set_deterministic_mode: _cython_3_2_5.cython_function_or_method
|
|
207
|
+
set_emulation_special_values_support: _cython_3_2_5.cython_function_or_method
|
|
208
|
+
set_emulation_strategy: _cython_3_2_5.cython_function_or_method
|
|
209
|
+
set_fixed_point_emulation_mantissa_bit_offset: _cython_3_2_5.cython_function_or_method
|
|
210
|
+
set_fixed_point_emulation_mantissa_control: _cython_3_2_5.cython_function_or_method
|
|
211
|
+
set_fixed_point_emulation_max_mantissa_bit_count: _cython_3_2_5.cython_function_or_method
|
|
212
|
+
set_math_mode: _cython_3_2_5.cython_function_or_method
|
|
213
|
+
set_stream: _cython_3_2_5.cython_function_or_method
|
|
214
|
+
sgebrd: _cython_3_2_5.cython_function_or_method
|
|
215
|
+
sgebrd_buffer_size: _cython_3_2_5.cython_function_or_method
|
|
216
|
+
sgeqrf: _cython_3_2_5.cython_function_or_method
|
|
217
|
+
sgeqrf_buffer_size: _cython_3_2_5.cython_function_or_method
|
|
218
|
+
sgesvd: _cython_3_2_5.cython_function_or_method
|
|
219
|
+
sgesvd_buffer_size: _cython_3_2_5.cython_function_or_method
|
|
220
|
+
sgesvda_strided_batched: _cython_3_2_5.cython_function_or_method
|
|
221
|
+
sgesvda_strided_batched_buffer_size: _cython_3_2_5.cython_function_or_method
|
|
222
|
+
sgesvdj: _cython_3_2_5.cython_function_or_method
|
|
223
|
+
sgesvdj_batched: _cython_3_2_5.cython_function_or_method
|
|
224
|
+
sgesvdj_batched_buffer_size: _cython_3_2_5.cython_function_or_method
|
|
225
|
+
sgesvdj_buffer_size: _cython_3_2_5.cython_function_or_method
|
|
226
|
+
sgetrf: _cython_3_2_5.cython_function_or_method
|
|
227
|
+
sgetrf_buffer_size: _cython_3_2_5.cython_function_or_method
|
|
228
|
+
sgetrs: _cython_3_2_5.cython_function_or_method
|
|
229
|
+
sh_gels: _cython_3_2_5.cython_function_or_method
|
|
230
|
+
sh_gels_buffer_size: _cython_3_2_5.cython_function_or_method
|
|
231
|
+
sh_gesv: _cython_3_2_5.cython_function_or_method
|
|
232
|
+
sh_gesv_buffer_size: _cython_3_2_5.cython_function_or_method
|
|
233
|
+
slaswp: _cython_3_2_5.cython_function_or_method
|
|
234
|
+
slauum: _cython_3_2_5.cython_function_or_method
|
|
235
|
+
slauum_buffer_size: _cython_3_2_5.cython_function_or_method
|
|
236
|
+
sorgbr: _cython_3_2_5.cython_function_or_method
|
|
237
|
+
sorgbr_buffer_size: _cython_3_2_5.cython_function_or_method
|
|
238
|
+
sorgqr: _cython_3_2_5.cython_function_or_method
|
|
239
|
+
sorgqr_buffer_size: _cython_3_2_5.cython_function_or_method
|
|
240
|
+
sorgtr: _cython_3_2_5.cython_function_or_method
|
|
241
|
+
sorgtr_buffer_size: _cython_3_2_5.cython_function_or_method
|
|
242
|
+
sormqr: _cython_3_2_5.cython_function_or_method
|
|
243
|
+
sormqr_buffer_size: _cython_3_2_5.cython_function_or_method
|
|
244
|
+
sormtr: _cython_3_2_5.cython_function_or_method
|
|
245
|
+
sormtr_buffer_size: _cython_3_2_5.cython_function_or_method
|
|
246
|
+
spotrf: _cython_3_2_5.cython_function_or_method
|
|
247
|
+
spotrf_batched: _cython_3_2_5.cython_function_or_method
|
|
248
|
+
spotrf_buffer_size: _cython_3_2_5.cython_function_or_method
|
|
249
|
+
spotri: _cython_3_2_5.cython_function_or_method
|
|
250
|
+
spotri_buffer_size: _cython_3_2_5.cython_function_or_method
|
|
251
|
+
spotrs: _cython_3_2_5.cython_function_or_method
|
|
252
|
+
spotrs_batched: _cython_3_2_5.cython_function_or_method
|
|
253
|
+
ss_gels: _cython_3_2_5.cython_function_or_method
|
|
254
|
+
ss_gels_buffer_size: _cython_3_2_5.cython_function_or_method
|
|
255
|
+
ss_gesv: _cython_3_2_5.cython_function_or_method
|
|
256
|
+
ss_gesv_buffer_size: _cython_3_2_5.cython_function_or_method
|
|
257
|
+
ssyevd: _cython_3_2_5.cython_function_or_method
|
|
258
|
+
ssyevd_buffer_size: _cython_3_2_5.cython_function_or_method
|
|
259
|
+
ssyevdx: _cython_3_2_5.cython_function_or_method
|
|
260
|
+
ssyevdx_buffer_size: _cython_3_2_5.cython_function_or_method
|
|
261
|
+
ssyevj: _cython_3_2_5.cython_function_or_method
|
|
262
|
+
ssyevj_batched: _cython_3_2_5.cython_function_or_method
|
|
263
|
+
ssyevj_batched_buffer_size: _cython_3_2_5.cython_function_or_method
|
|
264
|
+
ssyevj_buffer_size: _cython_3_2_5.cython_function_or_method
|
|
265
|
+
ssygvd: _cython_3_2_5.cython_function_or_method
|
|
266
|
+
ssygvd_buffer_size: _cython_3_2_5.cython_function_or_method
|
|
267
|
+
ssygvdx: _cython_3_2_5.cython_function_or_method
|
|
268
|
+
ssygvdx_buffer_size: _cython_3_2_5.cython_function_or_method
|
|
269
|
+
ssygvj: _cython_3_2_5.cython_function_or_method
|
|
270
|
+
ssygvj_buffer_size: _cython_3_2_5.cython_function_or_method
|
|
271
|
+
ssytrd: _cython_3_2_5.cython_function_or_method
|
|
272
|
+
ssytrd_buffer_size: _cython_3_2_5.cython_function_or_method
|
|
273
|
+
ssytrf: _cython_3_2_5.cython_function_or_method
|
|
274
|
+
ssytrf_buffer_size: _cython_3_2_5.cython_function_or_method
|
|
275
|
+
ssytri: _cython_3_2_5.cython_function_or_method
|
|
276
|
+
ssytri_buffer_size: _cython_3_2_5.cython_function_or_method
|
|
277
|
+
sx_gels: _cython_3_2_5.cython_function_or_method
|
|
278
|
+
sx_gels_buffer_size: _cython_3_2_5.cython_function_or_method
|
|
279
|
+
sx_gesv: _cython_3_2_5.cython_function_or_method
|
|
280
|
+
sx_gesv_buffer_size: _cython_3_2_5.cython_function_or_method
|
|
281
|
+
xgeev: _cython_3_2_5.cython_function_or_method
|
|
282
|
+
xgeev_buffer_size: _cython_3_2_5.cython_function_or_method
|
|
283
|
+
xgeqrf: _cython_3_2_5.cython_function_or_method
|
|
284
|
+
xgeqrf_buffer_size: _cython_3_2_5.cython_function_or_method
|
|
285
|
+
xgesvd: _cython_3_2_5.cython_function_or_method
|
|
286
|
+
xgesvd_buffer_size: _cython_3_2_5.cython_function_or_method
|
|
287
|
+
xgesvdj_get_residual: _cython_3_2_5.cython_function_or_method
|
|
288
|
+
xgesvdj_get_sweeps: _cython_3_2_5.cython_function_or_method
|
|
289
|
+
xgesvdj_set_max_sweeps: _cython_3_2_5.cython_function_or_method
|
|
290
|
+
xgesvdj_set_sort_eig: _cython_3_2_5.cython_function_or_method
|
|
291
|
+
xgesvdj_set_tolerance: _cython_3_2_5.cython_function_or_method
|
|
292
|
+
xgesvdp: _cython_3_2_5.cython_function_or_method
|
|
293
|
+
xgesvdp_buffer_size: _cython_3_2_5.cython_function_or_method
|
|
294
|
+
xgesvdr: _cython_3_2_5.cython_function_or_method
|
|
295
|
+
xgesvdr_buffer_size: _cython_3_2_5.cython_function_or_method
|
|
296
|
+
xgetrf: _cython_3_2_5.cython_function_or_method
|
|
297
|
+
xgetrf_buffer_size: _cython_3_2_5.cython_function_or_method
|
|
298
|
+
xgetrs: _cython_3_2_5.cython_function_or_method
|
|
299
|
+
xlarft: _cython_3_2_5.cython_function_or_method
|
|
300
|
+
xlarft_buffer_size: _cython_3_2_5.cython_function_or_method
|
|
301
|
+
xpolar: _cython_3_2_5.cython_function_or_method
|
|
302
|
+
xpolar_buffer_size: _cython_3_2_5.cython_function_or_method
|
|
303
|
+
xpotrf: _cython_3_2_5.cython_function_or_method
|
|
304
|
+
xpotrf_buffer_size: _cython_3_2_5.cython_function_or_method
|
|
305
|
+
xpotrs: _cython_3_2_5.cython_function_or_method
|
|
306
|
+
xstedc: _cython_3_2_5.cython_function_or_method
|
|
307
|
+
xstedc_buffer_size: _cython_3_2_5.cython_function_or_method
|
|
308
|
+
xsyev_batched: _cython_3_2_5.cython_function_or_method
|
|
309
|
+
xsyev_batched_buffer_size: _cython_3_2_5.cython_function_or_method
|
|
310
|
+
xsyevd: _cython_3_2_5.cython_function_or_method
|
|
311
|
+
xsyevd_buffer_size: _cython_3_2_5.cython_function_or_method
|
|
312
|
+
xsyevdx: _cython_3_2_5.cython_function_or_method
|
|
313
|
+
xsyevdx_buffer_size: _cython_3_2_5.cython_function_or_method
|
|
314
|
+
xsyevj_get_residual: _cython_3_2_5.cython_function_or_method
|
|
315
|
+
xsyevj_get_sweeps: _cython_3_2_5.cython_function_or_method
|
|
316
|
+
xsyevj_set_max_sweeps: _cython_3_2_5.cython_function_or_method
|
|
317
|
+
xsyevj_set_sort_eig: _cython_3_2_5.cython_function_or_method
|
|
318
|
+
xsyevj_set_tolerance: _cython_3_2_5.cython_function_or_method
|
|
319
|
+
xsygvd: _cython_3_2_5.cython_function_or_method
|
|
320
|
+
xsygvd_buffer_size: _cython_3_2_5.cython_function_or_method
|
|
321
|
+
xsygvdx: _cython_3_2_5.cython_function_or_method
|
|
322
|
+
xsygvdx_buffer_size: _cython_3_2_5.cython_function_or_method
|
|
323
|
+
xsytrs: _cython_3_2_5.cython_function_or_method
|
|
324
|
+
xsytrs_buffer_size: _cython_3_2_5.cython_function_or_method
|
|
325
|
+
xtrtri: _cython_3_2_5.cython_function_or_method
|
|
326
|
+
xtrtri_buffer_size: _cython_3_2_5.cython_function_or_method
|
|
327
|
+
zc_gels: _cython_3_2_5.cython_function_or_method
|
|
328
|
+
zc_gels_buffer_size: _cython_3_2_5.cython_function_or_method
|
|
329
|
+
zc_gesv: _cython_3_2_5.cython_function_or_method
|
|
330
|
+
zc_gesv_buffer_size: _cython_3_2_5.cython_function_or_method
|
|
331
|
+
ze_gels: _cython_3_2_5.cython_function_or_method
|
|
332
|
+
ze_gels_buffer_size: _cython_3_2_5.cython_function_or_method
|
|
333
|
+
ze_gesv: _cython_3_2_5.cython_function_or_method
|
|
334
|
+
ze_gesv_buffer_size: _cython_3_2_5.cython_function_or_method
|
|
335
|
+
zgebrd: _cython_3_2_5.cython_function_or_method
|
|
336
|
+
zgebrd_buffer_size: _cython_3_2_5.cython_function_or_method
|
|
337
|
+
zgeqrf: _cython_3_2_5.cython_function_or_method
|
|
338
|
+
zgeqrf_buffer_size: _cython_3_2_5.cython_function_or_method
|
|
339
|
+
zgesvd: _cython_3_2_5.cython_function_or_method
|
|
340
|
+
zgesvd_buffer_size: _cython_3_2_5.cython_function_or_method
|
|
341
|
+
zgesvda_strided_batched: _cython_3_2_5.cython_function_or_method
|
|
342
|
+
zgesvda_strided_batched_buffer_size: _cython_3_2_5.cython_function_or_method
|
|
343
|
+
zgesvdj: _cython_3_2_5.cython_function_or_method
|
|
344
|
+
zgesvdj_batched: _cython_3_2_5.cython_function_or_method
|
|
345
|
+
zgesvdj_batched_buffer_size: _cython_3_2_5.cython_function_or_method
|
|
346
|
+
zgesvdj_buffer_size: _cython_3_2_5.cython_function_or_method
|
|
347
|
+
zgetrf: _cython_3_2_5.cython_function_or_method
|
|
348
|
+
zgetrf_buffer_size: _cython_3_2_5.cython_function_or_method
|
|
349
|
+
zgetrs: _cython_3_2_5.cython_function_or_method
|
|
350
|
+
zheevd: _cython_3_2_5.cython_function_or_method
|
|
351
|
+
zheevd_buffer_size: _cython_3_2_5.cython_function_or_method
|
|
352
|
+
zheevdx: _cython_3_2_5.cython_function_or_method
|
|
353
|
+
zheevdx_buffer_size: _cython_3_2_5.cython_function_or_method
|
|
354
|
+
zheevj: _cython_3_2_5.cython_function_or_method
|
|
355
|
+
zheevj_batched: _cython_3_2_5.cython_function_or_method
|
|
356
|
+
zheevj_batched_buffer_size: _cython_3_2_5.cython_function_or_method
|
|
357
|
+
zheevj_buffer_size: _cython_3_2_5.cython_function_or_method
|
|
358
|
+
zhegvd: _cython_3_2_5.cython_function_or_method
|
|
359
|
+
zhegvd_buffer_size: _cython_3_2_5.cython_function_or_method
|
|
360
|
+
zhegvdx: _cython_3_2_5.cython_function_or_method
|
|
361
|
+
zhegvdx_buffer_size: _cython_3_2_5.cython_function_or_method
|
|
362
|
+
zhegvj: _cython_3_2_5.cython_function_or_method
|
|
363
|
+
zhegvj_buffer_size: _cython_3_2_5.cython_function_or_method
|
|
364
|
+
zhetrd: _cython_3_2_5.cython_function_or_method
|
|
365
|
+
zhetrd_buffer_size: _cython_3_2_5.cython_function_or_method
|
|
366
|
+
zk_gels: _cython_3_2_5.cython_function_or_method
|
|
367
|
+
zk_gels_buffer_size: _cython_3_2_5.cython_function_or_method
|
|
368
|
+
zk_gesv: _cython_3_2_5.cython_function_or_method
|
|
369
|
+
zk_gesv_buffer_size: _cython_3_2_5.cython_function_or_method
|
|
370
|
+
zlaswp: _cython_3_2_5.cython_function_or_method
|
|
371
|
+
zlauum: _cython_3_2_5.cython_function_or_method
|
|
372
|
+
zlauum_buffer_size: _cython_3_2_5.cython_function_or_method
|
|
373
|
+
zpotrf: _cython_3_2_5.cython_function_or_method
|
|
374
|
+
zpotrf_batched: _cython_3_2_5.cython_function_or_method
|
|
375
|
+
zpotrf_buffer_size: _cython_3_2_5.cython_function_or_method
|
|
376
|
+
zpotri: _cython_3_2_5.cython_function_or_method
|
|
377
|
+
zpotri_buffer_size: _cython_3_2_5.cython_function_or_method
|
|
378
|
+
zpotrs: _cython_3_2_5.cython_function_or_method
|
|
379
|
+
zpotrs_batched: _cython_3_2_5.cython_function_or_method
|
|
380
|
+
zsytrf: _cython_3_2_5.cython_function_or_method
|
|
381
|
+
zsytrf_buffer_size: _cython_3_2_5.cython_function_or_method
|
|
382
|
+
zsytri: _cython_3_2_5.cython_function_or_method
|
|
383
|
+
zsytri_buffer_size: _cython_3_2_5.cython_function_or_method
|
|
384
|
+
zungbr: _cython_3_2_5.cython_function_or_method
|
|
385
|
+
zungbr_buffer_size: _cython_3_2_5.cython_function_or_method
|
|
386
|
+
zungqr: _cython_3_2_5.cython_function_or_method
|
|
387
|
+
zungqr_buffer_size: _cython_3_2_5.cython_function_or_method
|
|
388
|
+
zungtr: _cython_3_2_5.cython_function_or_method
|
|
389
|
+
zungtr_buffer_size: _cython_3_2_5.cython_function_or_method
|
|
390
|
+
zunmqr: _cython_3_2_5.cython_function_or_method
|
|
391
|
+
zunmqr_buffer_size: _cython_3_2_5.cython_function_or_method
|
|
392
|
+
zunmtr: _cython_3_2_5.cython_function_or_method
|
|
393
|
+
zunmtr_buffer_size: _cython_3_2_5.cython_function_or_method
|
|
394
|
+
zy_gels: _cython_3_2_5.cython_function_or_method
|
|
395
|
+
zy_gels_buffer_size: _cython_3_2_5.cython_function_or_method
|
|
396
|
+
zy_gesv: _cython_3_2_5.cython_function_or_method
|
|
397
|
+
zy_gesv_buffer_size: _cython_3_2_5.cython_function_or_method
|
|
398
|
+
zz_gels: _cython_3_2_5.cython_function_or_method
|
|
399
|
+
zz_gels_buffer_size: _cython_3_2_5.cython_function_or_method
|
|
400
|
+
zz_gesv: _cython_3_2_5.cython_function_or_method
|
|
401
|
+
zz_gesv_buffer_size: _cython_3_2_5.cython_function_or_method
|
|
402
|
+
|
|
403
|
+
class Function(enum.IntEnum):
|
|
404
|
+
"""
|
|
405
|
+
See `cusolverDnFunction_t`.
|
|
406
|
+
"""
|
|
407
|
+
__new__: ClassVar[Callable] = ...
|
|
408
|
+
GEQRF: ClassVar[Function] = ...
|
|
409
|
+
GETRF: ClassVar[Function] = ...
|
|
410
|
+
POTRF: ClassVar[Function] = ...
|
|
411
|
+
SYEVBATCHED: ClassVar[Function] = ...
|
|
412
|
+
_generate_next_value_: ClassVar[Callable] = ...
|
|
413
|
+
_hashable_values_: ClassVar[list] = ...
|
|
414
|
+
_member_map_: ClassVar[dict] = ...
|
|
415
|
+
_member_names_: ClassVar[list] = ...
|
|
416
|
+
_member_type_: ClassVar[type[int]] = ...
|
|
417
|
+
_unhashable_values_: ClassVar[list] = ...
|
|
418
|
+
_unhashable_values_map_: ClassVar[dict] = ...
|
|
419
|
+
_use_args_: ClassVar[bool] = ...
|
|
420
|
+
_value2member_map_: ClassVar[dict] = ...
|
|
421
|
+
def __format__(self, *args, **kwargs) -> str:
|
|
422
|
+
"""Convert to a string according to format_spec."""
|
|
@@ -0,0 +1,98 @@
|
|
|
1
|
+
# Copyright (c) 2026, NVIDIA CORPORATION & AFFILIATES. ALL RIGHTS RESERVED.
|
|
2
|
+
#
|
|
3
|
+
# SPDX-License-Identifier: Apache-2.0
|
|
4
|
+
#
|
|
5
|
+
# This code was automatically generated across versions from 0.7.2 to 0.8.0, generator version 0.3.1.dev1471+g7122059e9. Do not modify it directly.
|
|
6
|
+
|
|
7
|
+
from libc.stdint cimport intptr_t
|
|
8
|
+
|
|
9
|
+
from .cycusolverMp cimport *
|
|
10
|
+
|
|
11
|
+
|
|
12
|
+
###############################################################################
|
|
13
|
+
# Types
|
|
14
|
+
###############################################################################
|
|
15
|
+
|
|
16
|
+
ctypedef ncclComm_t ncclComm
|
|
17
|
+
ctypedef cusolverMpGrid_t Grid
|
|
18
|
+
ctypedef cusolverMpMatrixDescriptor_t MatrixDescriptor
|
|
19
|
+
ctypedef cusolverMpHandle_t Handle
|
|
20
|
+
ctypedef cusolverMpNewtonSchulzDescriptor_t NewtonSchulzDescriptor
|
|
21
|
+
ctypedef cusolverMpLoggerCallback_t LoggerCallback
|
|
22
|
+
|
|
23
|
+
ctypedef cudaStream_t Stream
|
|
24
|
+
ctypedef cudaDataType DataType
|
|
25
|
+
ctypedef libraryPropertyType_t LibraryPropertyType
|
|
26
|
+
|
|
27
|
+
|
|
28
|
+
###############################################################################
|
|
29
|
+
# Enum
|
|
30
|
+
###############################################################################
|
|
31
|
+
|
|
32
|
+
ctypedef cusolverMpGridMapping_t _GridMapping
|
|
33
|
+
ctypedef cusolverMpNewtonSchulzDescriptorAttribute_t _NewtonSchulzDescriptorAttribute
|
|
34
|
+
|
|
35
|
+
|
|
36
|
+
###############################################################################
|
|
37
|
+
# Functions
|
|
38
|
+
###############################################################################
|
|
39
|
+
|
|
40
|
+
cpdef intptr_t create(int device_id, intptr_t stream) except? 0
|
|
41
|
+
cpdef destroy(intptr_t handle)
|
|
42
|
+
cpdef intptr_t get_stream(intptr_t handle) except? 0
|
|
43
|
+
cpdef int get_version(intptr_t handle) except? 0
|
|
44
|
+
cpdef intptr_t create_device_grid(intptr_t handle, intptr_t comm, int32_t num_row_devices, int32_t num_col_devices, int mapping) except? 0
|
|
45
|
+
cpdef destroy_grid(intptr_t grid)
|
|
46
|
+
cpdef intptr_t create_matrix_desc(intptr_t grid, int data_type, int64_t m_a, int64_t n_a, int64_t mb_a, int64_t nb_a, uint32_t rsrc_a, uint32_t csrc_a, int64_t lld_a) except? 0
|
|
47
|
+
cpdef destroy_matrix_desc(intptr_t desc)
|
|
48
|
+
cpdef int64_t numroc(int64_t n, int64_t nb, uint32_t iproc, uint32_t isrcproc, uint32_t nprocs) except? -1
|
|
49
|
+
cpdef matrix_scatter_h2d(intptr_t handle, int64_t m, int64_t n, intptr_t d_a, int64_t ia, int64_t ja, intptr_t desc_a, int root, intptr_t h_src, int64_t h_ldsrc)
|
|
50
|
+
cpdef matrix_gather_d2h(intptr_t handle, int64_t m, int64_t n, intptr_t d_a, int64_t ia, int64_t ja, intptr_t desc_a, int root, intptr_t h_dst, int64_t h_lddst)
|
|
51
|
+
cpdef tuple getrf_buffer_size(intptr_t handle, int64_t m, int64_t n, intptr_t d_a, int64_t ia, int64_t ja, intptr_t desc_a, intptr_t d_ipiv, int compute_type)
|
|
52
|
+
cpdef getrf(intptr_t handle, int64_t m, int64_t n, intptr_t d_a, int64_t ia, int64_t ja, intptr_t desc_a, intptr_t d_ipiv, int compute_type, intptr_t d_work, size_t workspace_in_bytes_on_device, intptr_t h_work, size_t workspace_in_bytes_on_host, intptr_t info)
|
|
53
|
+
cpdef tuple getrs_buffer_size(intptr_t handle, int trans, int64_t n, int64_t nrhs, intptr_t d_a, int64_t ia, int64_t ja, intptr_t desc_a, intptr_t d_ipiv, intptr_t d_b, int64_t ib, int64_t jb, intptr_t desc_b, int compute_type)
|
|
54
|
+
cpdef getrs(intptr_t handle, int trans, int64_t n, int64_t nrhs, intptr_t d_a, int64_t ia, int64_t ja, intptr_t desc_a, intptr_t d_ipiv, intptr_t d_b, int64_t ib, int64_t jb, intptr_t desc_b, int compute_type, intptr_t d_work, size_t workspace_in_bytes_on_device, intptr_t h_work, size_t workspace_in_bytes_on_host, intptr_t d_info)
|
|
55
|
+
cpdef tuple potrf_buffer_size(intptr_t handle, int uplo, int64_t n, intptr_t a, int64_t ia, int64_t ja, intptr_t desc_a, int compute_type)
|
|
56
|
+
cpdef potrf(intptr_t handle, int uplo, int64_t n, intptr_t a, int64_t ia, int64_t ja, intptr_t desc_a, int compute_type, intptr_t d_work, size_t workspace_in_bytes_on_device, intptr_t h_work, size_t workspace_in_bytes_on_host, intptr_t info)
|
|
57
|
+
cpdef tuple potrs_buffer_size(intptr_t handle, int uplo, int64_t n, int64_t nrhs, intptr_t a, int64_t ia, int64_t ja, intptr_t desc_a, intptr_t b, int64_t ib, int64_t jb, intptr_t desc_b, int compute_type)
|
|
58
|
+
cpdef potrs(intptr_t handle, int uplo, int64_t n, int64_t nrhs, intptr_t a, int64_t ia, int64_t ja, intptr_t desc_a, intptr_t b, int64_t ib, int64_t jb, intptr_t desc_b, int compute_type, intptr_t d_work, size_t workspace_in_bytes_on_device, intptr_t h_work, size_t workspace_in_bytes_on_host, intptr_t info)
|
|
59
|
+
cpdef tuple ormqr_buffer_size(intptr_t handle, int side, int trans, int64_t m, int64_t n, int64_t k, intptr_t a, int64_t ia, int64_t ja, intptr_t desc_a, intptr_t tau, intptr_t c, int64_t ic, int64_t jc, intptr_t desc_c, int compute_type)
|
|
60
|
+
cpdef ormqr(intptr_t handle, int side, int trans, int64_t m, int64_t n, int64_t k, intptr_t a, int64_t ia, int64_t ja, intptr_t desc_a, intptr_t tau, intptr_t c, int64_t ic, int64_t jc, intptr_t desc_c, int compute_type, intptr_t d_work, size_t workspace_in_bytes_on_device, intptr_t h_work, size_t workspace_in_bytes_on_host, intptr_t info)
|
|
61
|
+
cpdef tuple ormtr_buffer_size(intptr_t handle, int side, int uplo, int trans, int64_t m, int64_t n, intptr_t a, int64_t ia, int64_t ja, intptr_t desc_a, intptr_t tau, intptr_t c, int64_t ic, int64_t jc, intptr_t desc_c, int compute_type)
|
|
62
|
+
cpdef ormtr(intptr_t handle, int side, int uplo, int trans, int64_t m, int64_t n, intptr_t a, int64_t ia, int64_t ja, intptr_t desc_a, intptr_t tau, intptr_t c, int64_t ic, int64_t jc, intptr_t desc_c, int compute_type, intptr_t d_work, size_t workspace_in_bytes_on_device, intptr_t h_work, size_t workspace_in_bytes_on_host, intptr_t info)
|
|
63
|
+
cpdef tuple gels_buffer_size(intptr_t handle, int trans, int64_t m, int64_t n, int64_t nrhs, intptr_t a, int64_t ia, int64_t ja, intptr_t desc_a, intptr_t b, int64_t ib, int64_t jb, intptr_t desc_b, int compute_type)
|
|
64
|
+
cpdef gels(intptr_t handle, int trans, int64_t m, int64_t n, int64_t nrhs, intptr_t a, int64_t ia, int64_t ja, intptr_t desc_a, intptr_t b, int64_t ib, int64_t jb, intptr_t desc_b, int compute_type, intptr_t d_work, size_t workspace_in_bytes_on_device, intptr_t h_work, size_t workspace_in_bytes_on_host, intptr_t info)
|
|
65
|
+
cpdef tuple stedc_buffer_size(intptr_t handle, intptr_t compz, int64_t n, intptr_t d_d, intptr_t d_e, intptr_t d_q, int64_t iq, int64_t jq, intptr_t desc_q, int compute_type, intptr_t iwork)
|
|
66
|
+
cpdef stedc(intptr_t handle, intptr_t compz, int64_t n, intptr_t d_d, intptr_t d_e, intptr_t d_q, int64_t iq, int64_t jq, intptr_t desc_q, int compute_type, intptr_t d_work, size_t workspace_in_bytes_on_device, intptr_t h_work, size_t workspace_in_bytes_on_host, intptr_t info)
|
|
67
|
+
cpdef tuple geqrf_buffer_size(intptr_t handle, int64_t m, int64_t n, intptr_t d_a, int64_t ia, int64_t ja, intptr_t desc_a, int compute_type)
|
|
68
|
+
cpdef geqrf(intptr_t handle, int64_t m, int64_t n, intptr_t d_a, int64_t ia, int64_t ja, intptr_t desc_a, intptr_t d_tau, int compute_type, intptr_t d_work, size_t workspace_in_bytes_on_device, intptr_t h_work, size_t workspace_in_bytes_on_host, intptr_t info)
|
|
69
|
+
cpdef tuple sytrd_buffer_size(intptr_t handle, int uplo, int64_t n, intptr_t d_a, int64_t ia, int64_t ja, intptr_t desc_a, intptr_t d_d, intptr_t d_e, intptr_t d_tau, int compute_type)
|
|
70
|
+
cpdef sytrd(intptr_t handle, int uplo, int64_t n, intptr_t d_a, int64_t ia, int64_t ja, intptr_t desc_a, intptr_t d_d, intptr_t d_e, intptr_t d_tau, int compute_type, intptr_t d_work, size_t workspace_in_bytes_on_device, intptr_t h_work, size_t workspace_in_bytes_on_host, intptr_t info)
|
|
71
|
+
cpdef tuple syevd_buffer_size(intptr_t handle, intptr_t compz, int uplo, int64_t n, intptr_t d_a, int64_t ia, int64_t ja, intptr_t desc_a, intptr_t d_d, intptr_t d_q, int64_t iq, int64_t jq, intptr_t desc_q, int compute_type)
|
|
72
|
+
cpdef syevd(intptr_t handle, intptr_t compz, int uplo, int64_t n, intptr_t d_a, int64_t ia, int64_t ja, intptr_t desc_a, intptr_t d_d, intptr_t d_q, int64_t iq, int64_t jq, intptr_t desc_q, int compute_type, intptr_t d_work, size_t workspace_in_bytes_on_device, intptr_t h_work, size_t workspace_in_bytes_on_host, intptr_t d_info)
|
|
73
|
+
cpdef tuple sygst_buffer_size(intptr_t handle, int ibtype, int uplo, int64_t m, int64_t ia, int64_t ja, intptr_t desc_a, int64_t ib, int64_t jb, intptr_t desc_b, int compute_type)
|
|
74
|
+
cpdef sygst(intptr_t handle, int ibtype, int uplo, int64_t m, intptr_t a, int64_t ia, int64_t ja, intptr_t desc_a, intptr_t b, int64_t ib, int64_t jb, intptr_t desc_b, int compute_type, intptr_t d_work, size_t workspace_in_bytes_on_device, intptr_t h_work, size_t workspace_in_bytes_on_host, intptr_t info)
|
|
75
|
+
cpdef tuple sygvd_buffer_size(intptr_t handle, int ibtype, int jobz, int uplo, int64_t m, int64_t ia, int64_t ja, intptr_t desc_a, int64_t ib, int64_t jb, intptr_t desc_b, int64_t iz, int64_t jz, intptr_t desc_z, int compute_type)
|
|
76
|
+
cpdef sygvd(intptr_t handle, int ibtype, int jobz, int uplo, int64_t m, intptr_t a, int64_t ia, int64_t ja, intptr_t desc_a, intptr_t b, int64_t ib, int64_t jb, intptr_t desc_b, intptr_t w, intptr_t z, int64_t iz, int64_t jz, intptr_t desc_z, int compute_type, intptr_t d_work, size_t workspace_in_bytes_on_device, intptr_t h_work, size_t workspace_in_bytes_on_host, intptr_t info)
|
|
77
|
+
cpdef logger_set_file(intptr_t file)
|
|
78
|
+
cpdef logger_open_file(intptr_t log_file)
|
|
79
|
+
cpdef logger_set_level(int level)
|
|
80
|
+
cpdef logger_set_mask(int mask)
|
|
81
|
+
cpdef logger_force_disable()
|
|
82
|
+
cpdef set_math_mode(intptr_t handle, int mode)
|
|
83
|
+
cpdef int get_math_mode(intptr_t handle) except? -1
|
|
84
|
+
cpdef set_emulation_strategy(intptr_t handle, int strategy)
|
|
85
|
+
cpdef int get_emulation_strategy(intptr_t handle) except? -1
|
|
86
|
+
cpdef intptr_t newton_schulz_descriptor_create() except? 0
|
|
87
|
+
cpdef newton_schulz_descriptor_destroy(intptr_t ns_desc)
|
|
88
|
+
cpdef get_newton_schulz_descriptor_attribute_dtype(int attr)
|
|
89
|
+
cpdef newton_schulz_descriptor_set_attribute(intptr_t ns_desc, int attr, intptr_t buf, size_t size_in_bytes)
|
|
90
|
+
cpdef newton_schulz_descriptor_get_attribute(intptr_t ns_desc, int attr, intptr_t buf, size_t size_in_bytes, intptr_t size_in_bytes_written)
|
|
91
|
+
cpdef set_stream(intptr_t handle, intptr_t stream)
|
|
92
|
+
cpdef buffer_register(intptr_t grid, intptr_t ptr, size_t size)
|
|
93
|
+
cpdef buffer_deregister(intptr_t grid, intptr_t ptr)
|
|
94
|
+
cpdef tuple orgqr_buffer_size(intptr_t handle, int64_t m, int64_t n, int64_t k, intptr_t d_a, int64_t ia, int64_t ja, intptr_t desc_a, intptr_t d_tau, int compute_type)
|
|
95
|
+
cpdef orgqr(intptr_t handle, int64_t m, int64_t n, int64_t k, intptr_t d_a, int64_t ia, int64_t ja, intptr_t desc_a, intptr_t d_tau, int compute_type, intptr_t d_work, size_t workspace_in_bytes_on_device, intptr_t h_work, size_t workspace_in_bytes_on_host, intptr_t d_info)
|
|
96
|
+
cpdef laset(intptr_t handle, int uplo, int64_t m, int64_t n, intptr_t alpha, intptr_t beta, intptr_t d_a, int64_t ia, int64_t ja, intptr_t desc_a, intptr_t d_info)
|
|
97
|
+
cpdef tuple newton_schulz_buffer_size(intptr_t handle, intptr_t ns_desc, int64_t m, int64_t n, intptr_t d_x, int64_t ix, int64_t jx, intptr_t desc_x, int64_t number_of_newton_schulz_iterations, intptr_t h_coeffs, int compute_type)
|
|
98
|
+
cpdef newton_schulz(intptr_t handle, intptr_t ns_desc, int64_t m, int64_t n, intptr_t d_x, int64_t ix, int64_t jx, intptr_t desc_x, int64_t number_of_newton_schulz_iterations, intptr_t h_coeffs, int compute_type, intptr_t d_work, size_t workspace_in_bytes_on_device, intptr_t h_work, size_t workspace_in_bytes_on_host, intptr_t info)
|
|
@@ -0,0 +1,114 @@
|
|
|
1
|
+
# Copyright (c) 2026, NVIDIA CORPORATION & AFFILIATES. ALL RIGHTS RESERVED.
|
|
2
|
+
#
|
|
3
|
+
# SPDX-License-Identifier: Apache-2.0
|
|
4
|
+
|
|
5
|
+
import _cython_3_2_5
|
|
6
|
+
import enum
|
|
7
|
+
from typing import Any, Callable, ClassVar
|
|
8
|
+
|
|
9
|
+
__pyx_capi__: dict
|
|
10
|
+
__test__: dict
|
|
11
|
+
buffer_deregister: _cython_3_2_5.cython_function_or_method
|
|
12
|
+
buffer_register: _cython_3_2_5.cython_function_or_method
|
|
13
|
+
check_status: _cython_3_2_5.cython_function_or_method
|
|
14
|
+
create: _cython_3_2_5.cython_function_or_method
|
|
15
|
+
create_device_grid: _cython_3_2_5.cython_function_or_method
|
|
16
|
+
create_matrix_desc: _cython_3_2_5.cython_function_or_method
|
|
17
|
+
destroy: _cython_3_2_5.cython_function_or_method
|
|
18
|
+
destroy_grid: _cython_3_2_5.cython_function_or_method
|
|
19
|
+
destroy_matrix_desc: _cython_3_2_5.cython_function_or_method
|
|
20
|
+
gels: _cython_3_2_5.cython_function_or_method
|
|
21
|
+
gels_buffer_size: _cython_3_2_5.cython_function_or_method
|
|
22
|
+
geqrf: _cython_3_2_5.cython_function_or_method
|
|
23
|
+
geqrf_buffer_size: _cython_3_2_5.cython_function_or_method
|
|
24
|
+
get_emulation_strategy: _cython_3_2_5.cython_function_or_method
|
|
25
|
+
get_math_mode: _cython_3_2_5.cython_function_or_method
|
|
26
|
+
get_newton_schulz_descriptor_attribute_dtype: _cython_3_2_5.cython_function_or_method
|
|
27
|
+
get_stream: _cython_3_2_5.cython_function_or_method
|
|
28
|
+
get_version: _cython_3_2_5.cython_function_or_method
|
|
29
|
+
getrf: _cython_3_2_5.cython_function_or_method
|
|
30
|
+
getrf_buffer_size: _cython_3_2_5.cython_function_or_method
|
|
31
|
+
getrs: _cython_3_2_5.cython_function_or_method
|
|
32
|
+
getrs_buffer_size: _cython_3_2_5.cython_function_or_method
|
|
33
|
+
laset: _cython_3_2_5.cython_function_or_method
|
|
34
|
+
logger_force_disable: _cython_3_2_5.cython_function_or_method
|
|
35
|
+
logger_open_file: _cython_3_2_5.cython_function_or_method
|
|
36
|
+
logger_set_file: _cython_3_2_5.cython_function_or_method
|
|
37
|
+
logger_set_level: _cython_3_2_5.cython_function_or_method
|
|
38
|
+
logger_set_mask: _cython_3_2_5.cython_function_or_method
|
|
39
|
+
matrix_gather_d2h: _cython_3_2_5.cython_function_or_method
|
|
40
|
+
matrix_scatter_h2d: _cython_3_2_5.cython_function_or_method
|
|
41
|
+
newton_schulz: _cython_3_2_5.cython_function_or_method
|
|
42
|
+
newton_schulz_buffer_size: _cython_3_2_5.cython_function_or_method
|
|
43
|
+
newton_schulz_descriptor_create: _cython_3_2_5.cython_function_or_method
|
|
44
|
+
newton_schulz_descriptor_destroy: _cython_3_2_5.cython_function_or_method
|
|
45
|
+
newton_schulz_descriptor_get_attribute: _cython_3_2_5.cython_function_or_method
|
|
46
|
+
newton_schulz_descriptor_set_attribute: _cython_3_2_5.cython_function_or_method
|
|
47
|
+
numroc: _cython_3_2_5.cython_function_or_method
|
|
48
|
+
orgqr: _cython_3_2_5.cython_function_or_method
|
|
49
|
+
orgqr_buffer_size: _cython_3_2_5.cython_function_or_method
|
|
50
|
+
ormqr: _cython_3_2_5.cython_function_or_method
|
|
51
|
+
ormqr_buffer_size: _cython_3_2_5.cython_function_or_method
|
|
52
|
+
ormtr: _cython_3_2_5.cython_function_or_method
|
|
53
|
+
ormtr_buffer_size: _cython_3_2_5.cython_function_or_method
|
|
54
|
+
potrf: _cython_3_2_5.cython_function_or_method
|
|
55
|
+
potrf_buffer_size: _cython_3_2_5.cython_function_or_method
|
|
56
|
+
potrs: _cython_3_2_5.cython_function_or_method
|
|
57
|
+
potrs_buffer_size: _cython_3_2_5.cython_function_or_method
|
|
58
|
+
set_emulation_strategy: _cython_3_2_5.cython_function_or_method
|
|
59
|
+
set_math_mode: _cython_3_2_5.cython_function_or_method
|
|
60
|
+
set_stream: _cython_3_2_5.cython_function_or_method
|
|
61
|
+
stedc: _cython_3_2_5.cython_function_or_method
|
|
62
|
+
stedc_buffer_size: _cython_3_2_5.cython_function_or_method
|
|
63
|
+
syevd: _cython_3_2_5.cython_function_or_method
|
|
64
|
+
syevd_buffer_size: _cython_3_2_5.cython_function_or_method
|
|
65
|
+
sygst: _cython_3_2_5.cython_function_or_method
|
|
66
|
+
sygst_buffer_size: _cython_3_2_5.cython_function_or_method
|
|
67
|
+
sygvd: _cython_3_2_5.cython_function_or_method
|
|
68
|
+
sygvd_buffer_size: _cython_3_2_5.cython_function_or_method
|
|
69
|
+
sytrd: _cython_3_2_5.cython_function_or_method
|
|
70
|
+
sytrd_buffer_size: _cython_3_2_5.cython_function_or_method
|
|
71
|
+
|
|
72
|
+
class GridMapping(enum.IntEnum):
|
|
73
|
+
"""
|
|
74
|
+
See `cusolverMpGridMapping_t`.
|
|
75
|
+
"""
|
|
76
|
+
__new__: ClassVar[Callable] = ...
|
|
77
|
+
COL_MAJOR: ClassVar[GridMapping] = ...
|
|
78
|
+
ROW_MAJOR: ClassVar[GridMapping] = ...
|
|
79
|
+
_generate_next_value_: ClassVar[Callable] = ...
|
|
80
|
+
_hashable_values_: ClassVar[list] = ...
|
|
81
|
+
_member_map_: ClassVar[dict] = ...
|
|
82
|
+
_member_names_: ClassVar[list] = ...
|
|
83
|
+
_member_type_: ClassVar[type[int]] = ...
|
|
84
|
+
_unhashable_values_: ClassVar[list] = ...
|
|
85
|
+
_unhashable_values_map_: ClassVar[dict] = ...
|
|
86
|
+
_use_args_: ClassVar[bool] = ...
|
|
87
|
+
_value2member_map_: ClassVar[dict] = ...
|
|
88
|
+
def __format__(self, *args, **kwargs) -> str:
|
|
89
|
+
"""Convert to a string according to format_spec."""
|
|
90
|
+
|
|
91
|
+
class NewtonSchulzDescriptorAttribute(enum.IntEnum):
|
|
92
|
+
"""
|
|
93
|
+
See `cusolverMpNewtonSchulzDescriptorAttribute_t`.
|
|
94
|
+
"""
|
|
95
|
+
__new__: ClassVar[Callable] = ...
|
|
96
|
+
NORMALIZE: ClassVar[NewtonSchulzDescriptorAttribute] = ...
|
|
97
|
+
REDUCE_VIA_COMPUTE_TYPE: ClassVar[NewtonSchulzDescriptorAttribute] = ...
|
|
98
|
+
_generate_next_value_: ClassVar[Callable] = ...
|
|
99
|
+
_hashable_values_: ClassVar[list] = ...
|
|
100
|
+
_member_map_: ClassVar[dict] = ...
|
|
101
|
+
_member_names_: ClassVar[list] = ...
|
|
102
|
+
_member_type_: ClassVar[type[int]] = ...
|
|
103
|
+
_unhashable_values_: ClassVar[list] = ...
|
|
104
|
+
_unhashable_values_map_: ClassVar[dict] = ...
|
|
105
|
+
_use_args_: ClassVar[bool] = ...
|
|
106
|
+
_value2member_map_: ClassVar[dict] = ...
|
|
107
|
+
def __format__(self, *args, **kwargs) -> str:
|
|
108
|
+
"""Convert to a string according to format_spec."""
|
|
109
|
+
|
|
110
|
+
class cuSOLVERMpError(Exception):
|
|
111
|
+
def __init__(self, status) -> Any:
|
|
112
|
+
"""cuSOLVERMpError.__init__(self, status)"""
|
|
113
|
+
def __reduce__(self) -> Any:
|
|
114
|
+
"""cuSOLVERMpError.__reduce__(self)"""
|
|
Binary file
|