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,1562 @@
|
|
|
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
|
+
check_status: _cython_3_2_5.cython_function_or_method
|
|
12
|
+
commondx_create_code: _cython_3_2_5.cython_function_or_method
|
|
13
|
+
commondx_destroy_code: _cython_3_2_5.cython_function_or_method
|
|
14
|
+
commondx_get_code_ltoir: _cython_3_2_5.cython_function_or_method
|
|
15
|
+
commondx_get_code_ltoir_size: _cython_3_2_5.cython_function_or_method
|
|
16
|
+
commondx_get_code_ltoir_sizes: _cython_3_2_5.cython_function_or_method
|
|
17
|
+
commondx_get_code_ltoirs: _cython_3_2_5.cython_function_or_method
|
|
18
|
+
commondx_get_code_num_ltoirs: _cython_3_2_5.cython_function_or_method
|
|
19
|
+
commondx_get_code_option_int64: _cython_3_2_5.cython_function_or_method
|
|
20
|
+
commondx_get_code_options_int64s: _cython_3_2_5.cython_function_or_method
|
|
21
|
+
commondx_get_code_ptx: _cython_3_2_5.cython_function_or_method
|
|
22
|
+
commondx_get_code_ptx_size: _cython_3_2_5.cython_function_or_method
|
|
23
|
+
commondx_set_code_option_int64: _cython_3_2_5.cython_function_or_method
|
|
24
|
+
commondx_set_code_option_int64s: _cython_3_2_5.cython_function_or_method
|
|
25
|
+
commondx_set_code_option_str: _cython_3_2_5.cython_function_or_method
|
|
26
|
+
commondx_status_to_str: _cython_3_2_5.cython_function_or_method
|
|
27
|
+
cublasdx_create_descriptor: _cython_3_2_5.cython_function_or_method
|
|
28
|
+
cublasdx_create_device_function: _cython_3_2_5.cython_function_or_method
|
|
29
|
+
cublasdx_create_device_function_with_pipelines: _cython_3_2_5.cython_function_or_method
|
|
30
|
+
cublasdx_create_device_pipeline: _cython_3_2_5.cython_function_or_method
|
|
31
|
+
cublasdx_create_tensor: _cython_3_2_5.cython_function_or_method
|
|
32
|
+
cublasdx_create_tensor_strided: _cython_3_2_5.cython_function_or_method
|
|
33
|
+
cublasdx_create_tile_pipeline: _cython_3_2_5.cython_function_or_method
|
|
34
|
+
cublasdx_destroy_descriptor: _cython_3_2_5.cython_function_or_method
|
|
35
|
+
cublasdx_destroy_device_function: _cython_3_2_5.cython_function_or_method
|
|
36
|
+
cublasdx_destroy_pipeline: _cython_3_2_5.cython_function_or_method
|
|
37
|
+
cublasdx_destroy_tensor: _cython_3_2_5.cython_function_or_method
|
|
38
|
+
cublasdx_finalize: _cython_3_2_5.cython_function_or_method
|
|
39
|
+
cublasdx_finalize_code: _cython_3_2_5.cython_function_or_method
|
|
40
|
+
cublasdx_finalize_device_functions: _cython_3_2_5.cython_function_or_method
|
|
41
|
+
cublasdx_finalize_pipelines: _cython_3_2_5.cython_function_or_method
|
|
42
|
+
cublasdx_finalize_tensors: _cython_3_2_5.cython_function_or_method
|
|
43
|
+
cublasdx_get_device_function_trait_str: _cython_3_2_5.cython_function_or_method
|
|
44
|
+
cublasdx_get_device_function_trait_str_size: _cython_3_2_5.cython_function_or_method
|
|
45
|
+
cublasdx_get_ltoir: _cython_3_2_5.cython_function_or_method
|
|
46
|
+
cublasdx_get_ltoir_size: _cython_3_2_5.cython_function_or_method
|
|
47
|
+
cublasdx_get_pipeline_trait_int64: _cython_3_2_5.cython_function_or_method
|
|
48
|
+
cublasdx_get_pipeline_trait_int64s: _cython_3_2_5.cython_function_or_method
|
|
49
|
+
cublasdx_get_pipeline_trait_str: _cython_3_2_5.cython_function_or_method
|
|
50
|
+
cublasdx_get_pipeline_trait_str_size: _cython_3_2_5.cython_function_or_method
|
|
51
|
+
cublasdx_get_tensor_trait_int64: _cython_3_2_5.cython_function_or_method
|
|
52
|
+
cublasdx_get_tensor_trait_str: _cython_3_2_5.cython_function_or_method
|
|
53
|
+
cublasdx_get_tensor_trait_str_size: _cython_3_2_5.cython_function_or_method
|
|
54
|
+
cublasdx_get_trait_int64: _cython_3_2_5.cython_function_or_method
|
|
55
|
+
cublasdx_get_trait_int64s: _cython_3_2_5.cython_function_or_method
|
|
56
|
+
cublasdx_get_trait_str: _cython_3_2_5.cython_function_or_method
|
|
57
|
+
cublasdx_get_trait_str_size: _cython_3_2_5.cython_function_or_method
|
|
58
|
+
cublasdx_make_tensor_like: _cython_3_2_5.cython_function_or_method
|
|
59
|
+
cublasdx_operator_type_to_str: _cython_3_2_5.cython_function_or_method
|
|
60
|
+
cublasdx_set_operator_int64: _cython_3_2_5.cython_function_or_method
|
|
61
|
+
cublasdx_set_operator_int64s: _cython_3_2_5.cython_function_or_method
|
|
62
|
+
cublasdx_set_option_str: _cython_3_2_5.cython_function_or_method
|
|
63
|
+
cublasdx_set_tensor_option_int64: _cython_3_2_5.cython_function_or_method
|
|
64
|
+
cublasdx_trait_type_to_str: _cython_3_2_5.cython_function_or_method
|
|
65
|
+
cufftdx_create_descriptor: _cython_3_2_5.cython_function_or_method
|
|
66
|
+
cufftdx_destroy_descriptor: _cython_3_2_5.cython_function_or_method
|
|
67
|
+
cufftdx_finalize_code: _cython_3_2_5.cython_function_or_method
|
|
68
|
+
cufftdx_get_knob_int64s: _cython_3_2_5.cython_function_or_method
|
|
69
|
+
cufftdx_get_knob_int64size: _cython_3_2_5.cython_function_or_method
|
|
70
|
+
cufftdx_get_ltoir: _cython_3_2_5.cython_function_or_method
|
|
71
|
+
cufftdx_get_ltoir_size: _cython_3_2_5.cython_function_or_method
|
|
72
|
+
cufftdx_get_trait_commondx_data_type: _cython_3_2_5.cython_function_or_method
|
|
73
|
+
cufftdx_get_trait_int64: _cython_3_2_5.cython_function_or_method
|
|
74
|
+
cufftdx_get_trait_int64s: _cython_3_2_5.cython_function_or_method
|
|
75
|
+
cufftdx_get_trait_str: _cython_3_2_5.cython_function_or_method
|
|
76
|
+
cufftdx_get_trait_str_size: _cython_3_2_5.cython_function_or_method
|
|
77
|
+
cufftdx_operator_type_to_str: _cython_3_2_5.cython_function_or_method
|
|
78
|
+
cufftdx_set_operator_int64: _cython_3_2_5.cython_function_or_method
|
|
79
|
+
cufftdx_set_operator_int64s: _cython_3_2_5.cython_function_or_method
|
|
80
|
+
cufftdx_set_option_str: _cython_3_2_5.cython_function_or_method
|
|
81
|
+
cufftdx_trait_type_to_str: _cython_3_2_5.cython_function_or_method
|
|
82
|
+
curanddx_create_descriptor: _cython_3_2_5.cython_function_or_method
|
|
83
|
+
curanddx_destroy_descriptor: _cython_3_2_5.cython_function_or_method
|
|
84
|
+
curanddx_distribution_to_str: _cython_3_2_5.cython_function_or_method
|
|
85
|
+
curanddx_finalize_code: _cython_3_2_5.cython_function_or_method
|
|
86
|
+
curanddx_generate_method_to_str: _cython_3_2_5.cython_function_or_method
|
|
87
|
+
curanddx_generator_to_str: _cython_3_2_5.cython_function_or_method
|
|
88
|
+
curanddx_get_trait_int64: _cython_3_2_5.cython_function_or_method
|
|
89
|
+
curanddx_get_trait_str: _cython_3_2_5.cython_function_or_method
|
|
90
|
+
curanddx_get_trait_str_size: _cython_3_2_5.cython_function_or_method
|
|
91
|
+
curanddx_get_version: _cython_3_2_5.cython_function_or_method
|
|
92
|
+
curanddx_normal_method_to_str: _cython_3_2_5.cython_function_or_method
|
|
93
|
+
curanddx_operator_type_to_str: _cython_3_2_5.cython_function_or_method
|
|
94
|
+
curanddx_set_operator_doubles: _cython_3_2_5.cython_function_or_method
|
|
95
|
+
curanddx_set_operator_int64: _cython_3_2_5.cython_function_or_method
|
|
96
|
+
curanddx_set_option_str: _cython_3_2_5.cython_function_or_method
|
|
97
|
+
curanddx_trait_type_to_str: _cython_3_2_5.cython_function_or_method
|
|
98
|
+
cusolverdx_create_descriptor: _cython_3_2_5.cython_function_or_method
|
|
99
|
+
cusolverdx_destroy_descriptor: _cython_3_2_5.cython_function_or_method
|
|
100
|
+
cusolverdx_finalize_code: _cython_3_2_5.cython_function_or_method
|
|
101
|
+
cusolverdx_get_ltoir: _cython_3_2_5.cython_function_or_method
|
|
102
|
+
cusolverdx_get_ltoir_size: _cython_3_2_5.cython_function_or_method
|
|
103
|
+
cusolverdx_get_trait_commondx_data_types: _cython_3_2_5.cython_function_or_method
|
|
104
|
+
cusolverdx_get_trait_int64: _cython_3_2_5.cython_function_or_method
|
|
105
|
+
cusolverdx_get_trait_int64s: _cython_3_2_5.cython_function_or_method
|
|
106
|
+
cusolverdx_get_trait_str: _cython_3_2_5.cython_function_or_method
|
|
107
|
+
cusolverdx_get_trait_str_size: _cython_3_2_5.cython_function_or_method
|
|
108
|
+
cusolverdx_get_universal_fatbin: _cython_3_2_5.cython_function_or_method
|
|
109
|
+
cusolverdx_get_universal_fatbin_size: _cython_3_2_5.cython_function_or_method
|
|
110
|
+
cusolverdx_operator_type_to_str: _cython_3_2_5.cython_function_or_method
|
|
111
|
+
cusolverdx_set_operator_int64: _cython_3_2_5.cython_function_or_method
|
|
112
|
+
cusolverdx_set_operator_int64s: _cython_3_2_5.cython_function_or_method
|
|
113
|
+
cusolverdx_set_option_str: _cython_3_2_5.cython_function_or_method
|
|
114
|
+
cusolverdx_trait_type_to_str: _cython_3_2_5.cython_function_or_method
|
|
115
|
+
get_version: _cython_3_2_5.cython_function_or_method
|
|
116
|
+
get_version_ex: _cython_3_2_5.cython_function_or_method
|
|
117
|
+
|
|
118
|
+
class CommondxArchModifier(enum.IntEnum):
|
|
119
|
+
"""
|
|
120
|
+
The set of compute feature to target. See COMMONDX_OPTION_TARGET_SM
|
|
121
|
+
.See also https://docs.nvidia.com/cuda/cuda-c-programming-
|
|
122
|
+
guide/index.html#feature-set-compiler-targets
|
|
123
|
+
|
|
124
|
+
See `commondxArchModifier`.
|
|
125
|
+
"""
|
|
126
|
+
__new__: ClassVar[Callable] = ...
|
|
127
|
+
ARCH_SPECIFIC: ClassVar[CommondxArchModifier] = ...
|
|
128
|
+
FAMILY_SPECIFIC: ClassVar[CommondxArchModifier] = ...
|
|
129
|
+
GENERIC: ClassVar[CommondxArchModifier] = ...
|
|
130
|
+
_generate_next_value_: ClassVar[Callable] = ...
|
|
131
|
+
_hashable_values_: ClassVar[list] = ...
|
|
132
|
+
_member_map_: ClassVar[dict] = ...
|
|
133
|
+
_member_names_: ClassVar[list] = ...
|
|
134
|
+
_member_type_: ClassVar[type[int]] = ...
|
|
135
|
+
_unhashable_values_: ClassVar[list] = ...
|
|
136
|
+
_unhashable_values_map_: ClassVar[dict] = ...
|
|
137
|
+
_use_args_: ClassVar[bool] = ...
|
|
138
|
+
_value2member_map_: ClassVar[dict] = ...
|
|
139
|
+
def __format__(self, *args, **kwargs) -> str:
|
|
140
|
+
"""Convert to a string according to format_spec."""
|
|
141
|
+
|
|
142
|
+
class CommondxCodeContainer(enum.IntEnum):
|
|
143
|
+
"""
|
|
144
|
+
The set of device code container types supported by the library. See
|
|
145
|
+
commondxOption_t::COMMONDX_OPTION_CODE_CONTAINER.
|
|
146
|
+
|
|
147
|
+
See `commondxCodeContainer`.
|
|
148
|
+
"""
|
|
149
|
+
__new__: ClassVar[Callable] = ...
|
|
150
|
+
FATBIN: ClassVar[CommondxCodeContainer] = ...
|
|
151
|
+
LTOIR: ClassVar[CommondxCodeContainer] = ...
|
|
152
|
+
PTX: ClassVar[CommondxCodeContainer] = ...
|
|
153
|
+
_generate_next_value_: ClassVar[Callable] = ...
|
|
154
|
+
_hashable_values_: ClassVar[list] = ...
|
|
155
|
+
_member_map_: ClassVar[dict] = ...
|
|
156
|
+
_member_names_: ClassVar[list] = ...
|
|
157
|
+
_member_type_: ClassVar[type[int]] = ...
|
|
158
|
+
_unhashable_values_: ClassVar[list] = ...
|
|
159
|
+
_unhashable_values_map_: ClassVar[dict] = ...
|
|
160
|
+
_use_args_: ClassVar[bool] = ...
|
|
161
|
+
_value2member_map_: ClassVar[dict] = ...
|
|
162
|
+
def __format__(self, *args, **kwargs) -> str:
|
|
163
|
+
"""Convert to a string according to format_spec."""
|
|
164
|
+
|
|
165
|
+
class CommondxCodeType(enum.IntEnum):
|
|
166
|
+
"""
|
|
167
|
+
The set of code types supported by the library. See
|
|
168
|
+
commondxOption_t::COMMONDX_OPTION_TARGET_CODE.
|
|
169
|
+
|
|
170
|
+
See `commondxCodeType`.
|
|
171
|
+
"""
|
|
172
|
+
__new__: ClassVar[Callable] = ...
|
|
173
|
+
LTO: ClassVar[CommondxCodeType] = ...
|
|
174
|
+
PTX: ClassVar[CommondxCodeType] = ...
|
|
175
|
+
_generate_next_value_: ClassVar[Callable] = ...
|
|
176
|
+
_hashable_values_: ClassVar[list] = ...
|
|
177
|
+
_member_map_: ClassVar[dict] = ...
|
|
178
|
+
_member_names_: ClassVar[list] = ...
|
|
179
|
+
_member_type_: ClassVar[type[int]] = ...
|
|
180
|
+
_unhashable_values_: ClassVar[list] = ...
|
|
181
|
+
_unhashable_values_map_: ClassVar[dict] = ...
|
|
182
|
+
_use_args_: ClassVar[bool] = ...
|
|
183
|
+
_value2member_map_: ClassVar[dict] = ...
|
|
184
|
+
def __format__(self, *args, **kwargs) -> str:
|
|
185
|
+
"""Convert to a string according to format_spec."""
|
|
186
|
+
|
|
187
|
+
class CommondxExecution(enum.IntEnum):
|
|
188
|
+
"""
|
|
189
|
+
The set of execution modes supported by the library.
|
|
190
|
+
|
|
191
|
+
See `commondxExecution`.
|
|
192
|
+
"""
|
|
193
|
+
__new__: ClassVar[Callable] = ...
|
|
194
|
+
BLOCK: ClassVar[CommondxExecution] = ...
|
|
195
|
+
THREAD: ClassVar[CommondxExecution] = ...
|
|
196
|
+
_generate_next_value_: ClassVar[Callable] = ...
|
|
197
|
+
_hashable_values_: ClassVar[list] = ...
|
|
198
|
+
_member_map_: ClassVar[dict] = ...
|
|
199
|
+
_member_names_: ClassVar[list] = ...
|
|
200
|
+
_member_type_: ClassVar[type[int]] = ...
|
|
201
|
+
_unhashable_values_: ClassVar[list] = ...
|
|
202
|
+
_unhashable_values_map_: ClassVar[dict] = ...
|
|
203
|
+
_use_args_: ClassVar[bool] = ...
|
|
204
|
+
_value2member_map_: ClassVar[dict] = ...
|
|
205
|
+
def __format__(self, *args, **kwargs) -> str:
|
|
206
|
+
"""Convert to a string according to format_spec."""
|
|
207
|
+
|
|
208
|
+
class CommondxOption(enum.IntEnum):
|
|
209
|
+
"""
|
|
210
|
+
Options to tweak code generation.
|
|
211
|
+
|
|
212
|
+
See `commondxOption`.
|
|
213
|
+
"""
|
|
214
|
+
__new__: ClassVar[Callable] = ...
|
|
215
|
+
CODE_CONTAINER: ClassVar[CommondxOption] = ...
|
|
216
|
+
CODE_ISA: ClassVar[CommondxOption] = ...
|
|
217
|
+
EXTRA_NVRTC_ARGS: ClassVar[CommondxOption] = ...
|
|
218
|
+
SYMBOL_NAME: ClassVar[CommondxOption] = ...
|
|
219
|
+
TARGET_CODE: ClassVar[CommondxOption] = ...
|
|
220
|
+
TARGET_SM: ClassVar[CommondxOption] = ...
|
|
221
|
+
_generate_next_value_: ClassVar[Callable] = ...
|
|
222
|
+
_hashable_values_: ClassVar[list] = ...
|
|
223
|
+
_member_map_: ClassVar[dict] = ...
|
|
224
|
+
_member_names_: ClassVar[list] = ...
|
|
225
|
+
_member_type_: ClassVar[type[int]] = ...
|
|
226
|
+
_unhashable_values_: ClassVar[list] = ...
|
|
227
|
+
_unhashable_values_map_: ClassVar[dict] = ...
|
|
228
|
+
_use_args_: ClassVar[bool] = ...
|
|
229
|
+
_value2member_map_: ClassVar[dict] = ...
|
|
230
|
+
def __format__(self, *args, **kwargs) -> str:
|
|
231
|
+
"""Convert to a string according to format_spec."""
|
|
232
|
+
|
|
233
|
+
class CommondxPrecision(enum.IntEnum):
|
|
234
|
+
"""
|
|
235
|
+
The set of precisions supported by the library.For the actual input and
|
|
236
|
+
output value types, see commondxValueType_t
|
|
237
|
+
|
|
238
|
+
See `commondxPrecision`.
|
|
239
|
+
"""
|
|
240
|
+
__new__: ClassVar[Callable] = ...
|
|
241
|
+
BF16: ClassVar[CommondxPrecision] = ...
|
|
242
|
+
F16: ClassVar[CommondxPrecision] = ...
|
|
243
|
+
F32: ClassVar[CommondxPrecision] = ...
|
|
244
|
+
F64: ClassVar[CommondxPrecision] = ...
|
|
245
|
+
F8_E4M3: ClassVar[CommondxPrecision] = ...
|
|
246
|
+
F8_E5M2: ClassVar[CommondxPrecision] = ...
|
|
247
|
+
I16: ClassVar[CommondxPrecision] = ...
|
|
248
|
+
I32: ClassVar[CommondxPrecision] = ...
|
|
249
|
+
I64: ClassVar[CommondxPrecision] = ...
|
|
250
|
+
I8: ClassVar[CommondxPrecision] = ...
|
|
251
|
+
TF32: ClassVar[CommondxPrecision] = ...
|
|
252
|
+
UI16: ClassVar[CommondxPrecision] = ...
|
|
253
|
+
UI32: ClassVar[CommondxPrecision] = ...
|
|
254
|
+
UI64: ClassVar[CommondxPrecision] = ...
|
|
255
|
+
UI8: ClassVar[CommondxPrecision] = ...
|
|
256
|
+
_generate_next_value_: ClassVar[Callable] = ...
|
|
257
|
+
_hashable_values_: ClassVar[list] = ...
|
|
258
|
+
_member_map_: ClassVar[dict] = ...
|
|
259
|
+
_member_names_: ClassVar[list] = ...
|
|
260
|
+
_member_type_: ClassVar[type[int]] = ...
|
|
261
|
+
_unhashable_values_: ClassVar[list] = ...
|
|
262
|
+
_unhashable_values_map_: ClassVar[dict] = ...
|
|
263
|
+
_use_args_: ClassVar[bool] = ...
|
|
264
|
+
_value2member_map_: ClassVar[dict] = ...
|
|
265
|
+
def __format__(self, *args, **kwargs) -> str:
|
|
266
|
+
"""Convert to a string according to format_spec."""
|
|
267
|
+
|
|
268
|
+
class CommondxStatusType(enum.IntEnum):
|
|
269
|
+
"""
|
|
270
|
+
The set of status values that can be returned by APIs defined in the
|
|
271
|
+
library.
|
|
272
|
+
|
|
273
|
+
See `commondxStatusType`.
|
|
274
|
+
"""
|
|
275
|
+
__new__: ClassVar[Callable] = ...
|
|
276
|
+
COMPILATION_ERROR: ClassVar[CommondxStatusType] = ...
|
|
277
|
+
CUFFT_ERROR: ClassVar[CommondxStatusType] = ...
|
|
278
|
+
INTERNAL_ERROR: ClassVar[CommondxStatusType] = ...
|
|
279
|
+
INVALID_VALUE: ClassVar[CommondxStatusType] = ...
|
|
280
|
+
SUCCESS: ClassVar[CommondxStatusType] = ...
|
|
281
|
+
_generate_next_value_: ClassVar[Callable] = ...
|
|
282
|
+
_hashable_values_: ClassVar[list] = ...
|
|
283
|
+
_member_map_: ClassVar[dict] = ...
|
|
284
|
+
_member_names_: ClassVar[list] = ...
|
|
285
|
+
_member_type_: ClassVar[type[int]] = ...
|
|
286
|
+
_unhashable_values_: ClassVar[list] = ...
|
|
287
|
+
_unhashable_values_map_: ClassVar[dict] = ...
|
|
288
|
+
_use_args_: ClassVar[bool] = ...
|
|
289
|
+
_value2member_map_: ClassVar[dict] = ...
|
|
290
|
+
def __format__(self, *args, **kwargs) -> str:
|
|
291
|
+
"""Convert to a string according to format_spec."""
|
|
292
|
+
|
|
293
|
+
class CommondxValueType(enum.IntEnum):
|
|
294
|
+
"""
|
|
295
|
+
The set of value types supported by the library.Note that, for complex
|
|
296
|
+
numbers, this combines real and imaginary part, and that all complex
|
|
297
|
+
numbers are overaligned on their size.
|
|
298
|
+
|
|
299
|
+
See `commondxValueType`.
|
|
300
|
+
"""
|
|
301
|
+
__new__: ClassVar[Callable] = ...
|
|
302
|
+
C_16BF: ClassVar[CommondxValueType] = ...
|
|
303
|
+
C_16F: ClassVar[CommondxValueType] = ...
|
|
304
|
+
C_16F2: ClassVar[CommondxValueType] = ...
|
|
305
|
+
C_16I: ClassVar[CommondxValueType] = ...
|
|
306
|
+
C_16UI: ClassVar[CommondxValueType] = ...
|
|
307
|
+
C_32F: ClassVar[CommondxValueType] = ...
|
|
308
|
+
C_32I: ClassVar[CommondxValueType] = ...
|
|
309
|
+
C_32TF: ClassVar[CommondxValueType] = ...
|
|
310
|
+
C_32UI: ClassVar[CommondxValueType] = ...
|
|
311
|
+
C_64F: ClassVar[CommondxValueType] = ...
|
|
312
|
+
C_64I: ClassVar[CommondxValueType] = ...
|
|
313
|
+
C_64UI: ClassVar[CommondxValueType] = ...
|
|
314
|
+
C_8F_E4M3: ClassVar[CommondxValueType] = ...
|
|
315
|
+
C_8F_E5M2: ClassVar[CommondxValueType] = ...
|
|
316
|
+
C_8I: ClassVar[CommondxValueType] = ...
|
|
317
|
+
C_8UI: ClassVar[CommondxValueType] = ...
|
|
318
|
+
R_16BF: ClassVar[CommondxValueType] = ...
|
|
319
|
+
R_16F: ClassVar[CommondxValueType] = ...
|
|
320
|
+
R_16F2: ClassVar[CommondxValueType] = ...
|
|
321
|
+
R_16I: ClassVar[CommondxValueType] = ...
|
|
322
|
+
R_16UI: ClassVar[CommondxValueType] = ...
|
|
323
|
+
R_32F: ClassVar[CommondxValueType] = ...
|
|
324
|
+
R_32I: ClassVar[CommondxValueType] = ...
|
|
325
|
+
R_32TF: ClassVar[CommondxValueType] = ...
|
|
326
|
+
R_32UI: ClassVar[CommondxValueType] = ...
|
|
327
|
+
R_64F: ClassVar[CommondxValueType] = ...
|
|
328
|
+
R_64I: ClassVar[CommondxValueType] = ...
|
|
329
|
+
R_64UI: ClassVar[CommondxValueType] = ...
|
|
330
|
+
R_8F_E4M3: ClassVar[CommondxValueType] = ...
|
|
331
|
+
R_8F_E5M2: ClassVar[CommondxValueType] = ...
|
|
332
|
+
R_8I: ClassVar[CommondxValueType] = ...
|
|
333
|
+
R_8UI: ClassVar[CommondxValueType] = ...
|
|
334
|
+
_generate_next_value_: ClassVar[Callable] = ...
|
|
335
|
+
_hashable_values_: ClassVar[list] = ...
|
|
336
|
+
_member_map_: ClassVar[dict] = ...
|
|
337
|
+
_member_names_: ClassVar[list] = ...
|
|
338
|
+
_member_type_: ClassVar[type[int]] = ...
|
|
339
|
+
_unhashable_values_: ClassVar[list] = ...
|
|
340
|
+
_unhashable_values_map_: ClassVar[dict] = ...
|
|
341
|
+
_use_args_: ClassVar[bool] = ...
|
|
342
|
+
_value2member_map_: ClassVar[dict] = ...
|
|
343
|
+
def __format__(self, *args, **kwargs) -> str:
|
|
344
|
+
"""Convert to a string according to format_spec."""
|
|
345
|
+
|
|
346
|
+
class CublasdxApi(enum.IntEnum):
|
|
347
|
+
"""
|
|
348
|
+
Type of cublasDx API.Handling problems with default or custom/dynamic
|
|
349
|
+
leading dimensions. Check cublasdx::LeadingDimension operator
|
|
350
|
+
documentation for more details (https://docs.nvidia.com/cuda/cublasdx/a
|
|
351
|
+
pi/operators.html#leadingdimension-operator)
|
|
352
|
+
|
|
353
|
+
See `cublasdxApi`.
|
|
354
|
+
"""
|
|
355
|
+
__new__: ClassVar[Callable] = ...
|
|
356
|
+
SMEM: ClassVar[CublasdxApi] = ...
|
|
357
|
+
SMEM_DYNAMIC_LD: ClassVar[CublasdxApi] = ...
|
|
358
|
+
TENSORS: ClassVar[CublasdxApi] = ...
|
|
359
|
+
_generate_next_value_: ClassVar[Callable] = ...
|
|
360
|
+
_hashable_values_: ClassVar[list] = ...
|
|
361
|
+
_member_map_: ClassVar[dict] = ...
|
|
362
|
+
_member_names_: ClassVar[list] = ...
|
|
363
|
+
_member_type_: ClassVar[type[int]] = ...
|
|
364
|
+
_unhashable_values_: ClassVar[list] = ...
|
|
365
|
+
_unhashable_values_map_: ClassVar[dict] = ...
|
|
366
|
+
_use_args_: ClassVar[bool] = ...
|
|
367
|
+
_value2member_map_: ClassVar[dict] = ...
|
|
368
|
+
def __format__(self, *args, **kwargs) -> str:
|
|
369
|
+
"""Convert to a string according to format_spec."""
|
|
370
|
+
|
|
371
|
+
class CublasdxArrangement(enum.IntEnum):
|
|
372
|
+
"""
|
|
373
|
+
Data arrangement mode.Defines data arrangements in tensors' taking part
|
|
374
|
+
in the calculation. Check cubladx::TransposeMode operator
|
|
375
|
+
documentation for more details
|
|
376
|
+
(https://docs.nvidia.com/cuda/cublasdx/api/operators.html#arrangement-
|
|
377
|
+
operator)
|
|
378
|
+
|
|
379
|
+
See `cublasdxArrangement`.
|
|
380
|
+
"""
|
|
381
|
+
__new__: ClassVar[Callable] = ...
|
|
382
|
+
COL_MAJOR: ClassVar[CublasdxArrangement] = ...
|
|
383
|
+
ROW_MAJOR: ClassVar[CublasdxArrangement] = ...
|
|
384
|
+
_generate_next_value_: ClassVar[Callable] = ...
|
|
385
|
+
_hashable_values_: ClassVar[list] = ...
|
|
386
|
+
_member_map_: ClassVar[dict] = ...
|
|
387
|
+
_member_names_: ClassVar[list] = ...
|
|
388
|
+
_member_type_: ClassVar[type[int]] = ...
|
|
389
|
+
_unhashable_values_: ClassVar[list] = ...
|
|
390
|
+
_unhashable_values_map_: ClassVar[dict] = ...
|
|
391
|
+
_use_args_: ClassVar[bool] = ...
|
|
392
|
+
_value2member_map_: ClassVar[dict] = ...
|
|
393
|
+
def __format__(self, *args, **kwargs) -> str:
|
|
394
|
+
"""Convert to a string according to format_spec."""
|
|
395
|
+
|
|
396
|
+
class CublasdxBlockSizeStrategy(enum.IntEnum):
|
|
397
|
+
"""
|
|
398
|
+
Type of block size strategy used in device pipeline and tile pipeline
|
|
399
|
+
creation. This enum affects register usage in pipelining GEMM kernels
|
|
400
|
+
and should be considered for different architectures.
|
|
401
|
+
|
|
402
|
+
See `cublasdxBlockSizeStrategy`.
|
|
403
|
+
"""
|
|
404
|
+
__new__: ClassVar[Callable] = ...
|
|
405
|
+
FIXED: ClassVar[CublasdxBlockSizeStrategy] = ...
|
|
406
|
+
HEURISTIC: ClassVar[CublasdxBlockSizeStrategy] = ...
|
|
407
|
+
_generate_next_value_: ClassVar[Callable] = ...
|
|
408
|
+
_hashable_values_: ClassVar[list] = ...
|
|
409
|
+
_member_map_: ClassVar[dict] = ...
|
|
410
|
+
_member_names_: ClassVar[list] = ...
|
|
411
|
+
_member_type_: ClassVar[type[int]] = ...
|
|
412
|
+
_unhashable_values_: ClassVar[list] = ...
|
|
413
|
+
_unhashable_values_map_: ClassVar[dict] = ...
|
|
414
|
+
_use_args_: ClassVar[bool] = ...
|
|
415
|
+
_value2member_map_: ClassVar[dict] = ...
|
|
416
|
+
def __format__(self, *args, **kwargs) -> str:
|
|
417
|
+
"""Convert to a string according to format_spec."""
|
|
418
|
+
|
|
419
|
+
class CublasdxDeviceFunctionOption(enum.IntEnum):
|
|
420
|
+
"""
|
|
421
|
+
Device function options.
|
|
422
|
+
|
|
423
|
+
See `cublasdxDeviceFunctionOption`.
|
|
424
|
+
"""
|
|
425
|
+
__new__: ClassVar[Callable] = ...
|
|
426
|
+
CALLBACK: ClassVar[CublasdxDeviceFunctionOption] = ...
|
|
427
|
+
COPY_ALIGNMENT: ClassVar[CublasdxDeviceFunctionOption] = ...
|
|
428
|
+
NUM_THREADS: ClassVar[CublasdxDeviceFunctionOption] = ...
|
|
429
|
+
SYMBOL_NAME: ClassVar[CublasdxDeviceFunctionOption] = ...
|
|
430
|
+
_generate_next_value_: ClassVar[Callable] = ...
|
|
431
|
+
_hashable_values_: ClassVar[list] = ...
|
|
432
|
+
_member_map_: ClassVar[dict] = ...
|
|
433
|
+
_member_names_: ClassVar[list] = ...
|
|
434
|
+
_member_type_: ClassVar[type[int]] = ...
|
|
435
|
+
_unhashable_values_: ClassVar[list] = ...
|
|
436
|
+
_unhashable_values_map_: ClassVar[dict] = ...
|
|
437
|
+
_use_args_: ClassVar[bool] = ...
|
|
438
|
+
_value2member_map_: ClassVar[dict] = ...
|
|
439
|
+
def __format__(self, *args, **kwargs) -> str:
|
|
440
|
+
"""Convert to a string according to format_spec."""
|
|
441
|
+
|
|
442
|
+
class CublasdxDeviceFunctionTrait(enum.IntEnum):
|
|
443
|
+
"""
|
|
444
|
+
Device function traits, used to query informations.
|
|
445
|
+
|
|
446
|
+
See `cublasdxDeviceFunctionTrait`.
|
|
447
|
+
"""
|
|
448
|
+
__new__: ClassVar[Callable] = ...
|
|
449
|
+
SYMBOL: ClassVar[CublasdxDeviceFunctionTrait] = ...
|
|
450
|
+
_generate_next_value_: ClassVar[Callable] = ...
|
|
451
|
+
_hashable_values_: ClassVar[list] = ...
|
|
452
|
+
_member_map_: ClassVar[dict] = ...
|
|
453
|
+
_member_names_: ClassVar[list] = ...
|
|
454
|
+
_member_type_: ClassVar[type[int]] = ...
|
|
455
|
+
_unhashable_values_: ClassVar[list] = ...
|
|
456
|
+
_unhashable_values_map_: ClassVar[dict] = ...
|
|
457
|
+
_use_args_: ClassVar[bool] = ...
|
|
458
|
+
_value2member_map_: ClassVar[dict] = ...
|
|
459
|
+
def __format__(self, *args, **kwargs) -> str:
|
|
460
|
+
"""Convert to a string according to format_spec."""
|
|
461
|
+
|
|
462
|
+
class CublasdxDeviceFunctionType(enum.IntEnum):
|
|
463
|
+
"""
|
|
464
|
+
Device functions supported by the library.
|
|
465
|
+
|
|
466
|
+
See `cublasdxDeviceFunctionType`.
|
|
467
|
+
"""
|
|
468
|
+
__new__: ClassVar[Callable] = ...
|
|
469
|
+
AXPBY: ClassVar[CublasdxDeviceFunctionType] = ...
|
|
470
|
+
CLEAR: ClassVar[CublasdxDeviceFunctionType] = ...
|
|
471
|
+
COPY: ClassVar[CublasdxDeviceFunctionType] = ...
|
|
472
|
+
COPY_WAIT: ClassVar[CublasdxDeviceFunctionType] = ...
|
|
473
|
+
CREATE: ClassVar[CublasdxDeviceFunctionType] = ...
|
|
474
|
+
DESTROY: ClassVar[CublasdxDeviceFunctionType] = ...
|
|
475
|
+
EPILOGUE: ClassVar[CublasdxDeviceFunctionType] = ...
|
|
476
|
+
EXECUTE: ClassVar[CublasdxDeviceFunctionType] = ...
|
|
477
|
+
IS_INDEX_IN_BOUNDS: ClassVar[CublasdxDeviceFunctionType] = ...
|
|
478
|
+
IS_PREDICATED: ClassVar[CublasdxDeviceFunctionType] = ...
|
|
479
|
+
IS_THREAD_ACTIVE: ClassVar[CublasdxDeviceFunctionType] = ...
|
|
480
|
+
MAP_CRD2IDX: ClassVar[CublasdxDeviceFunctionType] = ...
|
|
481
|
+
MAP_IDX2CRD: ClassVar[CublasdxDeviceFunctionType] = ...
|
|
482
|
+
MAP_IDX2CRD_PARTITIONER: ClassVar[CublasdxDeviceFunctionType] = ...
|
|
483
|
+
RESET: ClassVar[CublasdxDeviceFunctionType] = ...
|
|
484
|
+
_generate_next_value_: ClassVar[Callable] = ...
|
|
485
|
+
_hashable_values_: ClassVar[list] = ...
|
|
486
|
+
_member_map_: ClassVar[dict] = ...
|
|
487
|
+
_member_names_: ClassVar[list] = ...
|
|
488
|
+
_member_type_: ClassVar[type[int]] = ...
|
|
489
|
+
_unhashable_values_: ClassVar[list] = ...
|
|
490
|
+
_unhashable_values_map_: ClassVar[dict] = ...
|
|
491
|
+
_use_args_: ClassVar[bool] = ...
|
|
492
|
+
_value2member_map_: ClassVar[dict] = ...
|
|
493
|
+
def __format__(self, *args, **kwargs) -> str:
|
|
494
|
+
"""Convert to a string according to format_spec."""
|
|
495
|
+
|
|
496
|
+
class CublasdxDevicePipelineType(enum.IntEnum):
|
|
497
|
+
"""
|
|
498
|
+
Type of device pipeline.
|
|
499
|
+
|
|
500
|
+
See `cublasdxDevicePipelineType`.
|
|
501
|
+
"""
|
|
502
|
+
__new__: ClassVar[Callable] = ...
|
|
503
|
+
SUGGESTED: ClassVar[CublasdxDevicePipelineType] = ...
|
|
504
|
+
_generate_next_value_: ClassVar[Callable] = ...
|
|
505
|
+
_hashable_values_: ClassVar[list] = ...
|
|
506
|
+
_member_map_: ClassVar[dict] = ...
|
|
507
|
+
_member_names_: ClassVar[list] = ...
|
|
508
|
+
_member_type_: ClassVar[type[int]] = ...
|
|
509
|
+
_unhashable_values_: ClassVar[list] = ...
|
|
510
|
+
_unhashable_values_map_: ClassVar[dict] = ...
|
|
511
|
+
_use_args_: ClassVar[bool] = ...
|
|
512
|
+
_value2member_map_: ClassVar[dict] = ...
|
|
513
|
+
def __format__(self, *args, **kwargs) -> str:
|
|
514
|
+
"""Convert to a string according to format_spec."""
|
|
515
|
+
|
|
516
|
+
class CublasdxFunction(enum.IntEnum):
|
|
517
|
+
"""
|
|
518
|
+
BLAS function.Sets the BLAS function that will be executed. Check
|
|
519
|
+
cubladx::Function operator documentation for more details
|
|
520
|
+
(https://docs.nvidia.com/cuda/cublasdx/api/operators.html#function-
|
|
521
|
+
operator)
|
|
522
|
+
|
|
523
|
+
See `cublasdxFunction`.
|
|
524
|
+
"""
|
|
525
|
+
__new__: ClassVar[Callable] = ...
|
|
526
|
+
MM: ClassVar[CublasdxFunction] = ...
|
|
527
|
+
_generate_next_value_: ClassVar[Callable] = ...
|
|
528
|
+
_hashable_values_: ClassVar[list] = ...
|
|
529
|
+
_member_map_: ClassVar[dict] = ...
|
|
530
|
+
_member_names_: ClassVar[list] = ...
|
|
531
|
+
_member_type_: ClassVar[type[int]] = ...
|
|
532
|
+
_unhashable_values_: ClassVar[list] = ...
|
|
533
|
+
_unhashable_values_map_: ClassVar[dict] = ...
|
|
534
|
+
_use_args_: ClassVar[bool] = ...
|
|
535
|
+
_value2member_map_: ClassVar[dict] = ...
|
|
536
|
+
def __format__(self, *args, **kwargs) -> str:
|
|
537
|
+
"""Convert to a string according to format_spec."""
|
|
538
|
+
|
|
539
|
+
class CublasdxMemorySpace(enum.IntEnum):
|
|
540
|
+
"""
|
|
541
|
+
Memory space.
|
|
542
|
+
|
|
543
|
+
See `cublasdxMemorySpace`.
|
|
544
|
+
"""
|
|
545
|
+
__new__: ClassVar[Callable] = ...
|
|
546
|
+
ANY: ClassVar[CublasdxMemorySpace] = ...
|
|
547
|
+
GMEM: ClassVar[CublasdxMemorySpace] = ...
|
|
548
|
+
RMEM: ClassVar[CublasdxMemorySpace] = ...
|
|
549
|
+
SMEM: ClassVar[CublasdxMemorySpace] = ...
|
|
550
|
+
_generate_next_value_: ClassVar[Callable] = ...
|
|
551
|
+
_hashable_values_: ClassVar[list] = ...
|
|
552
|
+
_member_map_: ClassVar[dict] = ...
|
|
553
|
+
_member_names_: ClassVar[list] = ...
|
|
554
|
+
_member_type_: ClassVar[type[int]] = ...
|
|
555
|
+
_unhashable_values_: ClassVar[list] = ...
|
|
556
|
+
_unhashable_values_map_: ClassVar[dict] = ...
|
|
557
|
+
_use_args_: ClassVar[bool] = ...
|
|
558
|
+
_value2member_map_: ClassVar[dict] = ...
|
|
559
|
+
def __format__(self, *args, **kwargs) -> str:
|
|
560
|
+
"""Convert to a string according to format_spec."""
|
|
561
|
+
|
|
562
|
+
class CublasdxOperatorType(enum.IntEnum):
|
|
563
|
+
"""
|
|
564
|
+
cublasDx operatorsThe set of supported cublasDx operators. Check
|
|
565
|
+
cublaDx description operator documentation for more details
|
|
566
|
+
(https://docs.nvidia.com/cuda/cublasdx/api/operators.html#function-
|
|
567
|
+
operator) Check cublasDx execution operator documentation for more
|
|
568
|
+
details
|
|
569
|
+
(https://docs.nvidia.com/cuda/cublasdx/api/operators.html#execution-
|
|
570
|
+
operators)
|
|
571
|
+
|
|
572
|
+
See `cublasdxOperatorType`.
|
|
573
|
+
"""
|
|
574
|
+
__new__: ClassVar[Callable] = ...
|
|
575
|
+
ALIGNMENT: ClassVar[CublasdxOperatorType] = ...
|
|
576
|
+
API: ClassVar[CublasdxOperatorType] = ...
|
|
577
|
+
ARRANGEMENT: ClassVar[CublasdxOperatorType] = ...
|
|
578
|
+
BLOCK_DIM: ClassVar[CublasdxOperatorType] = ...
|
|
579
|
+
ENABLE_INPUT_STREAMING: ClassVar[CublasdxOperatorType] = ...
|
|
580
|
+
EXECUTION: ClassVar[CublasdxOperatorType] = ...
|
|
581
|
+
FUNCTION: ClassVar[CublasdxOperatorType] = ...
|
|
582
|
+
LEADING_DIMENSION: ClassVar[CublasdxOperatorType] = ...
|
|
583
|
+
PRECISION: ClassVar[CublasdxOperatorType] = ...
|
|
584
|
+
SIZE: ClassVar[CublasdxOperatorType] = ...
|
|
585
|
+
SM: ClassVar[CublasdxOperatorType] = ...
|
|
586
|
+
STATIC_BLOCK_DIM: ClassVar[CublasdxOperatorType] = ...
|
|
587
|
+
TRANSPOSE_MODE: ClassVar[CublasdxOperatorType] = ...
|
|
588
|
+
TYPE: ClassVar[CublasdxOperatorType] = ...
|
|
589
|
+
WITH_PIPELINE: ClassVar[CublasdxOperatorType] = ...
|
|
590
|
+
_generate_next_value_: ClassVar[Callable] = ...
|
|
591
|
+
_hashable_values_: ClassVar[list] = ...
|
|
592
|
+
_member_map_: ClassVar[dict] = ...
|
|
593
|
+
_member_names_: ClassVar[list] = ...
|
|
594
|
+
_member_type_: ClassVar[type[int]] = ...
|
|
595
|
+
_unhashable_values_: ClassVar[list] = ...
|
|
596
|
+
_unhashable_values_map_: ClassVar[dict] = ...
|
|
597
|
+
_use_args_: ClassVar[bool] = ...
|
|
598
|
+
_value2member_map_: ClassVar[dict] = ...
|
|
599
|
+
def __format__(self, *args, **kwargs) -> str:
|
|
600
|
+
"""Convert to a string according to format_spec."""
|
|
601
|
+
|
|
602
|
+
class CublasdxPipelineTrait(enum.IntEnum):
|
|
603
|
+
"""
|
|
604
|
+
Pipeline traits, used to query informations.
|
|
605
|
+
|
|
606
|
+
See `cublasdxPipelineTrait`.
|
|
607
|
+
"""
|
|
608
|
+
__new__: ClassVar[Callable] = ...
|
|
609
|
+
BLOCK_DIM: ClassVar[CublasdxPipelineTrait] = ...
|
|
610
|
+
BUFFER_ALIGNMENT_BYTES: ClassVar[CublasdxPipelineTrait] = ...
|
|
611
|
+
BUFFER_SIZE: ClassVar[CublasdxPipelineTrait] = ...
|
|
612
|
+
OPAQUE_NAME: ClassVar[CublasdxPipelineTrait] = ...
|
|
613
|
+
STORAGE_ALIGNMENT_BYTES: ClassVar[CublasdxPipelineTrait] = ...
|
|
614
|
+
STORAGE_BYTES: ClassVar[CublasdxPipelineTrait] = ...
|
|
615
|
+
_generate_next_value_: ClassVar[Callable] = ...
|
|
616
|
+
_hashable_values_: ClassVar[list] = ...
|
|
617
|
+
_member_map_: ClassVar[dict] = ...
|
|
618
|
+
_member_names_: ClassVar[list] = ...
|
|
619
|
+
_member_type_: ClassVar[type[int]] = ...
|
|
620
|
+
_unhashable_values_: ClassVar[list] = ...
|
|
621
|
+
_unhashable_values_map_: ClassVar[dict] = ...
|
|
622
|
+
_use_args_: ClassVar[bool] = ...
|
|
623
|
+
_value2member_map_: ClassVar[dict] = ...
|
|
624
|
+
def __format__(self, *args, **kwargs) -> str:
|
|
625
|
+
"""Convert to a string according to format_spec."""
|
|
626
|
+
|
|
627
|
+
class CublasdxTensorOption(enum.IntEnum):
|
|
628
|
+
"""
|
|
629
|
+
Tensor options.
|
|
630
|
+
|
|
631
|
+
See `cublasdxTensorOption`.
|
|
632
|
+
"""
|
|
633
|
+
__new__: ClassVar[Callable] = ...
|
|
634
|
+
ALIGNMENT_BYTES: ClassVar[CublasdxTensorOption] = ...
|
|
635
|
+
_generate_next_value_: ClassVar[Callable] = ...
|
|
636
|
+
_hashable_values_: ClassVar[list] = ...
|
|
637
|
+
_member_map_: ClassVar[dict] = ...
|
|
638
|
+
_member_names_: ClassVar[list] = ...
|
|
639
|
+
_member_type_: ClassVar[type[int]] = ...
|
|
640
|
+
_unhashable_values_: ClassVar[list] = ...
|
|
641
|
+
_unhashable_values_map_: ClassVar[dict] = ...
|
|
642
|
+
_use_args_: ClassVar[bool] = ...
|
|
643
|
+
_value2member_map_: ClassVar[dict] = ...
|
|
644
|
+
def __format__(self, *args, **kwargs) -> str:
|
|
645
|
+
"""Convert to a string according to format_spec."""
|
|
646
|
+
|
|
647
|
+
class CublasdxTensorTrait(enum.IntEnum):
|
|
648
|
+
"""
|
|
649
|
+
Tensor traits, used to query informations.
|
|
650
|
+
|
|
651
|
+
See `cublasdxTensorTrait`.
|
|
652
|
+
"""
|
|
653
|
+
__new__: ClassVar[Callable] = ...
|
|
654
|
+
ALIGNMENT_BYTES: ClassVar[CublasdxTensorTrait] = ...
|
|
655
|
+
LOGICAL_SIZE: ClassVar[CublasdxTensorTrait] = ...
|
|
656
|
+
MEMORY_SPACE: ClassVar[CublasdxTensorTrait] = ...
|
|
657
|
+
OPAQUE_NAME: ClassVar[CublasdxTensorTrait] = ...
|
|
658
|
+
STORAGE_BYTES: ClassVar[CublasdxTensorTrait] = ...
|
|
659
|
+
UID: ClassVar[CublasdxTensorTrait] = ...
|
|
660
|
+
_generate_next_value_: ClassVar[Callable] = ...
|
|
661
|
+
_hashable_values_: ClassVar[list] = ...
|
|
662
|
+
_member_map_: ClassVar[dict] = ...
|
|
663
|
+
_member_names_: ClassVar[list] = ...
|
|
664
|
+
_member_type_: ClassVar[type[int]] = ...
|
|
665
|
+
_unhashable_values_: ClassVar[list] = ...
|
|
666
|
+
_unhashable_values_map_: ClassVar[dict] = ...
|
|
667
|
+
_use_args_: ClassVar[bool] = ...
|
|
668
|
+
_value2member_map_: ClassVar[dict] = ...
|
|
669
|
+
def __format__(self, *args, **kwargs) -> str:
|
|
670
|
+
"""Convert to a string according to format_spec."""
|
|
671
|
+
|
|
672
|
+
class CublasdxTensorType(enum.IntEnum):
|
|
673
|
+
"""
|
|
674
|
+
cuBLASDx desired tensor typeTensor types are opaque (layout is
|
|
675
|
+
unspecified), non-owning, and defined by - Memory space (global,
|
|
676
|
+
shared or register memory) - Size & alignment (in bytes) Tensor's
|
|
677
|
+
representation in-memory and in-device depends on their memory space.
|
|
678
|
+
Shared & register tensors are defined as **View CUDA Toolkit
|
|
679
|
+
Documentation for a C++ code example** where `ptr` points to the
|
|
680
|
+
associated data. Global memory tensors have an associated runtime
|
|
681
|
+
leading dimension (64b signed integer), and their representation is
|
|
682
|
+
**View CUDA Toolkit Documentation for a C++ code example** where `ptr`
|
|
683
|
+
points to the associated data and `strides[0]` is the leading
|
|
684
|
+
dimension. In either case, `ptr` must point to some storage (with
|
|
685
|
+
appropriate size and alignment, see below) and is not owning. The user
|
|
686
|
+
is expected to keep memory allocated beyond any use of the tensor.
|
|
687
|
+
`strides[1]` should be a signed, 64bit integer (`long long`) equal to
|
|
688
|
+
the leading dimension of the global memory tensor. The leading
|
|
689
|
+
dimension is the number of `elements` between two successive rows or
|
|
690
|
+
columns (not bytes), depending on the context. All tensor APIs take
|
|
691
|
+
their argument by value (not by pointer) and expect the struct to be
|
|
692
|
+
passed as-is on the stack. Each opaque tensor type is uniquely
|
|
693
|
+
identified by a unique ID and name, see cublasdxTensorTrait_t .
|
|
694
|
+
|
|
695
|
+
See `cublasdxTensorType`.
|
|
696
|
+
"""
|
|
697
|
+
__new__: ClassVar[Callable] = ...
|
|
698
|
+
ACCUMULATOR_C: ClassVar[CublasdxTensorType] = ...
|
|
699
|
+
GMEM_A: ClassVar[CublasdxTensorType] = ...
|
|
700
|
+
GMEM_B: ClassVar[CublasdxTensorType] = ...
|
|
701
|
+
GMEM_C: ClassVar[CublasdxTensorType] = ...
|
|
702
|
+
RMEM_C: ClassVar[CublasdxTensorType] = ...
|
|
703
|
+
SMEM_A: ClassVar[CublasdxTensorType] = ...
|
|
704
|
+
SMEM_B: ClassVar[CublasdxTensorType] = ...
|
|
705
|
+
SMEM_C: ClassVar[CublasdxTensorType] = ...
|
|
706
|
+
SUGGESTED_ACCUMULATOR_C: ClassVar[CublasdxTensorType] = ...
|
|
707
|
+
SUGGESTED_RMEM_C: ClassVar[CublasdxTensorType] = ...
|
|
708
|
+
SUGGESTED_SMEM_A: ClassVar[CublasdxTensorType] = ...
|
|
709
|
+
SUGGESTED_SMEM_B: ClassVar[CublasdxTensorType] = ...
|
|
710
|
+
SUGGESTED_SMEM_C: ClassVar[CublasdxTensorType] = ...
|
|
711
|
+
_generate_next_value_: ClassVar[Callable] = ...
|
|
712
|
+
_hashable_values_: ClassVar[list] = ...
|
|
713
|
+
_member_map_: ClassVar[dict] = ...
|
|
714
|
+
_member_names_: ClassVar[list] = ...
|
|
715
|
+
_member_type_: ClassVar[type[int]] = ...
|
|
716
|
+
_unhashable_values_: ClassVar[list] = ...
|
|
717
|
+
_unhashable_values_map_: ClassVar[dict] = ...
|
|
718
|
+
_use_args_: ClassVar[bool] = ...
|
|
719
|
+
_value2member_map_: ClassVar[dict] = ...
|
|
720
|
+
def __format__(self, *args, **kwargs) -> str:
|
|
721
|
+
"""Convert to a string according to format_spec."""
|
|
722
|
+
|
|
723
|
+
class CublasdxTilePipelineType(enum.IntEnum):
|
|
724
|
+
"""
|
|
725
|
+
Type of tile pipeline.
|
|
726
|
+
|
|
727
|
+
See `cublasdxTilePipelineType`.
|
|
728
|
+
"""
|
|
729
|
+
__new__: ClassVar[Callable] = ...
|
|
730
|
+
PIPELINE_DEFAULT: ClassVar[CublasdxTilePipelineType] = ...
|
|
731
|
+
_generate_next_value_: ClassVar[Callable] = ...
|
|
732
|
+
_hashable_values_: ClassVar[list] = ...
|
|
733
|
+
_member_map_: ClassVar[dict] = ...
|
|
734
|
+
_member_names_: ClassVar[list] = ...
|
|
735
|
+
_member_type_: ClassVar[type[int]] = ...
|
|
736
|
+
_unhashable_values_: ClassVar[list] = ...
|
|
737
|
+
_unhashable_values_map_: ClassVar[dict] = ...
|
|
738
|
+
_use_args_: ClassVar[bool] = ...
|
|
739
|
+
_value2member_map_: ClassVar[dict] = ...
|
|
740
|
+
def __format__(self, *args, **kwargs) -> str:
|
|
741
|
+
"""Convert to a string according to format_spec."""
|
|
742
|
+
|
|
743
|
+
class CublasdxTraitType(enum.IntEnum):
|
|
744
|
+
"""
|
|
745
|
+
cublasDx traitsThe set of supported types of traits that can be
|
|
746
|
+
accessed from finalized sources that use cublasDx. Check cublasDx
|
|
747
|
+
Execution Block Traits documentation for more details
|
|
748
|
+
(https://docs.nvidia.com/cuda/cublasdx/api/traits.html#block-traits)
|
|
749
|
+
|
|
750
|
+
See `cublasdxTraitType`.
|
|
751
|
+
"""
|
|
752
|
+
__new__: ClassVar[Callable] = ...
|
|
753
|
+
ALIGNMENT: ClassVar[CublasdxTraitType] = ...
|
|
754
|
+
ARRANGEMENT: ClassVar[CublasdxTraitType] = ...
|
|
755
|
+
BLOCK_DIM: ClassVar[CublasdxTraitType] = ...
|
|
756
|
+
BLOCK_SIZE: ClassVar[CublasdxTraitType] = ...
|
|
757
|
+
LEADING_DIMENSION: ClassVar[CublasdxTraitType] = ...
|
|
758
|
+
MAX_THREADS_PER_BLOCK: ClassVar[CublasdxTraitType] = ...
|
|
759
|
+
SIZE: ClassVar[CublasdxTraitType] = ...
|
|
760
|
+
SUGGESTED_BLOCK_DIM: ClassVar[CublasdxTraitType] = ...
|
|
761
|
+
SUGGESTED_LEADING_DIMENSION: ClassVar[CublasdxTraitType] = ...
|
|
762
|
+
SYMBOL_NAME: ClassVar[CublasdxTraitType] = ...
|
|
763
|
+
VALUE_TYPE: ClassVar[CublasdxTraitType] = ...
|
|
764
|
+
_generate_next_value_: ClassVar[Callable] = ...
|
|
765
|
+
_hashable_values_: ClassVar[list] = ...
|
|
766
|
+
_member_map_: ClassVar[dict] = ...
|
|
767
|
+
_member_names_: ClassVar[list] = ...
|
|
768
|
+
_member_type_: ClassVar[type[int]] = ...
|
|
769
|
+
_unhashable_values_: ClassVar[list] = ...
|
|
770
|
+
_unhashable_values_map_: ClassVar[dict] = ...
|
|
771
|
+
_use_args_: ClassVar[bool] = ...
|
|
772
|
+
_value2member_map_: ClassVar[dict] = ...
|
|
773
|
+
def __format__(self, *args, **kwargs) -> str:
|
|
774
|
+
"""Convert to a string according to format_spec."""
|
|
775
|
+
|
|
776
|
+
class CublasdxTransposeMode(enum.IntEnum):
|
|
777
|
+
"""
|
|
778
|
+
Tensor transpose mode.The transpose mode depends on
|
|
779
|
+
cubladx::TransposeMode operator which is deprecated since cublasDx
|
|
780
|
+
0.2.0 and might be removed in future versions of mathDx libraries
|
|
781
|
+
Check cubladx::TransposeMode operator documentation for more details (h
|
|
782
|
+
ttps://docs.nvidia.com/cuda/cublasdx/api/operators.html#transposemode-
|
|
783
|
+
operator)
|
|
784
|
+
|
|
785
|
+
See `cublasdxTransposeMode`.
|
|
786
|
+
"""
|
|
787
|
+
__new__: ClassVar[Callable] = ...
|
|
788
|
+
CONJ_TRANSPOSED: ClassVar[CublasdxTransposeMode] = ...
|
|
789
|
+
NON_TRANSPOSED: ClassVar[CublasdxTransposeMode] = ...
|
|
790
|
+
TRANSPOSED: ClassVar[CublasdxTransposeMode] = ...
|
|
791
|
+
_generate_next_value_: ClassVar[Callable] = ...
|
|
792
|
+
_hashable_values_: ClassVar[list] = ...
|
|
793
|
+
_member_map_: ClassVar[dict] = ...
|
|
794
|
+
_member_names_: ClassVar[list] = ...
|
|
795
|
+
_member_type_: ClassVar[type[int]] = ...
|
|
796
|
+
_unhashable_values_: ClassVar[list] = ...
|
|
797
|
+
_unhashable_values_map_: ClassVar[dict] = ...
|
|
798
|
+
_use_args_: ClassVar[bool] = ...
|
|
799
|
+
_value2member_map_: ClassVar[dict] = ...
|
|
800
|
+
def __format__(self, *args, **kwargs) -> str:
|
|
801
|
+
"""Convert to a string according to format_spec."""
|
|
802
|
+
|
|
803
|
+
class CublasdxType(enum.IntEnum):
|
|
804
|
+
"""
|
|
805
|
+
Type of computation data.Check cubladx::Type operator documentation for
|
|
806
|
+
more details
|
|
807
|
+
(https://docs.nvidia.com/cuda/cublasdx/api/operators.html#type-
|
|
808
|
+
operator)
|
|
809
|
+
|
|
810
|
+
See `cublasdxType`.
|
|
811
|
+
"""
|
|
812
|
+
__new__: ClassVar[Callable] = ...
|
|
813
|
+
COMPLEX: ClassVar[CublasdxType] = ...
|
|
814
|
+
REAL: ClassVar[CublasdxType] = ...
|
|
815
|
+
_generate_next_value_: ClassVar[Callable] = ...
|
|
816
|
+
_hashable_values_: ClassVar[list] = ...
|
|
817
|
+
_member_map_: ClassVar[dict] = ...
|
|
818
|
+
_member_names_: ClassVar[list] = ...
|
|
819
|
+
_member_type_: ClassVar[type[int]] = ...
|
|
820
|
+
_unhashable_values_: ClassVar[list] = ...
|
|
821
|
+
_unhashable_values_map_: ClassVar[dict] = ...
|
|
822
|
+
_use_args_: ClassVar[bool] = ...
|
|
823
|
+
_value2member_map_: ClassVar[dict] = ...
|
|
824
|
+
def __format__(self, *args, **kwargs) -> str:
|
|
825
|
+
"""Convert to a string according to format_spec."""
|
|
826
|
+
|
|
827
|
+
class CufftdxApi(enum.IntEnum):
|
|
828
|
+
"""
|
|
829
|
+
Type of cufftDx API.Handling problems with input in register or in
|
|
830
|
+
shared memory buffers. Check cufftdx::execute method documentation for
|
|
831
|
+
more details
|
|
832
|
+
(https://docs.nvidia.com/cuda/cufftdx/api/methods.html#block-execute-
|
|
833
|
+
method)
|
|
834
|
+
|
|
835
|
+
See `cufftdxApi`.
|
|
836
|
+
"""
|
|
837
|
+
__new__: ClassVar[Callable] = ...
|
|
838
|
+
LMEM: ClassVar[CufftdxApi] = ...
|
|
839
|
+
SMEM: ClassVar[CufftdxApi] = ...
|
|
840
|
+
_generate_next_value_: ClassVar[Callable] = ...
|
|
841
|
+
_hashable_values_: ClassVar[list] = ...
|
|
842
|
+
_member_map_: ClassVar[dict] = ...
|
|
843
|
+
_member_names_: ClassVar[list] = ...
|
|
844
|
+
_member_type_: ClassVar[type[int]] = ...
|
|
845
|
+
_unhashable_values_: ClassVar[list] = ...
|
|
846
|
+
_unhashable_values_map_: ClassVar[dict] = ...
|
|
847
|
+
_use_args_: ClassVar[bool] = ...
|
|
848
|
+
_value2member_map_: ClassVar[dict] = ...
|
|
849
|
+
def __format__(self, *args, **kwargs) -> str:
|
|
850
|
+
"""Convert to a string according to format_spec."""
|
|
851
|
+
|
|
852
|
+
class CufftdxCodeType(enum.IntEnum):
|
|
853
|
+
"""
|
|
854
|
+
Code type.Check cufftdx code type documentation for more details
|
|
855
|
+
(https://docs.nvidia.com/cuda/1.4.0-ea/cufftdx/api/methods.html#code-
|
|
856
|
+
type)
|
|
857
|
+
|
|
858
|
+
See `cufftdxCodeType`.
|
|
859
|
+
"""
|
|
860
|
+
__new__: ClassVar[Callable] = ...
|
|
861
|
+
LTOIR: ClassVar[CufftdxCodeType] = ...
|
|
862
|
+
PTX: ClassVar[CufftdxCodeType] = ...
|
|
863
|
+
_generate_next_value_: ClassVar[Callable] = ...
|
|
864
|
+
_hashable_values_: ClassVar[list] = ...
|
|
865
|
+
_member_map_: ClassVar[dict] = ...
|
|
866
|
+
_member_names_: ClassVar[list] = ...
|
|
867
|
+
_member_type_: ClassVar[type[int]] = ...
|
|
868
|
+
_unhashable_values_: ClassVar[list] = ...
|
|
869
|
+
_unhashable_values_map_: ClassVar[dict] = ...
|
|
870
|
+
_use_args_: ClassVar[bool] = ...
|
|
871
|
+
_value2member_map_: ClassVar[dict] = ...
|
|
872
|
+
def __format__(self, *args, **kwargs) -> str:
|
|
873
|
+
"""Convert to a string according to format_spec."""
|
|
874
|
+
|
|
875
|
+
class CufftdxComplexLayout(enum.IntEnum):
|
|
876
|
+
"""
|
|
877
|
+
Complex data layout.Check cufftdx layout documentation for more details
|
|
878
|
+
(https://docs.nvidia.com/cuda/cufftdx/api/methods.html#complex-element-
|
|
879
|
+
layouts)
|
|
880
|
+
|
|
881
|
+
See `cufftdxComplexLayout`.
|
|
882
|
+
"""
|
|
883
|
+
__new__: ClassVar[Callable] = ...
|
|
884
|
+
FULL: ClassVar[CufftdxComplexLayout] = ...
|
|
885
|
+
NATURAL: ClassVar[CufftdxComplexLayout] = ...
|
|
886
|
+
PACKED: ClassVar[CufftdxComplexLayout] = ...
|
|
887
|
+
_generate_next_value_: ClassVar[Callable] = ...
|
|
888
|
+
_hashable_values_: ClassVar[list] = ...
|
|
889
|
+
_member_map_: ClassVar[dict] = ...
|
|
890
|
+
_member_names_: ClassVar[list] = ...
|
|
891
|
+
_member_type_: ClassVar[type[int]] = ...
|
|
892
|
+
_unhashable_values_: ClassVar[list] = ...
|
|
893
|
+
_unhashable_values_map_: ClassVar[dict] = ...
|
|
894
|
+
_use_args_: ClassVar[bool] = ...
|
|
895
|
+
_value2member_map_: ClassVar[dict] = ...
|
|
896
|
+
def __format__(self, *args, **kwargs) -> str:
|
|
897
|
+
"""Convert to a string according to format_spec."""
|
|
898
|
+
|
|
899
|
+
class CufftdxDirection(enum.IntEnum):
|
|
900
|
+
"""
|
|
901
|
+
FFT direction.Check cufftdx::Direction operator documentation for more
|
|
902
|
+
details
|
|
903
|
+
(https://docs.nvidia.com/cuda/cufftdx/api/operators.html#direction-
|
|
904
|
+
operator)
|
|
905
|
+
|
|
906
|
+
See `cufftdxDirection`.
|
|
907
|
+
"""
|
|
908
|
+
__new__: ClassVar[Callable] = ...
|
|
909
|
+
FORWARD: ClassVar[CufftdxDirection] = ...
|
|
910
|
+
INVERSE: ClassVar[CufftdxDirection] = ...
|
|
911
|
+
_generate_next_value_: ClassVar[Callable] = ...
|
|
912
|
+
_hashable_values_: ClassVar[list] = ...
|
|
913
|
+
_member_map_: ClassVar[dict] = ...
|
|
914
|
+
_member_names_: ClassVar[list] = ...
|
|
915
|
+
_member_type_: ClassVar[type[int]] = ...
|
|
916
|
+
_unhashable_values_: ClassVar[list] = ...
|
|
917
|
+
_unhashable_values_map_: ClassVar[dict] = ...
|
|
918
|
+
_use_args_: ClassVar[bool] = ...
|
|
919
|
+
_value2member_map_: ClassVar[dict] = ...
|
|
920
|
+
def __format__(self, *args, **kwargs) -> str:
|
|
921
|
+
"""Convert to a string according to format_spec."""
|
|
922
|
+
|
|
923
|
+
class CufftdxKnobType(enum.IntEnum):
|
|
924
|
+
"""
|
|
925
|
+
cufftDx configuration knobsThe set of supported knobs used for
|
|
926
|
+
accessing cufftDx operator's performance configuration details. Check
|
|
927
|
+
cufftDx Execution Traits documentation for more details
|
|
928
|
+
(https://docs.nvidia.com/cuda/cufftdx/api/traits.html#execution-traits)
|
|
929
|
+
|
|
930
|
+
See `cufftdxKnobType`.
|
|
931
|
+
"""
|
|
932
|
+
__new__: ClassVar[Callable] = ...
|
|
933
|
+
ELEMENTS_PER_THREAD: ClassVar[CufftdxKnobType] = ...
|
|
934
|
+
FFTS_PER_BLOCK: ClassVar[CufftdxKnobType] = ...
|
|
935
|
+
_generate_next_value_: ClassVar[Callable] = ...
|
|
936
|
+
_hashable_values_: ClassVar[list] = ...
|
|
937
|
+
_member_map_: ClassVar[dict] = ...
|
|
938
|
+
_member_names_: ClassVar[list] = ...
|
|
939
|
+
_member_type_: ClassVar[type[int]] = ...
|
|
940
|
+
_unhashable_values_: ClassVar[list] = ...
|
|
941
|
+
_unhashable_values_map_: ClassVar[dict] = ...
|
|
942
|
+
_use_args_: ClassVar[bool] = ...
|
|
943
|
+
_value2member_map_: ClassVar[dict] = ...
|
|
944
|
+
def __format__(self, *args, **kwargs) -> str:
|
|
945
|
+
"""Convert to a string according to format_spec."""
|
|
946
|
+
|
|
947
|
+
class CufftdxOperatorType(enum.IntEnum):
|
|
948
|
+
"""
|
|
949
|
+
cufftDx operatorsThe set of supported cufftDx operators. Check cufftDx
|
|
950
|
+
description operators documentation for more details
|
|
951
|
+
(https://docs.nvidia.com/cuda/cufftdx/api/operators.html#description-
|
|
952
|
+
operators) Check cufftDx execution operators documentation for more
|
|
953
|
+
details
|
|
954
|
+
(https://docs.nvidia.com/cuda/cufftdx/api/operators.html#execution-
|
|
955
|
+
operators)
|
|
956
|
+
|
|
957
|
+
See `cufftdxOperatorType`.
|
|
958
|
+
"""
|
|
959
|
+
__new__: ClassVar[Callable] = ...
|
|
960
|
+
API: ClassVar[CufftdxOperatorType] = ...
|
|
961
|
+
BLOCK_DIM: ClassVar[CufftdxOperatorType] = ...
|
|
962
|
+
CODE_TYPE: ClassVar[CufftdxOperatorType] = ...
|
|
963
|
+
DIRECTION: ClassVar[CufftdxOperatorType] = ...
|
|
964
|
+
ELEMENTS_PER_THREAD: ClassVar[CufftdxOperatorType] = ...
|
|
965
|
+
EXECUTION: ClassVar[CufftdxOperatorType] = ...
|
|
966
|
+
FFTS_PER_BLOCK: ClassVar[CufftdxOperatorType] = ...
|
|
967
|
+
PRECISION: ClassVar[CufftdxOperatorType] = ...
|
|
968
|
+
REAL_FFT_OPTIONS: ClassVar[CufftdxOperatorType] = ...
|
|
969
|
+
SIZE: ClassVar[CufftdxOperatorType] = ...
|
|
970
|
+
SM: ClassVar[CufftdxOperatorType] = ...
|
|
971
|
+
TYPE: ClassVar[CufftdxOperatorType] = ...
|
|
972
|
+
_generate_next_value_: ClassVar[Callable] = ...
|
|
973
|
+
_hashable_values_: ClassVar[list] = ...
|
|
974
|
+
_member_map_: ClassVar[dict] = ...
|
|
975
|
+
_member_names_: ClassVar[list] = ...
|
|
976
|
+
_member_type_: ClassVar[type[int]] = ...
|
|
977
|
+
_unhashable_values_: ClassVar[list] = ...
|
|
978
|
+
_unhashable_values_map_: ClassVar[dict] = ...
|
|
979
|
+
_use_args_: ClassVar[bool] = ...
|
|
980
|
+
_value2member_map_: ClassVar[dict] = ...
|
|
981
|
+
def __format__(self, *args, **kwargs) -> str:
|
|
982
|
+
"""Convert to a string according to format_spec."""
|
|
983
|
+
|
|
984
|
+
class CufftdxRealMode(enum.IntEnum):
|
|
985
|
+
"""
|
|
986
|
+
Real data mode.Check cufftdx real data mode documentation for more
|
|
987
|
+
details (https://docs.nvidia.com/cuda/cufftdx/api/methods.html#real-
|
|
988
|
+
element-layouts)
|
|
989
|
+
|
|
990
|
+
See `cufftdxRealMode`.
|
|
991
|
+
"""
|
|
992
|
+
__new__: ClassVar[Callable] = ...
|
|
993
|
+
FOLDED: ClassVar[CufftdxRealMode] = ...
|
|
994
|
+
NORMAL: ClassVar[CufftdxRealMode] = ...
|
|
995
|
+
_generate_next_value_: ClassVar[Callable] = ...
|
|
996
|
+
_hashable_values_: ClassVar[list] = ...
|
|
997
|
+
_member_map_: ClassVar[dict] = ...
|
|
998
|
+
_member_names_: ClassVar[list] = ...
|
|
999
|
+
_member_type_: ClassVar[type[int]] = ...
|
|
1000
|
+
_unhashable_values_: ClassVar[list] = ...
|
|
1001
|
+
_unhashable_values_map_: ClassVar[dict] = ...
|
|
1002
|
+
_use_args_: ClassVar[bool] = ...
|
|
1003
|
+
_value2member_map_: ClassVar[dict] = ...
|
|
1004
|
+
def __format__(self, *args, **kwargs) -> str:
|
|
1005
|
+
"""Convert to a string according to format_spec."""
|
|
1006
|
+
|
|
1007
|
+
class CufftdxTraitType(enum.IntEnum):
|
|
1008
|
+
"""
|
|
1009
|
+
cufftDx traitsThe set of supported types of traits that can be accessed
|
|
1010
|
+
from finalized sources that use cufftDx. Check cufftDx Execution
|
|
1011
|
+
Thread Traits documentation for more details
|
|
1012
|
+
(https://docs.nvidia.com/cuda/cufftdx/api/traits.html#thread-traits)
|
|
1013
|
+
Check cufftDx Execution Block Traits documentation for more details
|
|
1014
|
+
(https://docs.nvidia.com/cuda/cufftdx/api/traits.html#block-traits)
|
|
1015
|
+
|
|
1016
|
+
See `cufftdxTraitType`.
|
|
1017
|
+
"""
|
|
1018
|
+
__new__: ClassVar[Callable] = ...
|
|
1019
|
+
BLOCK_DIM: ClassVar[CufftdxTraitType] = ...
|
|
1020
|
+
ELEMENTS_PER_THREAD: ClassVar[CufftdxTraitType] = ...
|
|
1021
|
+
FFTS_PER_BLOCK: ClassVar[CufftdxTraitType] = ...
|
|
1022
|
+
IMPLICIT_TYPE_BATCHING: ClassVar[CufftdxTraitType] = ...
|
|
1023
|
+
INPUT_ELEMENTS_PER_THREAD: ClassVar[CufftdxTraitType] = ...
|
|
1024
|
+
INPUT_LENGTH: ClassVar[CufftdxTraitType] = ...
|
|
1025
|
+
INPUT_TYPE: ClassVar[CufftdxTraitType] = ...
|
|
1026
|
+
OUTPUT_ELEMENTS_PER_THREAD: ClassVar[CufftdxTraitType] = ...
|
|
1027
|
+
OUTPUT_LENGTH: ClassVar[CufftdxTraitType] = ...
|
|
1028
|
+
OUTPUT_TYPE: ClassVar[CufftdxTraitType] = ...
|
|
1029
|
+
SHARED_MEMORY_SIZE: ClassVar[CufftdxTraitType] = ...
|
|
1030
|
+
STORAGE_SIZE: ClassVar[CufftdxTraitType] = ...
|
|
1031
|
+
STRIDE: ClassVar[CufftdxTraitType] = ...
|
|
1032
|
+
SUGGESTED_FFTS_PER_BLOCK: ClassVar[CufftdxTraitType] = ...
|
|
1033
|
+
SYMBOL_NAME: ClassVar[CufftdxTraitType] = ...
|
|
1034
|
+
VALUE_TYPE: ClassVar[CufftdxTraitType] = ...
|
|
1035
|
+
_generate_next_value_: ClassVar[Callable] = ...
|
|
1036
|
+
_hashable_values_: ClassVar[list] = ...
|
|
1037
|
+
_member_map_: ClassVar[dict] = ...
|
|
1038
|
+
_member_names_: ClassVar[list] = ...
|
|
1039
|
+
_member_type_: ClassVar[type[int]] = ...
|
|
1040
|
+
_unhashable_values_: ClassVar[list] = ...
|
|
1041
|
+
_unhashable_values_map_: ClassVar[dict] = ...
|
|
1042
|
+
_use_args_: ClassVar[bool] = ...
|
|
1043
|
+
_value2member_map_: ClassVar[dict] = ...
|
|
1044
|
+
def __format__(self, *args, **kwargs) -> str:
|
|
1045
|
+
"""Convert to a string according to format_spec."""
|
|
1046
|
+
|
|
1047
|
+
class CufftdxType(enum.IntEnum):
|
|
1048
|
+
"""
|
|
1049
|
+
Type of computation data.Check cufftdx::Type operator documentation for
|
|
1050
|
+
more details
|
|
1051
|
+
(https://docs.nvidia.com/cuda/cufftdx/api/methods.html#block-execute-
|
|
1052
|
+
method)
|
|
1053
|
+
|
|
1054
|
+
See `cufftdxType`.
|
|
1055
|
+
"""
|
|
1056
|
+
__new__: ClassVar[Callable] = ...
|
|
1057
|
+
C2C: ClassVar[CufftdxType] = ...
|
|
1058
|
+
C2R: ClassVar[CufftdxType] = ...
|
|
1059
|
+
R2C: ClassVar[CufftdxType] = ...
|
|
1060
|
+
_generate_next_value_: ClassVar[Callable] = ...
|
|
1061
|
+
_hashable_values_: ClassVar[list] = ...
|
|
1062
|
+
_member_map_: ClassVar[dict] = ...
|
|
1063
|
+
_member_names_: ClassVar[list] = ...
|
|
1064
|
+
_member_type_: ClassVar[type[int]] = ...
|
|
1065
|
+
_unhashable_values_: ClassVar[list] = ...
|
|
1066
|
+
_unhashable_values_map_: ClassVar[dict] = ...
|
|
1067
|
+
_use_args_: ClassVar[bool] = ...
|
|
1068
|
+
_value2member_map_: ClassVar[dict] = ...
|
|
1069
|
+
def __format__(self, *args, **kwargs) -> str:
|
|
1070
|
+
"""Convert to a string according to format_spec."""
|
|
1071
|
+
|
|
1072
|
+
class CuranddxDeviceFunctionType(enum.IntEnum):
|
|
1073
|
+
"""
|
|
1074
|
+
Device functions.The set of supported device functions. Must be passed
|
|
1075
|
+
using CURANDDX_OPERATOR_DEVICE_FUNCTIONS and can be or'ed to indicate
|
|
1076
|
+
that libmathdx should generate multiple device functions.
|
|
1077
|
+
|
|
1078
|
+
See `curanddxDeviceFunctionType`.
|
|
1079
|
+
"""
|
|
1080
|
+
__new__: ClassVar[Callable] = ...
|
|
1081
|
+
DESTROY_STATE: ClassVar[CuranddxDeviceFunctionType] = ...
|
|
1082
|
+
GENERATE: ClassVar[CuranddxDeviceFunctionType] = ...
|
|
1083
|
+
INIT_STATE: ClassVar[CuranddxDeviceFunctionType] = ...
|
|
1084
|
+
SKIP_OFFSET: ClassVar[CuranddxDeviceFunctionType] = ...
|
|
1085
|
+
SKIP_SEQUENCE: ClassVar[CuranddxDeviceFunctionType] = ...
|
|
1086
|
+
SKIP_SUBSEQUENCE: ClassVar[CuranddxDeviceFunctionType] = ...
|
|
1087
|
+
_generate_next_value_: ClassVar[Callable] = ...
|
|
1088
|
+
_hashable_values_: ClassVar[list] = ...
|
|
1089
|
+
_member_map_: ClassVar[dict] = ...
|
|
1090
|
+
_member_names_: ClassVar[list] = ...
|
|
1091
|
+
_member_type_: ClassVar[type[int]] = ...
|
|
1092
|
+
_unhashable_values_: ClassVar[list] = ...
|
|
1093
|
+
_unhashable_values_map_: ClassVar[dict] = ...
|
|
1094
|
+
_use_args_: ClassVar[bool] = ...
|
|
1095
|
+
_value2member_map_: ClassVar[dict] = ...
|
|
1096
|
+
def __format__(self, *args, **kwargs) -> str:
|
|
1097
|
+
"""Convert to a string according to format_spec."""
|
|
1098
|
+
|
|
1099
|
+
class CuranddxDistribution(enum.IntEnum):
|
|
1100
|
+
"""
|
|
1101
|
+
Distribution type.Indicate the random number generation to use. Can be
|
|
1102
|
+
set using CURANDDX_OPERATOR_DISTRIBUTION and further customized using
|
|
1103
|
+
CURANDDX_OPERATOR_NORMAL_METHOD. See cuRANDDx distributions in the
|
|
1104
|
+
documentation:
|
|
1105
|
+
https://docs.nvidia.com/cuda/curanddx/api/methods.html#random-number-
|
|
1106
|
+
generation-with-distributions.
|
|
1107
|
+
|
|
1108
|
+
See `curanddxDistribution`.
|
|
1109
|
+
"""
|
|
1110
|
+
__new__: ClassVar[Callable] = ...
|
|
1111
|
+
LOG_NORMAL: ClassVar[CuranddxDistribution] = ...
|
|
1112
|
+
NORMAL: ClassVar[CuranddxDistribution] = ...
|
|
1113
|
+
POISSON: ClassVar[CuranddxDistribution] = ...
|
|
1114
|
+
UNIFORM: ClassVar[CuranddxDistribution] = ...
|
|
1115
|
+
UNIFORM_BITS: ClassVar[CuranddxDistribution] = ...
|
|
1116
|
+
_generate_next_value_: ClassVar[Callable] = ...
|
|
1117
|
+
_hashable_values_: ClassVar[list] = ...
|
|
1118
|
+
_member_map_: ClassVar[dict] = ...
|
|
1119
|
+
_member_names_: ClassVar[list] = ...
|
|
1120
|
+
_member_type_: ClassVar[type[int]] = ...
|
|
1121
|
+
_unhashable_values_: ClassVar[list] = ...
|
|
1122
|
+
_unhashable_values_map_: ClassVar[dict] = ...
|
|
1123
|
+
_use_args_: ClassVar[bool] = ...
|
|
1124
|
+
_value2member_map_: ClassVar[dict] = ...
|
|
1125
|
+
def __format__(self, *args, **kwargs) -> str:
|
|
1126
|
+
"""Convert to a string according to format_spec."""
|
|
1127
|
+
|
|
1128
|
+
class CuranddxGenerateMethod(enum.IntEnum):
|
|
1129
|
+
"""
|
|
1130
|
+
Generation method type.Specifies which generation method to use for
|
|
1131
|
+
cuRANDDx distributions. Set using CURANDDX_OPERATOR_GENERATE_METHOD .
|
|
1132
|
+
See cuRANDDx execution methods in the documentation:
|
|
1133
|
+
https://docs.nvidia.com/cuda/curanddx/api/methods.html
|
|
1134
|
+
|
|
1135
|
+
See `curanddxGenerateMethod`.
|
|
1136
|
+
"""
|
|
1137
|
+
__new__: ClassVar[Callable] = ...
|
|
1138
|
+
PAIR: ClassVar[CuranddxGenerateMethod] = ...
|
|
1139
|
+
QUAD: ClassVar[CuranddxGenerateMethod] = ...
|
|
1140
|
+
SINGLE: ClassVar[CuranddxGenerateMethod] = ...
|
|
1141
|
+
_generate_next_value_: ClassVar[Callable] = ...
|
|
1142
|
+
_hashable_values_: ClassVar[list] = ...
|
|
1143
|
+
_member_map_: ClassVar[dict] = ...
|
|
1144
|
+
_member_names_: ClassVar[list] = ...
|
|
1145
|
+
_member_type_: ClassVar[type[int]] = ...
|
|
1146
|
+
_unhashable_values_: ClassVar[list] = ...
|
|
1147
|
+
_unhashable_values_map_: ClassVar[dict] = ...
|
|
1148
|
+
_use_args_: ClassVar[bool] = ...
|
|
1149
|
+
_value2member_map_: ClassVar[dict] = ...
|
|
1150
|
+
def __format__(self, *args, **kwargs) -> str:
|
|
1151
|
+
"""Convert to a string according to format_spec."""
|
|
1152
|
+
|
|
1153
|
+
class CuranddxGenerator(enum.IntEnum):
|
|
1154
|
+
"""
|
|
1155
|
+
Generator type.Matches cuRANDDx generator kinds. Set using
|
|
1156
|
+
CURANDDX_OPERATOR_GENERATOR . See cuRANDDx generators in the
|
|
1157
|
+
documentation: https://docs.nvidia.com/cuda/curanddx/api/description_op
|
|
1158
|
+
s.html#generator-operator-label
|
|
1159
|
+
|
|
1160
|
+
See `curanddxGenerator`.
|
|
1161
|
+
"""
|
|
1162
|
+
__new__: ClassVar[Callable] = ...
|
|
1163
|
+
MRG32K3A: ClassVar[CuranddxGenerator] = ...
|
|
1164
|
+
PCG: ClassVar[CuranddxGenerator] = ...
|
|
1165
|
+
PHILOX4_32: ClassVar[CuranddxGenerator] = ...
|
|
1166
|
+
SCRAMBLED_SOBOL32: ClassVar[CuranddxGenerator] = ...
|
|
1167
|
+
SCRAMBLED_SOBOL64: ClassVar[CuranddxGenerator] = ...
|
|
1168
|
+
SOBOL32: ClassVar[CuranddxGenerator] = ...
|
|
1169
|
+
SOBOL64: ClassVar[CuranddxGenerator] = ...
|
|
1170
|
+
XORWOW: ClassVar[CuranddxGenerator] = ...
|
|
1171
|
+
_generate_next_value_: ClassVar[Callable] = ...
|
|
1172
|
+
_hashable_values_: ClassVar[list] = ...
|
|
1173
|
+
_member_map_: ClassVar[dict] = ...
|
|
1174
|
+
_member_names_: ClassVar[list] = ...
|
|
1175
|
+
_member_type_: ClassVar[type[int]] = ...
|
|
1176
|
+
_unhashable_values_: ClassVar[list] = ...
|
|
1177
|
+
_unhashable_values_map_: ClassVar[dict] = ...
|
|
1178
|
+
_use_args_: ClassVar[bool] = ...
|
|
1179
|
+
_value2member_map_: ClassVar[dict] = ...
|
|
1180
|
+
def __format__(self, *args, **kwargs) -> str:
|
|
1181
|
+
"""Convert to a string according to format_spec."""
|
|
1182
|
+
|
|
1183
|
+
class CuranddxNormalMethod(enum.IntEnum):
|
|
1184
|
+
"""
|
|
1185
|
+
Normal distribution method type.Specifies the method to use for normal
|
|
1186
|
+
and log-normal distributions. Set using CURANDDX_OPERATOR_NORMAL_METHOD
|
|
1187
|
+
. See cuRANDDx normal distribution documentation:
|
|
1188
|
+
https://docs.nvidia.com/cuda/curanddx/api/methods.html#normal-
|
|
1189
|
+
distribution
|
|
1190
|
+
|
|
1191
|
+
See `curanddxNormalMethod`.
|
|
1192
|
+
"""
|
|
1193
|
+
__new__: ClassVar[Callable] = ...
|
|
1194
|
+
BOX_MULLER: ClassVar[CuranddxNormalMethod] = ...
|
|
1195
|
+
ICDF: ClassVar[CuranddxNormalMethod] = ...
|
|
1196
|
+
_generate_next_value_: ClassVar[Callable] = ...
|
|
1197
|
+
_hashable_values_: ClassVar[list] = ...
|
|
1198
|
+
_member_map_: ClassVar[dict] = ...
|
|
1199
|
+
_member_names_: ClassVar[list] = ...
|
|
1200
|
+
_member_type_: ClassVar[type[int]] = ...
|
|
1201
|
+
_unhashable_values_: ClassVar[list] = ...
|
|
1202
|
+
_unhashable_values_map_: ClassVar[dict] = ...
|
|
1203
|
+
_use_args_: ClassVar[bool] = ...
|
|
1204
|
+
_value2member_map_: ClassVar[dict] = ...
|
|
1205
|
+
def __format__(self, *args, **kwargs) -> str:
|
|
1206
|
+
"""Convert to a string according to format_spec."""
|
|
1207
|
+
|
|
1208
|
+
class CuranddxOperatorType(enum.IntEnum):
|
|
1209
|
+
"""
|
|
1210
|
+
Operators.The set of supported curandDx operators.
|
|
1211
|
+
|
|
1212
|
+
See `curanddxOperatorType`.
|
|
1213
|
+
"""
|
|
1214
|
+
__new__: ClassVar[Callable] = ...
|
|
1215
|
+
DEVICE_FUNCTIONS: ClassVar[CuranddxOperatorType] = ...
|
|
1216
|
+
DISTRIBUTION: ClassVar[CuranddxOperatorType] = ...
|
|
1217
|
+
DISTRIBUTION_PARAMETERS: ClassVar[CuranddxOperatorType] = ...
|
|
1218
|
+
EXECUTION: ClassVar[CuranddxOperatorType] = ...
|
|
1219
|
+
GENERATE_METHOD: ClassVar[CuranddxOperatorType] = ...
|
|
1220
|
+
GENERATOR: ClassVar[CuranddxOperatorType] = ...
|
|
1221
|
+
NORMAL_METHOD: ClassVar[CuranddxOperatorType] = ...
|
|
1222
|
+
OUTPUT_TYPE: ClassVar[CuranddxOperatorType] = ...
|
|
1223
|
+
PHILOX_ROUNDS: ClassVar[CuranddxOperatorType] = ...
|
|
1224
|
+
SM: ClassVar[CuranddxOperatorType] = ...
|
|
1225
|
+
_generate_next_value_: ClassVar[Callable] = ...
|
|
1226
|
+
_hashable_values_: ClassVar[list] = ...
|
|
1227
|
+
_member_map_: ClassVar[dict] = ...
|
|
1228
|
+
_member_names_: ClassVar[list] = ...
|
|
1229
|
+
_member_type_: ClassVar[type[int]] = ...
|
|
1230
|
+
_unhashable_values_: ClassVar[list] = ...
|
|
1231
|
+
_unhashable_values_map_: ClassVar[dict] = ...
|
|
1232
|
+
_use_args_: ClassVar[bool] = ...
|
|
1233
|
+
_value2member_map_: ClassVar[dict] = ...
|
|
1234
|
+
def __format__(self, *args, **kwargs) -> str:
|
|
1235
|
+
"""Convert to a string according to format_spec."""
|
|
1236
|
+
|
|
1237
|
+
class CuranddxTraitType(enum.IntEnum):
|
|
1238
|
+
"""
|
|
1239
|
+
Traits.The set of supported types of traits that can be accessed from
|
|
1240
|
+
finalized sources that use curanddx.
|
|
1241
|
+
|
|
1242
|
+
See `curanddxTraitType`.
|
|
1243
|
+
"""
|
|
1244
|
+
__new__: ClassVar[Callable] = ...
|
|
1245
|
+
STATE_ALIGNMENT: ClassVar[CuranddxTraitType] = ...
|
|
1246
|
+
STATE_SIZE: ClassVar[CuranddxTraitType] = ...
|
|
1247
|
+
SYMBOL_DESTROY_STATE_NAME: ClassVar[CuranddxTraitType] = ...
|
|
1248
|
+
SYMBOL_GENERATE_NAME: ClassVar[CuranddxTraitType] = ...
|
|
1249
|
+
SYMBOL_INIT_STATE_NAME: ClassVar[CuranddxTraitType] = ...
|
|
1250
|
+
SYMBOL_SKIP_OFFSET_NAME: ClassVar[CuranddxTraitType] = ...
|
|
1251
|
+
SYMBOL_SKIP_SEQUENCE_NAME: ClassVar[CuranddxTraitType] = ...
|
|
1252
|
+
SYMBOL_SKIP_SUBSEQUENCE_NAME: ClassVar[CuranddxTraitType] = ...
|
|
1253
|
+
_generate_next_value_: ClassVar[Callable] = ...
|
|
1254
|
+
_hashable_values_: ClassVar[list] = ...
|
|
1255
|
+
_member_map_: ClassVar[dict] = ...
|
|
1256
|
+
_member_names_: ClassVar[list] = ...
|
|
1257
|
+
_member_type_: ClassVar[type[int]] = ...
|
|
1258
|
+
_unhashable_values_: ClassVar[list] = ...
|
|
1259
|
+
_unhashable_values_map_: ClassVar[dict] = ...
|
|
1260
|
+
_use_args_: ClassVar[bool] = ...
|
|
1261
|
+
_value2member_map_: ClassVar[dict] = ...
|
|
1262
|
+
def __format__(self, *args, **kwargs) -> str:
|
|
1263
|
+
"""Convert to a string according to format_spec."""
|
|
1264
|
+
|
|
1265
|
+
class CusolverdxApi(enum.IntEnum):
|
|
1266
|
+
"""
|
|
1267
|
+
Type of cusolverdx API.
|
|
1268
|
+
|
|
1269
|
+
See `cusolverdxApi`.
|
|
1270
|
+
"""
|
|
1271
|
+
__new__: ClassVar[Callable] = ...
|
|
1272
|
+
SMEM: ClassVar[CusolverdxApi] = ...
|
|
1273
|
+
SMEM_DYNAMIC_LD: ClassVar[CusolverdxApi] = ...
|
|
1274
|
+
_generate_next_value_: ClassVar[Callable] = ...
|
|
1275
|
+
_hashable_values_: ClassVar[list] = ...
|
|
1276
|
+
_member_map_: ClassVar[dict] = ...
|
|
1277
|
+
_member_names_: ClassVar[list] = ...
|
|
1278
|
+
_member_type_: ClassVar[type[int]] = ...
|
|
1279
|
+
_unhashable_values_: ClassVar[list] = ...
|
|
1280
|
+
_unhashable_values_map_: ClassVar[dict] = ...
|
|
1281
|
+
_use_args_: ClassVar[bool] = ...
|
|
1282
|
+
_value2member_map_: ClassVar[dict] = ...
|
|
1283
|
+
def __format__(self, *args, **kwargs) -> str:
|
|
1284
|
+
"""Convert to a string according to format_spec."""
|
|
1285
|
+
|
|
1286
|
+
class CusolverdxArrangement(enum.IntEnum):
|
|
1287
|
+
"""
|
|
1288
|
+
Data arrangement mode.Defines data arrangements in tensors' taking part
|
|
1289
|
+
in the calculation.
|
|
1290
|
+
|
|
1291
|
+
See `cusolverdxArrangement`.
|
|
1292
|
+
"""
|
|
1293
|
+
__new__: ClassVar[Callable] = ...
|
|
1294
|
+
COL_MAJOR: ClassVar[CusolverdxArrangement] = ...
|
|
1295
|
+
ROW_MAJOR: ClassVar[CusolverdxArrangement] = ...
|
|
1296
|
+
_generate_next_value_: ClassVar[Callable] = ...
|
|
1297
|
+
_hashable_values_: ClassVar[list] = ...
|
|
1298
|
+
_member_map_: ClassVar[dict] = ...
|
|
1299
|
+
_member_names_: ClassVar[list] = ...
|
|
1300
|
+
_member_type_: ClassVar[type[int]] = ...
|
|
1301
|
+
_unhashable_values_: ClassVar[list] = ...
|
|
1302
|
+
_unhashable_values_map_: ClassVar[dict] = ...
|
|
1303
|
+
_use_args_: ClassVar[bool] = ...
|
|
1304
|
+
_value2member_map_: ClassVar[dict] = ...
|
|
1305
|
+
def __format__(self, *args, **kwargs) -> str:
|
|
1306
|
+
"""Convert to a string according to format_spec."""
|
|
1307
|
+
|
|
1308
|
+
class CusolverdxDiag(enum.IntEnum):
|
|
1309
|
+
"""
|
|
1310
|
+
Diag operator.Indicates whether the matrix diagonal is unit (all ones)
|
|
1311
|
+
or non-unit.
|
|
1312
|
+
|
|
1313
|
+
See `cusolverdxDiag`.
|
|
1314
|
+
"""
|
|
1315
|
+
__new__: ClassVar[Callable] = ...
|
|
1316
|
+
NON_UNIT: ClassVar[CusolverdxDiag] = ...
|
|
1317
|
+
UNIT: ClassVar[CusolverdxDiag] = ...
|
|
1318
|
+
_generate_next_value_: ClassVar[Callable] = ...
|
|
1319
|
+
_hashable_values_: ClassVar[list] = ...
|
|
1320
|
+
_member_map_: ClassVar[dict] = ...
|
|
1321
|
+
_member_names_: ClassVar[list] = ...
|
|
1322
|
+
_member_type_: ClassVar[type[int]] = ...
|
|
1323
|
+
_unhashable_values_: ClassVar[list] = ...
|
|
1324
|
+
_unhashable_values_map_: ClassVar[dict] = ...
|
|
1325
|
+
_use_args_: ClassVar[bool] = ...
|
|
1326
|
+
_value2member_map_: ClassVar[dict] = ...
|
|
1327
|
+
def __format__(self, *args, **kwargs) -> str:
|
|
1328
|
+
"""Convert to a string according to format_spec."""
|
|
1329
|
+
|
|
1330
|
+
class CusolverdxFillMode(enum.IntEnum):
|
|
1331
|
+
"""
|
|
1332
|
+
Tensor fill mode.For symmetric matrix the fill mode can be upper or
|
|
1333
|
+
lower triangular
|
|
1334
|
+
|
|
1335
|
+
See `cusolverdxFillMode`.
|
|
1336
|
+
"""
|
|
1337
|
+
__new__: ClassVar[Callable] = ...
|
|
1338
|
+
LOWER: ClassVar[CusolverdxFillMode] = ...
|
|
1339
|
+
UPPER: ClassVar[CusolverdxFillMode] = ...
|
|
1340
|
+
_generate_next_value_: ClassVar[Callable] = ...
|
|
1341
|
+
_hashable_values_: ClassVar[list] = ...
|
|
1342
|
+
_member_map_: ClassVar[dict] = ...
|
|
1343
|
+
_member_names_: ClassVar[list] = ...
|
|
1344
|
+
_member_type_: ClassVar[type[int]] = ...
|
|
1345
|
+
_unhashable_values_: ClassVar[list] = ...
|
|
1346
|
+
_unhashable_values_map_: ClassVar[dict] = ...
|
|
1347
|
+
_use_args_: ClassVar[bool] = ...
|
|
1348
|
+
_value2member_map_: ClassVar[dict] = ...
|
|
1349
|
+
def __format__(self, *args, **kwargs) -> str:
|
|
1350
|
+
"""Convert to a string according to format_spec."""
|
|
1351
|
+
|
|
1352
|
+
class CusolverdxFunction(enum.IntEnum):
|
|
1353
|
+
'''
|
|
1354
|
+
Type of device function.The function signatures depend on the selected
|
|
1355
|
+
cusolverdxApi_t: - For CUSOLVERDX_API_SMEM: Leading dimensions are
|
|
1356
|
+
fixed at compile time. - For CUSOLVERDX_API_SMEM_DYNAMIC_LD: Leading
|
|
1357
|
+
dimensions are passed as pointers to `unsigned` at runtime. Type
|
|
1358
|
+
definitions used in function signatures: - `value_type`: The value
|
|
1359
|
+
type determined by CUSOLVERDX_OPERATOR_TYPE and
|
|
1360
|
+
CUSOLVERDX_OPERATOR_PRECISION. For real types: `float` or `double`. For
|
|
1361
|
+
complex types: `cuFloatComplex` or `cuDoubleComplex`. - `status_type`:
|
|
1362
|
+
A signed 32b integer (`int`) used to return status information. All
|
|
1363
|
+
functions are `extern "C"` and the symbol name can be queried using
|
|
1364
|
+
CUSOLVERDX_TRAIT_SYMBOL_NAME.
|
|
1365
|
+
|
|
1366
|
+
See `cusolverdxFunction`.
|
|
1367
|
+
'''
|
|
1368
|
+
__new__: ClassVar[Callable] = ...
|
|
1369
|
+
GELQF: ClassVar[CusolverdxFunction] = ...
|
|
1370
|
+
GELS: ClassVar[CusolverdxFunction] = ...
|
|
1371
|
+
GEQRF: ClassVar[CusolverdxFunction] = ...
|
|
1372
|
+
GESV_NO_PIVOT: ClassVar[CusolverdxFunction] = ...
|
|
1373
|
+
GESV_PARTIAL_PIVOT: ClassVar[CusolverdxFunction] = ...
|
|
1374
|
+
GETRF_NO_PIVOT: ClassVar[CusolverdxFunction] = ...
|
|
1375
|
+
GETRF_PARTIAL_PIVOT: ClassVar[CusolverdxFunction] = ...
|
|
1376
|
+
GETRS_NO_PIVOT: ClassVar[CusolverdxFunction] = ...
|
|
1377
|
+
GETRS_PARTIAL_PIVOT: ClassVar[CusolverdxFunction] = ...
|
|
1378
|
+
GTSV_NO_PIVOT: ClassVar[CusolverdxFunction] = ...
|
|
1379
|
+
HEEV: ClassVar[CusolverdxFunction] = ...
|
|
1380
|
+
HTEV: ClassVar[CusolverdxFunction] = ...
|
|
1381
|
+
POSV: ClassVar[CusolverdxFunction] = ...
|
|
1382
|
+
POTRF: ClassVar[CusolverdxFunction] = ...
|
|
1383
|
+
POTRS: ClassVar[CusolverdxFunction] = ...
|
|
1384
|
+
TRSM: ClassVar[CusolverdxFunction] = ...
|
|
1385
|
+
UNGLQ: ClassVar[CusolverdxFunction] = ...
|
|
1386
|
+
UNGQR: ClassVar[CusolverdxFunction] = ...
|
|
1387
|
+
UNMLQ: ClassVar[CusolverdxFunction] = ...
|
|
1388
|
+
UNMQR: ClassVar[CusolverdxFunction] = ...
|
|
1389
|
+
_generate_next_value_: ClassVar[Callable] = ...
|
|
1390
|
+
_hashable_values_: ClassVar[list] = ...
|
|
1391
|
+
_member_map_: ClassVar[dict] = ...
|
|
1392
|
+
_member_names_: ClassVar[list] = ...
|
|
1393
|
+
_member_type_: ClassVar[type[int]] = ...
|
|
1394
|
+
_unhashable_values_: ClassVar[list] = ...
|
|
1395
|
+
_unhashable_values_map_: ClassVar[dict] = ...
|
|
1396
|
+
_use_args_: ClassVar[bool] = ...
|
|
1397
|
+
_value2member_map_: ClassVar[dict] = ...
|
|
1398
|
+
def __format__(self, *args, **kwargs) -> str:
|
|
1399
|
+
"""Convert to a string according to format_spec."""
|
|
1400
|
+
|
|
1401
|
+
class CusolverdxJob(enum.IntEnum):
|
|
1402
|
+
"""
|
|
1403
|
+
Job operator.Specifies whether and how eigenvectors are computed and
|
|
1404
|
+
stored in eigenvalue operations.
|
|
1405
|
+
|
|
1406
|
+
See `cusolverdxJob`.
|
|
1407
|
+
"""
|
|
1408
|
+
__new__: ClassVar[Callable] = ...
|
|
1409
|
+
ALL_VECTORS: ClassVar[CusolverdxJob] = ...
|
|
1410
|
+
MULTIPLY_VECTORS: ClassVar[CusolverdxJob] = ...
|
|
1411
|
+
NO_VECTORS: ClassVar[CusolverdxJob] = ...
|
|
1412
|
+
OVERWRITE_VECTORS: ClassVar[CusolverdxJob] = ...
|
|
1413
|
+
_generate_next_value_: ClassVar[Callable] = ...
|
|
1414
|
+
_hashable_values_: ClassVar[list] = ...
|
|
1415
|
+
_member_map_: ClassVar[dict] = ...
|
|
1416
|
+
_member_names_: ClassVar[list] = ...
|
|
1417
|
+
_member_type_: ClassVar[type[int]] = ...
|
|
1418
|
+
_unhashable_values_: ClassVar[list] = ...
|
|
1419
|
+
_unhashable_values_map_: ClassVar[dict] = ...
|
|
1420
|
+
_use_args_: ClassVar[bool] = ...
|
|
1421
|
+
_value2member_map_: ClassVar[dict] = ...
|
|
1422
|
+
def __format__(self, *args, **kwargs) -> str:
|
|
1423
|
+
"""Convert to a string according to format_spec."""
|
|
1424
|
+
|
|
1425
|
+
class CusolverdxOperatorType(enum.IntEnum):
|
|
1426
|
+
"""
|
|
1427
|
+
Operators.The set of supported cusolverDx operators.
|
|
1428
|
+
|
|
1429
|
+
See `cusolverdxOperatorType`.
|
|
1430
|
+
"""
|
|
1431
|
+
__new__: ClassVar[Callable] = ...
|
|
1432
|
+
API: ClassVar[CusolverdxOperatorType] = ...
|
|
1433
|
+
ARRANGEMENT: ClassVar[CusolverdxOperatorType] = ...
|
|
1434
|
+
BATCHES_PER_BLOCK: ClassVar[CusolverdxOperatorType] = ...
|
|
1435
|
+
BLOCK_DIM: ClassVar[CusolverdxOperatorType] = ...
|
|
1436
|
+
DIAG: ClassVar[CusolverdxOperatorType] = ...
|
|
1437
|
+
EXECUTION: ClassVar[CusolverdxOperatorType] = ...
|
|
1438
|
+
FILL_MODE: ClassVar[CusolverdxOperatorType] = ...
|
|
1439
|
+
FUNCTION: ClassVar[CusolverdxOperatorType] = ...
|
|
1440
|
+
JOB: ClassVar[CusolverdxOperatorType] = ...
|
|
1441
|
+
LEADING_DIMENSION: ClassVar[CusolverdxOperatorType] = ...
|
|
1442
|
+
PRECISION: ClassVar[CusolverdxOperatorType] = ...
|
|
1443
|
+
SIDE: ClassVar[CusolverdxOperatorType] = ...
|
|
1444
|
+
SIZE: ClassVar[CusolverdxOperatorType] = ...
|
|
1445
|
+
SM: ClassVar[CusolverdxOperatorType] = ...
|
|
1446
|
+
TRANSPOSE_MODE: ClassVar[CusolverdxOperatorType] = ...
|
|
1447
|
+
TYPE: ClassVar[CusolverdxOperatorType] = ...
|
|
1448
|
+
_generate_next_value_: ClassVar[Callable] = ...
|
|
1449
|
+
_hashable_values_: ClassVar[list] = ...
|
|
1450
|
+
_member_map_: ClassVar[dict] = ...
|
|
1451
|
+
_member_names_: ClassVar[list] = ...
|
|
1452
|
+
_member_type_: ClassVar[type[int]] = ...
|
|
1453
|
+
_unhashable_values_: ClassVar[list] = ...
|
|
1454
|
+
_unhashable_values_map_: ClassVar[dict] = ...
|
|
1455
|
+
_use_args_: ClassVar[bool] = ...
|
|
1456
|
+
_value2member_map_: ClassVar[dict] = ...
|
|
1457
|
+
def __format__(self, *args, **kwargs) -> str:
|
|
1458
|
+
"""Convert to a string according to format_spec."""
|
|
1459
|
+
|
|
1460
|
+
class CusolverdxSide(enum.IntEnum):
|
|
1461
|
+
"""
|
|
1462
|
+
Side operator.Indicates which side an operator is applied from in
|
|
1463
|
+
operations such as matrix multiplication.
|
|
1464
|
+
|
|
1465
|
+
See `cusolverdxSide`.
|
|
1466
|
+
"""
|
|
1467
|
+
__new__: ClassVar[Callable] = ...
|
|
1468
|
+
LEFT: ClassVar[CusolverdxSide] = ...
|
|
1469
|
+
RIGHT: ClassVar[CusolverdxSide] = ...
|
|
1470
|
+
_generate_next_value_: ClassVar[Callable] = ...
|
|
1471
|
+
_hashable_values_: ClassVar[list] = ...
|
|
1472
|
+
_member_map_: ClassVar[dict] = ...
|
|
1473
|
+
_member_names_: ClassVar[list] = ...
|
|
1474
|
+
_member_type_: ClassVar[type[int]] = ...
|
|
1475
|
+
_unhashable_values_: ClassVar[list] = ...
|
|
1476
|
+
_unhashable_values_map_: ClassVar[dict] = ...
|
|
1477
|
+
_use_args_: ClassVar[bool] = ...
|
|
1478
|
+
_value2member_map_: ClassVar[dict] = ...
|
|
1479
|
+
def __format__(self, *args, **kwargs) -> str:
|
|
1480
|
+
"""Convert to a string according to format_spec."""
|
|
1481
|
+
|
|
1482
|
+
class CusolverdxTraitType(enum.IntEnum):
|
|
1483
|
+
"""
|
|
1484
|
+
Traits.The set of supported types of traits that can be accessed from
|
|
1485
|
+
finalized sources that use cusolverdx.
|
|
1486
|
+
|
|
1487
|
+
See `cusolverdxTraitType`.
|
|
1488
|
+
"""
|
|
1489
|
+
__new__: ClassVar[Callable] = ...
|
|
1490
|
+
ARRANGEMENT: ClassVar[CusolverdxTraitType] = ...
|
|
1491
|
+
BATCHES_PER_BLOCK: ClassVar[CusolverdxTraitType] = ...
|
|
1492
|
+
BLOCK_DIM: ClassVar[CusolverdxTraitType] = ...
|
|
1493
|
+
LEADING_DIMENSION: ClassVar[CusolverdxTraitType] = ...
|
|
1494
|
+
SHARED_MEMORY_SIZE: ClassVar[CusolverdxTraitType] = ...
|
|
1495
|
+
SUGGESTED_BATCHES_PER_BLOCK: ClassVar[CusolverdxTraitType] = ...
|
|
1496
|
+
SUGGESTED_BLOCK_DIM: ClassVar[CusolverdxTraitType] = ...
|
|
1497
|
+
SUGGESTED_LEADING_DIMENSION: ClassVar[CusolverdxTraitType] = ...
|
|
1498
|
+
SYMBOL_NAME: ClassVar[CusolverdxTraitType] = ...
|
|
1499
|
+
TYPE: ClassVar[CusolverdxTraitType] = ...
|
|
1500
|
+
VALUE_TYPE: ClassVar[CusolverdxTraitType] = ...
|
|
1501
|
+
WORKSPACE_SIZE: ClassVar[CusolverdxTraitType] = ...
|
|
1502
|
+
_generate_next_value_: ClassVar[Callable] = ...
|
|
1503
|
+
_hashable_values_: ClassVar[list] = ...
|
|
1504
|
+
_member_map_: ClassVar[dict] = ...
|
|
1505
|
+
_member_names_: ClassVar[list] = ...
|
|
1506
|
+
_member_type_: ClassVar[type[int]] = ...
|
|
1507
|
+
_unhashable_values_: ClassVar[list] = ...
|
|
1508
|
+
_unhashable_values_map_: ClassVar[dict] = ...
|
|
1509
|
+
_use_args_: ClassVar[bool] = ...
|
|
1510
|
+
_value2member_map_: ClassVar[dict] = ...
|
|
1511
|
+
def __format__(self, *args, **kwargs) -> str:
|
|
1512
|
+
"""Convert to a string according to format_spec."""
|
|
1513
|
+
|
|
1514
|
+
class CusolverdxTransposeMode(enum.IntEnum):
|
|
1515
|
+
"""
|
|
1516
|
+
Transpose mode.Indicates inputs or outputs must be considered
|
|
1517
|
+
transposed.
|
|
1518
|
+
|
|
1519
|
+
See `cusolverdxTransposeMode`.
|
|
1520
|
+
"""
|
|
1521
|
+
__new__: ClassVar[Callable] = ...
|
|
1522
|
+
CONJ_TRANSPOSED: ClassVar[CusolverdxTransposeMode] = ...
|
|
1523
|
+
NON_TRANSPOSED: ClassVar[CusolverdxTransposeMode] = ...
|
|
1524
|
+
TRANSPOSED: ClassVar[CusolverdxTransposeMode] = ...
|
|
1525
|
+
_generate_next_value_: ClassVar[Callable] = ...
|
|
1526
|
+
_hashable_values_: ClassVar[list] = ...
|
|
1527
|
+
_member_map_: ClassVar[dict] = ...
|
|
1528
|
+
_member_names_: ClassVar[list] = ...
|
|
1529
|
+
_member_type_: ClassVar[type[int]] = ...
|
|
1530
|
+
_unhashable_values_: ClassVar[list] = ...
|
|
1531
|
+
_unhashable_values_map_: ClassVar[dict] = ...
|
|
1532
|
+
_use_args_: ClassVar[bool] = ...
|
|
1533
|
+
_value2member_map_: ClassVar[dict] = ...
|
|
1534
|
+
def __format__(self, *args, **kwargs) -> str:
|
|
1535
|
+
"""Convert to a string according to format_spec."""
|
|
1536
|
+
|
|
1537
|
+
class CusolverdxType(enum.IntEnum):
|
|
1538
|
+
"""
|
|
1539
|
+
Type of input values.
|
|
1540
|
+
|
|
1541
|
+
See `cusolverdxType`.
|
|
1542
|
+
"""
|
|
1543
|
+
__new__: ClassVar[Callable] = ...
|
|
1544
|
+
COMPLEX: ClassVar[CusolverdxType] = ...
|
|
1545
|
+
REAL: ClassVar[CusolverdxType] = ...
|
|
1546
|
+
_generate_next_value_: ClassVar[Callable] = ...
|
|
1547
|
+
_hashable_values_: ClassVar[list] = ...
|
|
1548
|
+
_member_map_: ClassVar[dict] = ...
|
|
1549
|
+
_member_names_: ClassVar[list] = ...
|
|
1550
|
+
_member_type_: ClassVar[type[int]] = ...
|
|
1551
|
+
_unhashable_values_: ClassVar[list] = ...
|
|
1552
|
+
_unhashable_values_map_: ClassVar[dict] = ...
|
|
1553
|
+
_use_args_: ClassVar[bool] = ...
|
|
1554
|
+
_value2member_map_: ClassVar[dict] = ...
|
|
1555
|
+
def __format__(self, *args, **kwargs) -> str:
|
|
1556
|
+
"""Convert to a string according to format_spec."""
|
|
1557
|
+
|
|
1558
|
+
class LibMathDxError(Exception):
|
|
1559
|
+
def __init__(self, status) -> Any:
|
|
1560
|
+
"""LibMathDxError.__init__(self, status)"""
|
|
1561
|
+
def __reduce__(self) -> Any:
|
|
1562
|
+
"""LibMathDxError.__reduce__(self)"""
|