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,552 @@
|
|
|
1
|
+
# This code was automatically generated across versions from 0.3.1 to 0.3.2, generator version 0.3.1.dev1418+g63712a33a.d20260318. Do not modify it directly.
|
|
2
|
+
# This layer exposes the C header to Cython as-is.
|
|
3
|
+
|
|
4
|
+
from libc.stdint cimport int64_t
|
|
5
|
+
|
|
6
|
+
|
|
7
|
+
###############################################################################
|
|
8
|
+
# Types (structs, enums, ...)
|
|
9
|
+
###############################################################################
|
|
10
|
+
|
|
11
|
+
# enums
|
|
12
|
+
ctypedef enum commondxValueType "commondxValueType":
|
|
13
|
+
COMMONDX_R_8F_E5M2 "COMMONDX_R_8F_E5M2" = 0
|
|
14
|
+
COMMONDX_C_8F_E5M2 "COMMONDX_C_8F_E5M2" = 1
|
|
15
|
+
COMMONDX_R_8F_E4M3 "COMMONDX_R_8F_E4M3" = 2
|
|
16
|
+
COMMONDX_C_8F_E4M3 "COMMONDX_C_8F_E4M3" = 3
|
|
17
|
+
COMMONDX_R_16BF "COMMONDX_R_16BF" = 4
|
|
18
|
+
COMMONDX_C_16BF "COMMONDX_C_16BF" = 5
|
|
19
|
+
COMMONDX_R_16F2 "COMMONDX_R_16F2" = 6
|
|
20
|
+
COMMONDX_R_16F "COMMONDX_R_16F" = 7
|
|
21
|
+
COMMONDX_C_16F "COMMONDX_C_16F" = 8
|
|
22
|
+
COMMONDX_C_16F2 "COMMONDX_C_16F2" = 9
|
|
23
|
+
COMMONDX_R_32TF "COMMONDX_R_32TF" = 10
|
|
24
|
+
COMMONDX_C_32TF "COMMONDX_C_32TF" = 11
|
|
25
|
+
COMMONDX_R_32F "COMMONDX_R_32F" = 12
|
|
26
|
+
COMMONDX_C_32F "COMMONDX_C_32F" = 13
|
|
27
|
+
COMMONDX_R_64F "COMMONDX_R_64F" = 14
|
|
28
|
+
COMMONDX_C_64F "COMMONDX_C_64F" = 15
|
|
29
|
+
COMMONDX_R_8I "COMMONDX_R_8I" = 16
|
|
30
|
+
COMMONDX_C_8I "COMMONDX_C_8I" = 17
|
|
31
|
+
COMMONDX_R_16I "COMMONDX_R_16I" = 18
|
|
32
|
+
COMMONDX_C_16I "COMMONDX_C_16I" = 19
|
|
33
|
+
COMMONDX_R_32I "COMMONDX_R_32I" = 20
|
|
34
|
+
COMMONDX_C_32I "COMMONDX_C_32I" = 21
|
|
35
|
+
COMMONDX_R_64I "COMMONDX_R_64I" = 22
|
|
36
|
+
COMMONDX_C_64I "COMMONDX_C_64I" = 23
|
|
37
|
+
COMMONDX_R_8UI "COMMONDX_R_8UI" = 24
|
|
38
|
+
COMMONDX_C_8UI "COMMONDX_C_8UI" = 25
|
|
39
|
+
COMMONDX_R_16UI "COMMONDX_R_16UI" = 26
|
|
40
|
+
COMMONDX_C_16UI "COMMONDX_C_16UI" = 27
|
|
41
|
+
COMMONDX_R_32UI "COMMONDX_R_32UI" = 28
|
|
42
|
+
COMMONDX_C_32UI "COMMONDX_C_32UI" = 29
|
|
43
|
+
COMMONDX_R_64UI "COMMONDX_R_64UI" = 30
|
|
44
|
+
COMMONDX_C_64UI "COMMONDX_C_64UI" = 31
|
|
45
|
+
|
|
46
|
+
ctypedef enum commondxStatusType "commondxStatusType":
|
|
47
|
+
COMMONDX_SUCCESS "COMMONDX_SUCCESS" = 0
|
|
48
|
+
COMMONDX_INVALID_VALUE "COMMONDX_INVALID_VALUE" = 1
|
|
49
|
+
COMMONDX_INTERNAL_ERROR "COMMONDX_INTERNAL_ERROR" = 2
|
|
50
|
+
COMMONDX_COMPILATION_ERROR "COMMONDX_COMPILATION_ERROR" = 3
|
|
51
|
+
COMMONDX_CUFFT_ERROR "COMMONDX_CUFFT_ERROR" = 4
|
|
52
|
+
_COMMONDXSTATUSTYPE_INTERNAL_LOADING_ERROR "_COMMONDXSTATUSTYPE_INTERNAL_LOADING_ERROR" = -42
|
|
53
|
+
|
|
54
|
+
ctypedef enum commondxPrecision "commondxPrecision":
|
|
55
|
+
COMMONDX_PRECISION_F8_E5M2 "COMMONDX_PRECISION_F8_E5M2" = 0
|
|
56
|
+
COMMONDX_PRECISION_F8_E4M3 "COMMONDX_PRECISION_F8_E4M3" = 1
|
|
57
|
+
COMMONDX_PRECISION_BF16 "COMMONDX_PRECISION_BF16" = 2
|
|
58
|
+
COMMONDX_PRECISION_F16 "COMMONDX_PRECISION_F16" = 3
|
|
59
|
+
COMMONDX_PRECISION_TF32 "COMMONDX_PRECISION_TF32" = 4
|
|
60
|
+
COMMONDX_PRECISION_F32 "COMMONDX_PRECISION_F32" = 5
|
|
61
|
+
COMMONDX_PRECISION_F64 "COMMONDX_PRECISION_F64" = 6
|
|
62
|
+
COMMONDX_PRECISION_I8 "COMMONDX_PRECISION_I8" = 7
|
|
63
|
+
COMMONDX_PRECISION_I16 "COMMONDX_PRECISION_I16" = 8
|
|
64
|
+
COMMONDX_PRECISION_I32 "COMMONDX_PRECISION_I32" = 9
|
|
65
|
+
COMMONDX_PRECISION_I64 "COMMONDX_PRECISION_I64" = 10
|
|
66
|
+
COMMONDX_PRECISION_UI8 "COMMONDX_PRECISION_UI8" = 11
|
|
67
|
+
COMMONDX_PRECISION_UI16 "COMMONDX_PRECISION_UI16" = 12
|
|
68
|
+
COMMONDX_PRECISION_UI32 "COMMONDX_PRECISION_UI32" = 13
|
|
69
|
+
COMMONDX_PRECISION_UI64 "COMMONDX_PRECISION_UI64" = 14
|
|
70
|
+
|
|
71
|
+
ctypedef enum commondxArchModifier "commondxArchModifier":
|
|
72
|
+
COMMONDX_ARCH_MODIFIER_GENERIC "COMMONDX_ARCH_MODIFIER_GENERIC" = 0
|
|
73
|
+
COMMONDX_ARCH_MODIFIER_ARCH_SPECIFIC "COMMONDX_ARCH_MODIFIER_ARCH_SPECIFIC" = 1
|
|
74
|
+
COMMONDX_ARCH_MODIFIER_FAMILY_SPECIFIC "COMMONDX_ARCH_MODIFIER_FAMILY_SPECIFIC" = 2
|
|
75
|
+
|
|
76
|
+
ctypedef enum commondxOption "commondxOption":
|
|
77
|
+
COMMONDX_OPTION_SYMBOL_NAME "COMMONDX_OPTION_SYMBOL_NAME" = 0
|
|
78
|
+
COMMONDX_OPTION_TARGET_SM "COMMONDX_OPTION_TARGET_SM" = 1
|
|
79
|
+
COMMONDX_OPTION_CODE_CONTAINER "COMMONDX_OPTION_CODE_CONTAINER" = 2
|
|
80
|
+
COMMONDX_OPTION_CODE_ISA "COMMONDX_OPTION_CODE_ISA" = 3
|
|
81
|
+
COMMONDX_OPTION_EXTRA_NVRTC_ARGS "COMMONDX_OPTION_EXTRA_NVRTC_ARGS" = 4
|
|
82
|
+
COMMONDX_OPTION_TARGET_CODE "COMMONDX_OPTION_TARGET_CODE" = 5
|
|
83
|
+
|
|
84
|
+
ctypedef enum commondxExecution "commondxExecution":
|
|
85
|
+
COMMONDX_EXECUTION_THREAD "COMMONDX_EXECUTION_THREAD" = 0
|
|
86
|
+
COMMONDX_EXECUTION_BLOCK "COMMONDX_EXECUTION_BLOCK" = 1
|
|
87
|
+
|
|
88
|
+
ctypedef enum commondxCodeContainer "commondxCodeContainer":
|
|
89
|
+
COMMONDX_CODE_CONTAINER_LTOIR "COMMONDX_CODE_CONTAINER_LTOIR" = 0
|
|
90
|
+
COMMONDX_CODE_CONTAINER_FATBIN "COMMONDX_CODE_CONTAINER_FATBIN" = 1
|
|
91
|
+
COMMONDX_CODE_CONTAINER_PTX "COMMONDX_CODE_CONTAINER_PTX" = 2
|
|
92
|
+
|
|
93
|
+
ctypedef enum cublasdxApi "cublasdxApi":
|
|
94
|
+
CUBLASDX_API_SMEM "CUBLASDX_API_SMEM" = 0
|
|
95
|
+
CUBLASDX_API_SMEM_DYNAMIC_LD "CUBLASDX_API_SMEM_DYNAMIC_LD" = 1
|
|
96
|
+
CUBLASDX_API_TENSORS "CUBLASDX_API_TENSORS" = 2
|
|
97
|
+
|
|
98
|
+
ctypedef enum cublasdxDevicePipelineType "cublasdxDevicePipelineType":
|
|
99
|
+
CUBLASDX_DEVICE_PIPELINE_SUGGESTED "CUBLASDX_DEVICE_PIPELINE_SUGGESTED" = 0
|
|
100
|
+
|
|
101
|
+
ctypedef enum cublasdxTilePipelineType "cublasdxTilePipelineType":
|
|
102
|
+
CUBLASDX_TILE_PIPELINE_DEFAULT "CUBLASDX_TILE_PIPELINE_DEFAULT" = 0
|
|
103
|
+
|
|
104
|
+
ctypedef enum cublasdxBlockSizeStrategy "cublasdxBlockSizeStrategy":
|
|
105
|
+
CUBLASDX_BLOCK_SIZE_STRATEGY_HEURISTIC "CUBLASDX_BLOCK_SIZE_STRATEGY_HEURISTIC" = 0
|
|
106
|
+
CUBLASDX_BLOCK_SIZE_STRATEGY_FIXED "CUBLASDX_BLOCK_SIZE_STRATEGY_FIXED" = 1
|
|
107
|
+
|
|
108
|
+
ctypedef enum cublasdxType "cublasdxType":
|
|
109
|
+
CUBLASDX_TYPE_REAL "CUBLASDX_TYPE_REAL" = 0
|
|
110
|
+
CUBLASDX_TYPE_COMPLEX "CUBLASDX_TYPE_COMPLEX" = 1
|
|
111
|
+
|
|
112
|
+
ctypedef enum cublasdxMemorySpace "cublasdxMemorySpace":
|
|
113
|
+
CUBLASDX_MEMORY_SPACE_RMEM "CUBLASDX_MEMORY_SPACE_RMEM" = 0
|
|
114
|
+
CUBLASDX_MEMORY_SPACE_SMEM "CUBLASDX_MEMORY_SPACE_SMEM" = 1
|
|
115
|
+
CUBLASDX_MEMORY_SPACE_GMEM "CUBLASDX_MEMORY_SPACE_GMEM" = 2
|
|
116
|
+
CUBLASDX_MEMORY_SPACE_ANY "CUBLASDX_MEMORY_SPACE_ANY" = 3
|
|
117
|
+
|
|
118
|
+
ctypedef enum cublasdxTransposeMode "cublasdxTransposeMode":
|
|
119
|
+
CUBLASDX_TRANSPOSE_MODE_NON_TRANSPOSED "CUBLASDX_TRANSPOSE_MODE_NON_TRANSPOSED" = 0
|
|
120
|
+
CUBLASDX_TRANSPOSE_MODE_TRANSPOSED "CUBLASDX_TRANSPOSE_MODE_TRANSPOSED" = 1
|
|
121
|
+
CUBLASDX_TRANSPOSE_MODE_CONJ_TRANSPOSED "CUBLASDX_TRANSPOSE_MODE_CONJ_TRANSPOSED" = 2
|
|
122
|
+
|
|
123
|
+
ctypedef enum cublasdxArrangement "cublasdxArrangement":
|
|
124
|
+
CUBLASDX_ARRANGEMENT_COL_MAJOR "CUBLASDX_ARRANGEMENT_COL_MAJOR" = 0
|
|
125
|
+
CUBLASDX_ARRANGEMENT_ROW_MAJOR "CUBLASDX_ARRANGEMENT_ROW_MAJOR" = 1
|
|
126
|
+
|
|
127
|
+
ctypedef enum cublasdxFunction "cublasdxFunction":
|
|
128
|
+
CUBLASDX_FUNCTION_MM "CUBLASDX_FUNCTION_MM" = 0
|
|
129
|
+
|
|
130
|
+
ctypedef enum cublasdxOperatorType "cublasdxOperatorType":
|
|
131
|
+
CUBLASDX_OPERATOR_FUNCTION "CUBLASDX_OPERATOR_FUNCTION" = 0
|
|
132
|
+
CUBLASDX_OPERATOR_SIZE "CUBLASDX_OPERATOR_SIZE" = 1
|
|
133
|
+
CUBLASDX_OPERATOR_TYPE "CUBLASDX_OPERATOR_TYPE" = 2
|
|
134
|
+
CUBLASDX_OPERATOR_PRECISION "CUBLASDX_OPERATOR_PRECISION" = 3
|
|
135
|
+
CUBLASDX_OPERATOR_SM "CUBLASDX_OPERATOR_SM" = 4
|
|
136
|
+
CUBLASDX_OPERATOR_EXECUTION "CUBLASDX_OPERATOR_EXECUTION" = 5
|
|
137
|
+
CUBLASDX_OPERATOR_BLOCK_DIM "CUBLASDX_OPERATOR_BLOCK_DIM" = 6
|
|
138
|
+
CUBLASDX_OPERATOR_LEADING_DIMENSION "CUBLASDX_OPERATOR_LEADING_DIMENSION" = 7
|
|
139
|
+
CUBLASDX_OPERATOR_TRANSPOSE_MODE "CUBLASDX_OPERATOR_TRANSPOSE_MODE" = 8
|
|
140
|
+
CUBLASDX_OPERATOR_API "CUBLASDX_OPERATOR_API" = 9
|
|
141
|
+
CUBLASDX_OPERATOR_ARRANGEMENT "CUBLASDX_OPERATOR_ARRANGEMENT" = 10
|
|
142
|
+
CUBLASDX_OPERATOR_ALIGNMENT "CUBLASDX_OPERATOR_ALIGNMENT" = 11
|
|
143
|
+
CUBLASDX_OPERATOR_STATIC_BLOCK_DIM "CUBLASDX_OPERATOR_STATIC_BLOCK_DIM" = 12
|
|
144
|
+
CUBLASDX_OPERATOR_ENABLE_INPUT_STREAMING "CUBLASDX_OPERATOR_ENABLE_INPUT_STREAMING" = 13
|
|
145
|
+
CUBLASDX_OPERATOR_WITH_PIPELINE "CUBLASDX_OPERATOR_WITH_PIPELINE" = 14
|
|
146
|
+
|
|
147
|
+
ctypedef enum cublasdxTraitType "cublasdxTraitType":
|
|
148
|
+
CUBLASDX_TRAIT_VALUE_TYPE "CUBLASDX_TRAIT_VALUE_TYPE" = 0
|
|
149
|
+
CUBLASDX_TRAIT_SIZE "CUBLASDX_TRAIT_SIZE" = 1
|
|
150
|
+
CUBLASDX_TRAIT_BLOCK_SIZE "CUBLASDX_TRAIT_BLOCK_SIZE" = 2
|
|
151
|
+
CUBLASDX_TRAIT_BLOCK_DIM "CUBLASDX_TRAIT_BLOCK_DIM" = 3
|
|
152
|
+
CUBLASDX_TRAIT_LEADING_DIMENSION "CUBLASDX_TRAIT_LEADING_DIMENSION" = 4
|
|
153
|
+
CUBLASDX_TRAIT_SYMBOL_NAME "CUBLASDX_TRAIT_SYMBOL_NAME" = 5
|
|
154
|
+
CUBLASDX_TRAIT_ARRANGEMENT "CUBLASDX_TRAIT_ARRANGEMENT" = 6
|
|
155
|
+
CUBLASDX_TRAIT_ALIGNMENT "CUBLASDX_TRAIT_ALIGNMENT" = 7
|
|
156
|
+
CUBLASDX_TRAIT_SUGGESTED_LEADING_DIMENSION "CUBLASDX_TRAIT_SUGGESTED_LEADING_DIMENSION" = 8
|
|
157
|
+
CUBLASDX_TRAIT_SUGGESTED_BLOCK_DIM "CUBLASDX_TRAIT_SUGGESTED_BLOCK_DIM" = 9
|
|
158
|
+
CUBLASDX_TRAIT_MAX_THREADS_PER_BLOCK "CUBLASDX_TRAIT_MAX_THREADS_PER_BLOCK" = 10
|
|
159
|
+
|
|
160
|
+
ctypedef enum cublasdxTensorType "cublasdxTensorType":
|
|
161
|
+
CUBLASDX_TENSOR_SMEM_A "CUBLASDX_TENSOR_SMEM_A" = 0
|
|
162
|
+
CUBLASDX_TENSOR_SMEM_B "CUBLASDX_TENSOR_SMEM_B" = 1
|
|
163
|
+
CUBLASDX_TENSOR_SMEM_C "CUBLASDX_TENSOR_SMEM_C" = 2
|
|
164
|
+
CUBLASDX_TENSOR_SUGGESTED_SMEM_A "CUBLASDX_TENSOR_SUGGESTED_SMEM_A" = 3
|
|
165
|
+
CUBLASDX_TENSOR_SUGGESTED_SMEM_B "CUBLASDX_TENSOR_SUGGESTED_SMEM_B" = 4
|
|
166
|
+
CUBLASDX_TENSOR_SUGGESTED_SMEM_C "CUBLASDX_TENSOR_SUGGESTED_SMEM_C" = 5
|
|
167
|
+
CUBLASDX_TENSOR_SUGGESTED_RMEM_C "CUBLASDX_TENSOR_SUGGESTED_RMEM_C" = 6
|
|
168
|
+
CUBLASDX_TENSOR_GMEM_A "CUBLASDX_TENSOR_GMEM_A" = 7
|
|
169
|
+
CUBLASDX_TENSOR_GMEM_B "CUBLASDX_TENSOR_GMEM_B" = 8
|
|
170
|
+
CUBLASDX_TENSOR_GMEM_C "CUBLASDX_TENSOR_GMEM_C" = 9
|
|
171
|
+
CUBLASDX_TENSOR_SUGGESTED_ACCUMULATOR_C "CUBLASDX_TENSOR_SUGGESTED_ACCUMULATOR_C" = 10
|
|
172
|
+
CUBLASDX_TENSOR_RMEM_C "CUBLASDX_TENSOR_RMEM_C" = 11
|
|
173
|
+
CUBLASDX_TENSOR_ACCUMULATOR_C "CUBLASDX_TENSOR_ACCUMULATOR_C" = 12
|
|
174
|
+
|
|
175
|
+
ctypedef enum cublasdxTensorOption "cublasdxTensorOption":
|
|
176
|
+
CUBLASDX_TENSOR_OPTION_ALIGNMENT_BYTES "CUBLASDX_TENSOR_OPTION_ALIGNMENT_BYTES" = 0
|
|
177
|
+
|
|
178
|
+
ctypedef enum cublasdxTensorTrait "cublasdxTensorTrait":
|
|
179
|
+
CUBLASDX_TENSOR_TRAIT_STORAGE_BYTES "CUBLASDX_TENSOR_TRAIT_STORAGE_BYTES" = 0
|
|
180
|
+
CUBLASDX_TENSOR_TRAIT_ALIGNMENT_BYTES "CUBLASDX_TENSOR_TRAIT_ALIGNMENT_BYTES" = 1
|
|
181
|
+
CUBLASDX_TENSOR_TRAIT_UID "CUBLASDX_TENSOR_TRAIT_UID" = 2
|
|
182
|
+
CUBLASDX_TENSOR_TRAIT_OPAQUE_NAME "CUBLASDX_TENSOR_TRAIT_OPAQUE_NAME" = 4
|
|
183
|
+
CUBLASDX_TENSOR_TRAIT_LOGICAL_SIZE "CUBLASDX_TENSOR_TRAIT_LOGICAL_SIZE" = 5
|
|
184
|
+
CUBLASDX_TENSOR_TRAIT_MEMORY_SPACE "CUBLASDX_TENSOR_TRAIT_MEMORY_SPACE" = 6
|
|
185
|
+
|
|
186
|
+
ctypedef enum cublasdxPipelineTrait "cublasdxPipelineTrait":
|
|
187
|
+
CUBLASDX_PIPELINE_TRAIT_STORAGE_BYTES "CUBLASDX_PIPELINE_TRAIT_STORAGE_BYTES" = 0
|
|
188
|
+
CUBLASDX_PIPELINE_TRAIT_STORAGE_ALIGNMENT_BYTES "CUBLASDX_PIPELINE_TRAIT_STORAGE_ALIGNMENT_BYTES" = 1
|
|
189
|
+
CUBLASDX_PIPELINE_TRAIT_BUFFER_SIZE "CUBLASDX_PIPELINE_TRAIT_BUFFER_SIZE" = 2
|
|
190
|
+
CUBLASDX_PIPELINE_TRAIT_BUFFER_ALIGNMENT_BYTES "CUBLASDX_PIPELINE_TRAIT_BUFFER_ALIGNMENT_BYTES" = 3
|
|
191
|
+
CUBLASDX_PIPELINE_TRAIT_OPAQUE_NAME "CUBLASDX_PIPELINE_TRAIT_OPAQUE_NAME" = 4
|
|
192
|
+
CUBLASDX_PIPELINE_TRAIT_BLOCK_DIM "CUBLASDX_PIPELINE_TRAIT_BLOCK_DIM" = 5
|
|
193
|
+
|
|
194
|
+
ctypedef enum cublasdxDeviceFunctionTrait "cublasdxDeviceFunctionTrait":
|
|
195
|
+
CUBLASDX_DEVICE_FUNCTION_TRAIT_SYMBOL "CUBLASDX_DEVICE_FUNCTION_TRAIT_SYMBOL" = 1
|
|
196
|
+
|
|
197
|
+
ctypedef enum cublasdxDeviceFunctionOption "cublasdxDeviceFunctionOption":
|
|
198
|
+
CUBLASDX_DEVICE_FUNCTION_OPTION_SYMBOL_NAME "CUBLASDX_DEVICE_FUNCTION_OPTION_SYMBOL_NAME" = 0
|
|
199
|
+
CUBLASDX_DEVICE_FUNCTION_OPTION_COPY_ALIGNMENT "CUBLASDX_DEVICE_FUNCTION_OPTION_COPY_ALIGNMENT" = 1
|
|
200
|
+
CUBLASDX_DEVICE_FUNCTION_OPTION_CALLBACK "CUBLASDX_DEVICE_FUNCTION_OPTION_CALLBACK" = 2
|
|
201
|
+
CUBLASDX_DEVICE_FUNCTION_OPTION_NUM_THREADS "CUBLASDX_DEVICE_FUNCTION_OPTION_NUM_THREADS" = 3
|
|
202
|
+
|
|
203
|
+
ctypedef enum cublasdxDeviceFunctionType "cublasdxDeviceFunctionType":
|
|
204
|
+
CUBLASDX_DEVICE_FUNCTION_EXECUTE "CUBLASDX_DEVICE_FUNCTION_EXECUTE" = 0
|
|
205
|
+
CUBLASDX_DEVICE_FUNCTION_COPY "CUBLASDX_DEVICE_FUNCTION_COPY" = 1
|
|
206
|
+
CUBLASDX_DEVICE_FUNCTION_COPY_WAIT "CUBLASDX_DEVICE_FUNCTION_COPY_WAIT" = 2
|
|
207
|
+
CUBLASDX_DEVICE_FUNCTION_CLEAR "CUBLASDX_DEVICE_FUNCTION_CLEAR" = 3
|
|
208
|
+
CUBLASDX_DEVICE_FUNCTION_AXPBY "CUBLASDX_DEVICE_FUNCTION_AXPBY" = 4
|
|
209
|
+
CUBLASDX_DEVICE_FUNCTION_MAP_IDX2CRD "CUBLASDX_DEVICE_FUNCTION_MAP_IDX2CRD" = 5
|
|
210
|
+
CUBLASDX_DEVICE_FUNCTION_MAP_IDX2CRD_PARTITIONER "CUBLASDX_DEVICE_FUNCTION_MAP_IDX2CRD_PARTITIONER" = 6
|
|
211
|
+
CUBLASDX_DEVICE_FUNCTION_MAP_CRD2IDX "CUBLASDX_DEVICE_FUNCTION_MAP_CRD2IDX" = 7
|
|
212
|
+
CUBLASDX_DEVICE_FUNCTION_IS_THREAD_ACTIVE "CUBLASDX_DEVICE_FUNCTION_IS_THREAD_ACTIVE" = 8
|
|
213
|
+
CUBLASDX_DEVICE_FUNCTION_IS_PREDICATED "CUBLASDX_DEVICE_FUNCTION_IS_PREDICATED" = 9
|
|
214
|
+
CUBLASDX_DEVICE_FUNCTION_IS_INDEX_IN_BOUNDS "CUBLASDX_DEVICE_FUNCTION_IS_INDEX_IN_BOUNDS" = 10
|
|
215
|
+
CUBLASDX_DEVICE_FUNCTION_CREATE "CUBLASDX_DEVICE_FUNCTION_CREATE" = 11
|
|
216
|
+
CUBLASDX_DEVICE_FUNCTION_DESTROY "CUBLASDX_DEVICE_FUNCTION_DESTROY" = 12
|
|
217
|
+
CUBLASDX_DEVICE_FUNCTION_RESET "CUBLASDX_DEVICE_FUNCTION_RESET" = 13
|
|
218
|
+
CUBLASDX_DEVICE_FUNCTION_EPILOGUE "CUBLASDX_DEVICE_FUNCTION_EPILOGUE" = 14
|
|
219
|
+
|
|
220
|
+
ctypedef enum cufftdxApi "cufftdxApi":
|
|
221
|
+
CUFFTDX_API_LMEM "CUFFTDX_API_LMEM" = 0
|
|
222
|
+
CUFFTDX_API_SMEM "CUFFTDX_API_SMEM" = 1
|
|
223
|
+
|
|
224
|
+
ctypedef enum cufftdxType "cufftdxType":
|
|
225
|
+
CUFFTDX_TYPE_C2C "CUFFTDX_TYPE_C2C" = 0
|
|
226
|
+
CUFFTDX_TYPE_R2C "CUFFTDX_TYPE_R2C" = 1
|
|
227
|
+
CUFFTDX_TYPE_C2R "CUFFTDX_TYPE_C2R" = 2
|
|
228
|
+
|
|
229
|
+
ctypedef enum cufftdxDirection "cufftdxDirection":
|
|
230
|
+
CUFFTDX_DIRECTION_FORWARD "CUFFTDX_DIRECTION_FORWARD" = 0
|
|
231
|
+
CUFFTDX_DIRECTION_INVERSE "CUFFTDX_DIRECTION_INVERSE" = 1
|
|
232
|
+
|
|
233
|
+
ctypedef enum cufftdxComplexLayout "cufftdxComplexLayout":
|
|
234
|
+
CUFFTDX_COMPLEX_LAYOUT_NATURAL "CUFFTDX_COMPLEX_LAYOUT_NATURAL" = 0
|
|
235
|
+
CUFFTDX_COMPLEX_LAYOUT_PACKED "CUFFTDX_COMPLEX_LAYOUT_PACKED" = 1
|
|
236
|
+
CUFFTDX_COMPLEX_LAYOUT_FULL "CUFFTDX_COMPLEX_LAYOUT_FULL" = 2
|
|
237
|
+
|
|
238
|
+
ctypedef enum cufftdxRealMode "cufftdxRealMode":
|
|
239
|
+
CUFFTDX_REAL_MODE_NORMAL "CUFFTDX_REAL_MODE_NORMAL" = 0
|
|
240
|
+
CUFFTDX_REAL_MODE_FOLDED "CUFFTDX_REAL_MODE_FOLDED" = 1
|
|
241
|
+
|
|
242
|
+
ctypedef enum cufftdxCodeType "cufftdxCodeType":
|
|
243
|
+
CUFFTDX_CODE_TYPE_PTX "CUFFTDX_CODE_TYPE_PTX" = 0
|
|
244
|
+
CUFFTDX_CODE_TYPE_LTOIR "CUFFTDX_CODE_TYPE_LTOIR" = 1
|
|
245
|
+
|
|
246
|
+
ctypedef enum cufftdxOperatorType "cufftdxOperatorType":
|
|
247
|
+
CUFFTDX_OPERATOR_SIZE "CUFFTDX_OPERATOR_SIZE" = 0
|
|
248
|
+
CUFFTDX_OPERATOR_DIRECTION "CUFFTDX_OPERATOR_DIRECTION" = 1
|
|
249
|
+
CUFFTDX_OPERATOR_TYPE "CUFFTDX_OPERATOR_TYPE" = 2
|
|
250
|
+
CUFFTDX_OPERATOR_PRECISION "CUFFTDX_OPERATOR_PRECISION" = 3
|
|
251
|
+
CUFFTDX_OPERATOR_SM "CUFFTDX_OPERATOR_SM" = 4
|
|
252
|
+
CUFFTDX_OPERATOR_EXECUTION "CUFFTDX_OPERATOR_EXECUTION" = 5
|
|
253
|
+
CUFFTDX_OPERATOR_FFTS_PER_BLOCK "CUFFTDX_OPERATOR_FFTS_PER_BLOCK" = 6
|
|
254
|
+
CUFFTDX_OPERATOR_ELEMENTS_PER_THREAD "CUFFTDX_OPERATOR_ELEMENTS_PER_THREAD" = 7
|
|
255
|
+
CUFFTDX_OPERATOR_BLOCK_DIM "CUFFTDX_OPERATOR_BLOCK_DIM" = 8
|
|
256
|
+
CUFFTDX_OPERATOR_REAL_FFT_OPTIONS "CUFFTDX_OPERATOR_REAL_FFT_OPTIONS" = 9
|
|
257
|
+
CUFFTDX_OPERATOR_API "CUFFTDX_OPERATOR_API" = 10
|
|
258
|
+
CUFFTDX_OPERATOR_CODE_TYPE "CUFFTDX_OPERATOR_CODE_TYPE" = 11
|
|
259
|
+
|
|
260
|
+
ctypedef enum cufftdxKnobType "cufftdxKnobType":
|
|
261
|
+
CUFFTDX_KNOB_ELEMENTS_PER_THREAD "CUFFTDX_KNOB_ELEMENTS_PER_THREAD" = 0
|
|
262
|
+
CUFFTDX_KNOB_FFTS_PER_BLOCK "CUFFTDX_KNOB_FFTS_PER_BLOCK" = 1
|
|
263
|
+
|
|
264
|
+
ctypedef enum cufftdxTraitType "cufftdxTraitType":
|
|
265
|
+
CUFFTDX_TRAIT_VALUE_TYPE "CUFFTDX_TRAIT_VALUE_TYPE" = 0
|
|
266
|
+
CUFFTDX_TRAIT_INPUT_TYPE "CUFFTDX_TRAIT_INPUT_TYPE" = 1
|
|
267
|
+
CUFFTDX_TRAIT_OUTPUT_TYPE "CUFFTDX_TRAIT_OUTPUT_TYPE" = 2
|
|
268
|
+
CUFFTDX_TRAIT_IMPLICIT_TYPE_BATCHING "CUFFTDX_TRAIT_IMPLICIT_TYPE_BATCHING" = 3
|
|
269
|
+
CUFFTDX_TRAIT_ELEMENTS_PER_THREAD "CUFFTDX_TRAIT_ELEMENTS_PER_THREAD" = 4
|
|
270
|
+
CUFFTDX_TRAIT_STORAGE_SIZE "CUFFTDX_TRAIT_STORAGE_SIZE" = 5
|
|
271
|
+
CUFFTDX_TRAIT_STRIDE "CUFFTDX_TRAIT_STRIDE" = 6
|
|
272
|
+
CUFFTDX_TRAIT_BLOCK_DIM "CUFFTDX_TRAIT_BLOCK_DIM" = 7
|
|
273
|
+
CUFFTDX_TRAIT_SHARED_MEMORY_SIZE "CUFFTDX_TRAIT_SHARED_MEMORY_SIZE" = 8
|
|
274
|
+
CUFFTDX_TRAIT_FFTS_PER_BLOCK "CUFFTDX_TRAIT_FFTS_PER_BLOCK" = 9
|
|
275
|
+
CUFFTDX_TRAIT_SYMBOL_NAME "CUFFTDX_TRAIT_SYMBOL_NAME" = 10
|
|
276
|
+
CUFFTDX_TRAIT_INPUT_LENGTH "CUFFTDX_TRAIT_INPUT_LENGTH" = 11
|
|
277
|
+
CUFFTDX_TRAIT_OUTPUT_LENGTH "CUFFTDX_TRAIT_OUTPUT_LENGTH" = 12
|
|
278
|
+
CUFFTDX_TRAIT_INPUT_ELEMENTS_PER_THREAD "CUFFTDX_TRAIT_INPUT_ELEMENTS_PER_THREAD" = 13
|
|
279
|
+
CUFFTDX_TRAIT_OUTPUT_ELEMENTS_PER_THREAD "CUFFTDX_TRAIT_OUTPUT_ELEMENTS_PER_THREAD" = 14
|
|
280
|
+
CUFFTDX_TRAIT_SUGGESTED_FFTS_PER_BLOCK "CUFFTDX_TRAIT_SUGGESTED_FFTS_PER_BLOCK" = 15
|
|
281
|
+
|
|
282
|
+
ctypedef enum cusolverdxApi "cusolverdxApi":
|
|
283
|
+
CUSOLVERDX_API_SMEM "CUSOLVERDX_API_SMEM" = 0
|
|
284
|
+
CUSOLVERDX_API_SMEM_DYNAMIC_LD "CUSOLVERDX_API_SMEM_DYNAMIC_LD" = 1
|
|
285
|
+
|
|
286
|
+
ctypedef enum cusolverdxType "cusolverdxType":
|
|
287
|
+
CUSOLVERDX_TYPE_REAL "CUSOLVERDX_TYPE_REAL" = 0
|
|
288
|
+
CUSOLVERDX_TYPE_COMPLEX "CUSOLVERDX_TYPE_COMPLEX" = 1
|
|
289
|
+
|
|
290
|
+
ctypedef enum cusolverdxFunction "cusolverdxFunction":
|
|
291
|
+
CUSOLVERDX_FUNCTION_GETRF_NO_PIVOT "CUSOLVERDX_FUNCTION_GETRF_NO_PIVOT" = 0
|
|
292
|
+
CUSOLVERDX_FUNCTION_GETRS_NO_PIVOT "CUSOLVERDX_FUNCTION_GETRS_NO_PIVOT" = 1
|
|
293
|
+
CUSOLVERDX_FUNCTION_POTRF "CUSOLVERDX_FUNCTION_POTRF" = 2
|
|
294
|
+
CUSOLVERDX_FUNCTION_POTRS "CUSOLVERDX_FUNCTION_POTRS" = 3
|
|
295
|
+
CUSOLVERDX_FUNCTION_TRSM "CUSOLVERDX_FUNCTION_TRSM" = 4
|
|
296
|
+
CUSOLVERDX_FUNCTION_GETRF_PARTIAL_PIVOT "CUSOLVERDX_FUNCTION_GETRF_PARTIAL_PIVOT" = 5
|
|
297
|
+
CUSOLVERDX_FUNCTION_GETRS_PARTIAL_PIVOT "CUSOLVERDX_FUNCTION_GETRS_PARTIAL_PIVOT" = 6
|
|
298
|
+
CUSOLVERDX_FUNCTION_GEQRF "CUSOLVERDX_FUNCTION_GEQRF" = 7
|
|
299
|
+
CUSOLVERDX_FUNCTION_UNMQR "CUSOLVERDX_FUNCTION_UNMQR" = 8
|
|
300
|
+
CUSOLVERDX_FUNCTION_GELQF "CUSOLVERDX_FUNCTION_GELQF" = 9
|
|
301
|
+
CUSOLVERDX_FUNCTION_UNMLQ "CUSOLVERDX_FUNCTION_UNMLQ" = 10
|
|
302
|
+
CUSOLVERDX_FUNCTION_POSV "CUSOLVERDX_FUNCTION_POSV" = 11
|
|
303
|
+
CUSOLVERDX_FUNCTION_GESV_NO_PIVOT "CUSOLVERDX_FUNCTION_GESV_NO_PIVOT" = 12
|
|
304
|
+
CUSOLVERDX_FUNCTION_GESV_PARTIAL_PIVOT "CUSOLVERDX_FUNCTION_GESV_PARTIAL_PIVOT" = 13
|
|
305
|
+
CUSOLVERDX_FUNCTION_GELS "CUSOLVERDX_FUNCTION_GELS" = 14
|
|
306
|
+
CUSOLVERDX_FUNCTION_UNGQR "CUSOLVERDX_FUNCTION_UNGQR" = 15
|
|
307
|
+
CUSOLVERDX_FUNCTION_UNGLQ "CUSOLVERDX_FUNCTION_UNGLQ" = 16
|
|
308
|
+
CUSOLVERDX_FUNCTION_GTSV_NO_PIVOT "CUSOLVERDX_FUNCTION_GTSV_NO_PIVOT" = 17
|
|
309
|
+
CUSOLVERDX_FUNCTION_HTEV "CUSOLVERDX_FUNCTION_HTEV" = 18
|
|
310
|
+
CUSOLVERDX_FUNCTION_HEEV "CUSOLVERDX_FUNCTION_HEEV" = 19
|
|
311
|
+
|
|
312
|
+
ctypedef enum cusolverdxArrangement "cusolverdxArrangement":
|
|
313
|
+
CUSOLVERDX_ARRANGEMENT_COL_MAJOR "CUSOLVERDX_ARRANGEMENT_COL_MAJOR" = 0
|
|
314
|
+
CUSOLVERDX_ARRANGEMENT_ROW_MAJOR "CUSOLVERDX_ARRANGEMENT_ROW_MAJOR" = 1
|
|
315
|
+
|
|
316
|
+
ctypedef enum cusolverdxTransposeMode "cusolverdxTransposeMode":
|
|
317
|
+
CUSOLVERDX_TRANSPOSE_MODE_NON_TRANSPOSED "CUSOLVERDX_TRANSPOSE_MODE_NON_TRANSPOSED" = 0
|
|
318
|
+
CUSOLVERDX_TRANSPOSE_MODE_TRANSPOSED "CUSOLVERDX_TRANSPOSE_MODE_TRANSPOSED" = 1
|
|
319
|
+
CUSOLVERDX_TRANSPOSE_MODE_CONJ_TRANSPOSED "CUSOLVERDX_TRANSPOSE_MODE_CONJ_TRANSPOSED" = 2
|
|
320
|
+
|
|
321
|
+
ctypedef enum cusolverdxFillMode "cusolverdxFillMode":
|
|
322
|
+
CUSOLVERDX_FILL_MODE_UPPER "CUSOLVERDX_FILL_MODE_UPPER" = 0
|
|
323
|
+
CUSOLVERDX_FILL_MODE_LOWER "CUSOLVERDX_FILL_MODE_LOWER" = 1
|
|
324
|
+
|
|
325
|
+
ctypedef enum cusolverdxSide "cusolverdxSide":
|
|
326
|
+
CUSOLVERDX_SIDE_LEFT "CUSOLVERDX_SIDE_LEFT" = 0
|
|
327
|
+
CUSOLVERDX_SIDE_RIGHT "CUSOLVERDX_SIDE_RIGHT" = 1
|
|
328
|
+
|
|
329
|
+
ctypedef enum cusolverdxDiag "cusolverdxDiag":
|
|
330
|
+
CUSOLVERDX_DIAG_UNIT "CUSOLVERDX_DIAG_UNIT" = 0
|
|
331
|
+
CUSOLVERDX_DIAG_NON_UNIT "CUSOLVERDX_DIAG_NON_UNIT" = 1
|
|
332
|
+
|
|
333
|
+
ctypedef enum cusolverdxOperatorType "cusolverdxOperatorType":
|
|
334
|
+
CUSOLVERDX_OPERATOR_SIZE "CUSOLVERDX_OPERATOR_SIZE" = 0
|
|
335
|
+
CUSOLVERDX_OPERATOR_TYPE "CUSOLVERDX_OPERATOR_TYPE" = 1
|
|
336
|
+
CUSOLVERDX_OPERATOR_PRECISION "CUSOLVERDX_OPERATOR_PRECISION" = 2
|
|
337
|
+
CUSOLVERDX_OPERATOR_SM "CUSOLVERDX_OPERATOR_SM" = 3
|
|
338
|
+
CUSOLVERDX_OPERATOR_EXECUTION "CUSOLVERDX_OPERATOR_EXECUTION" = 4
|
|
339
|
+
CUSOLVERDX_OPERATOR_BLOCK_DIM "CUSOLVERDX_OPERATOR_BLOCK_DIM" = 5
|
|
340
|
+
CUSOLVERDX_OPERATOR_API "CUSOLVERDX_OPERATOR_API" = 6
|
|
341
|
+
CUSOLVERDX_OPERATOR_FUNCTION "CUSOLVERDX_OPERATOR_FUNCTION" = 7
|
|
342
|
+
CUSOLVERDX_OPERATOR_ARRANGEMENT "CUSOLVERDX_OPERATOR_ARRANGEMENT" = 8
|
|
343
|
+
CUSOLVERDX_OPERATOR_FILL_MODE "CUSOLVERDX_OPERATOR_FILL_MODE" = 9
|
|
344
|
+
CUSOLVERDX_OPERATOR_SIDE "CUSOLVERDX_OPERATOR_SIDE" = 10
|
|
345
|
+
CUSOLVERDX_OPERATOR_DIAG "CUSOLVERDX_OPERATOR_DIAG" = 11
|
|
346
|
+
CUSOLVERDX_OPERATOR_TRANSPOSE_MODE "CUSOLVERDX_OPERATOR_TRANSPOSE_MODE" = 12
|
|
347
|
+
CUSOLVERDX_OPERATOR_LEADING_DIMENSION "CUSOLVERDX_OPERATOR_LEADING_DIMENSION" = 13
|
|
348
|
+
CUSOLVERDX_OPERATOR_BATCHES_PER_BLOCK "CUSOLVERDX_OPERATOR_BATCHES_PER_BLOCK" = 14
|
|
349
|
+
CUSOLVERDX_OPERATOR_JOB "CUSOLVERDX_OPERATOR_JOB" = 15
|
|
350
|
+
|
|
351
|
+
ctypedef enum cusolverdxTraitType "cusolverdxTraitType":
|
|
352
|
+
CUSOLVERDX_TRAIT_SHARED_MEMORY_SIZE "CUSOLVERDX_TRAIT_SHARED_MEMORY_SIZE" = 1
|
|
353
|
+
CUSOLVERDX_TRAIT_SYMBOL_NAME "CUSOLVERDX_TRAIT_SYMBOL_NAME" = 2
|
|
354
|
+
CUSOLVERDX_TRAIT_BLOCK_DIM "CUSOLVERDX_TRAIT_BLOCK_DIM" = 3
|
|
355
|
+
CUSOLVERDX_TRAIT_SUGGESTED_BLOCK_DIM "CUSOLVERDX_TRAIT_SUGGESTED_BLOCK_DIM" = 5
|
|
356
|
+
CUSOLVERDX_TRAIT_SUGGESTED_BATCHES_PER_BLOCK "CUSOLVERDX_TRAIT_SUGGESTED_BATCHES_PER_BLOCK" = 6
|
|
357
|
+
CUSOLVERDX_TRAIT_ARRANGEMENT "CUSOLVERDX_TRAIT_ARRANGEMENT" = 7
|
|
358
|
+
CUSOLVERDX_TRAIT_BATCHES_PER_BLOCK "CUSOLVERDX_TRAIT_BATCHES_PER_BLOCK" = 8
|
|
359
|
+
CUSOLVERDX_TRAIT_TYPE "CUSOLVERDX_TRAIT_TYPE" = 9
|
|
360
|
+
CUSOLVERDX_TRAIT_LEADING_DIMENSION "CUSOLVERDX_TRAIT_LEADING_DIMENSION" = 10
|
|
361
|
+
CUSOLVERDX_TRAIT_VALUE_TYPE "CUSOLVERDX_TRAIT_VALUE_TYPE" = 11
|
|
362
|
+
CUSOLVERDX_TRAIT_WORKSPACE_SIZE "CUSOLVERDX_TRAIT_WORKSPACE_SIZE" = 12
|
|
363
|
+
CUSOLVERDX_TRAIT_SUGGESTED_LEADING_DIMENSION "CUSOLVERDX_TRAIT_SUGGESTED_LEADING_DIMENSION" = 4
|
|
364
|
+
|
|
365
|
+
ctypedef enum curanddxDistribution "curanddxDistribution":
|
|
366
|
+
CURANDDX_DISTRIBUTION_UNIFORM_BITS "CURANDDX_DISTRIBUTION_UNIFORM_BITS" = 0
|
|
367
|
+
CURANDDX_DISTRIBUTION_UNIFORM "CURANDDX_DISTRIBUTION_UNIFORM" = 1
|
|
368
|
+
CURANDDX_DISTRIBUTION_NORMAL "CURANDDX_DISTRIBUTION_NORMAL" = 2
|
|
369
|
+
CURANDDX_DISTRIBUTION_LOG_NORMAL "CURANDDX_DISTRIBUTION_LOG_NORMAL" = 3
|
|
370
|
+
CURANDDX_DISTRIBUTION_POISSON "CURANDDX_DISTRIBUTION_POISSON" = 4
|
|
371
|
+
|
|
372
|
+
ctypedef enum curanddxNormalMethod "curanddxNormalMethod":
|
|
373
|
+
CURANDDX_NORMAL_METHOD_ICDF "CURANDDX_NORMAL_METHOD_ICDF" = 0
|
|
374
|
+
CURANDDX_NORMAL_METHOD_BOX_MULLER "CURANDDX_NORMAL_METHOD_BOX_MULLER" = 1
|
|
375
|
+
|
|
376
|
+
ctypedef enum curanddxGenerateMethod "curanddxGenerateMethod":
|
|
377
|
+
CURANDDX_GENERATE_METHOD_SINGLE "CURANDDX_GENERATE_METHOD_SINGLE" = 0
|
|
378
|
+
CURANDDX_GENERATE_METHOD_PAIR "CURANDDX_GENERATE_METHOD_PAIR" = 1
|
|
379
|
+
CURANDDX_GENERATE_METHOD_QUAD "CURANDDX_GENERATE_METHOD_QUAD" = 2
|
|
380
|
+
|
|
381
|
+
ctypedef enum curanddxGenerator "curanddxGenerator":
|
|
382
|
+
CURANDDX_GENERATOR_XORWOW "CURANDDX_GENERATOR_XORWOW" = 0
|
|
383
|
+
CURANDDX_GENERATOR_MRG32K3A "CURANDDX_GENERATOR_MRG32K3A" = 1
|
|
384
|
+
CURANDDX_GENERATOR_PHILOX4_32 "CURANDDX_GENERATOR_PHILOX4_32" = 2
|
|
385
|
+
CURANDDX_GENERATOR_PCG "CURANDDX_GENERATOR_PCG" = 3
|
|
386
|
+
CURANDDX_GENERATOR_SOBOL32 "CURANDDX_GENERATOR_SOBOL32" = 4
|
|
387
|
+
CURANDDX_GENERATOR_SCRAMBLED_SOBOL32 "CURANDDX_GENERATOR_SCRAMBLED_SOBOL32" = 5
|
|
388
|
+
CURANDDX_GENERATOR_SOBOL64 "CURANDDX_GENERATOR_SOBOL64" = 6
|
|
389
|
+
CURANDDX_GENERATOR_SCRAMBLED_SOBOL64 "CURANDDX_GENERATOR_SCRAMBLED_SOBOL64" = 7
|
|
390
|
+
|
|
391
|
+
ctypedef enum curanddxOperatorType "curanddxOperatorType":
|
|
392
|
+
CURANDDX_OPERATOR_GENERATOR "CURANDDX_OPERATOR_GENERATOR" = 0
|
|
393
|
+
CURANDDX_OPERATOR_PHILOX_ROUNDS "CURANDDX_OPERATOR_PHILOX_ROUNDS" = 1
|
|
394
|
+
CURANDDX_OPERATOR_SM "CURANDDX_OPERATOR_SM" = 2
|
|
395
|
+
CURANDDX_OPERATOR_EXECUTION "CURANDDX_OPERATOR_EXECUTION" = 3
|
|
396
|
+
CURANDDX_OPERATOR_DISTRIBUTION "CURANDDX_OPERATOR_DISTRIBUTION" = 4
|
|
397
|
+
CURANDDX_OPERATOR_OUTPUT_TYPE "CURANDDX_OPERATOR_OUTPUT_TYPE" = 5
|
|
398
|
+
CURANDDX_OPERATOR_GENERATE_METHOD "CURANDDX_OPERATOR_GENERATE_METHOD" = 6
|
|
399
|
+
CURANDDX_OPERATOR_NORMAL_METHOD "CURANDDX_OPERATOR_NORMAL_METHOD" = 7
|
|
400
|
+
CURANDDX_OPERATOR_DEVICE_FUNCTIONS "CURANDDX_OPERATOR_DEVICE_FUNCTIONS" = 8
|
|
401
|
+
CURANDDX_OPERATOR_DISTRIBUTION_PARAMETERS "CURANDDX_OPERATOR_DISTRIBUTION_PARAMETERS" = 9
|
|
402
|
+
|
|
403
|
+
ctypedef enum curanddxDeviceFunctionType "curanddxDeviceFunctionType":
|
|
404
|
+
CURANDDX_DEVICE_FUNCTION_GENERATE "CURANDDX_DEVICE_FUNCTION_GENERATE" = (1 << 1)
|
|
405
|
+
CURANDDX_DEVICE_FUNCTION_INIT_STATE "CURANDDX_DEVICE_FUNCTION_INIT_STATE" = (1 << 2)
|
|
406
|
+
CURANDDX_DEVICE_FUNCTION_DESTROY_STATE "CURANDDX_DEVICE_FUNCTION_DESTROY_STATE" = (1 << 3)
|
|
407
|
+
CURANDDX_DEVICE_FUNCTION_SKIP_OFFSET "CURANDDX_DEVICE_FUNCTION_SKIP_OFFSET" = (1 << 4)
|
|
408
|
+
CURANDDX_DEVICE_FUNCTION_SKIP_SUBSEQUENCE "CURANDDX_DEVICE_FUNCTION_SKIP_SUBSEQUENCE" = (1 << 5)
|
|
409
|
+
CURANDDX_DEVICE_FUNCTION_SKIP_SEQUENCE "CURANDDX_DEVICE_FUNCTION_SKIP_SEQUENCE" = (1 << 6)
|
|
410
|
+
|
|
411
|
+
ctypedef enum curanddxTraitType "curanddxTraitType":
|
|
412
|
+
CURANDDX_TRAIT_SYMBOL_GENERATE_NAME "CURANDDX_TRAIT_SYMBOL_GENERATE_NAME" = 0
|
|
413
|
+
CURANDDX_TRAIT_SYMBOL_INIT_STATE_NAME "CURANDDX_TRAIT_SYMBOL_INIT_STATE_NAME" = 1
|
|
414
|
+
CURANDDX_TRAIT_SYMBOL_DESTROY_STATE_NAME "CURANDDX_TRAIT_SYMBOL_DESTROY_STATE_NAME" = 2
|
|
415
|
+
CURANDDX_TRAIT_SYMBOL_SKIP_OFFSET_NAME "CURANDDX_TRAIT_SYMBOL_SKIP_OFFSET_NAME" = 3
|
|
416
|
+
CURANDDX_TRAIT_SYMBOL_SKIP_SUBSEQUENCE_NAME "CURANDDX_TRAIT_SYMBOL_SKIP_SUBSEQUENCE_NAME" = 4
|
|
417
|
+
CURANDDX_TRAIT_SYMBOL_SKIP_SEQUENCE_NAME "CURANDDX_TRAIT_SYMBOL_SKIP_SEQUENCE_NAME" = 5
|
|
418
|
+
CURANDDX_TRAIT_STATE_SIZE "CURANDDX_TRAIT_STATE_SIZE" = 6
|
|
419
|
+
CURANDDX_TRAIT_STATE_ALIGNMENT "CURANDDX_TRAIT_STATE_ALIGNMENT" = 7
|
|
420
|
+
|
|
421
|
+
ctypedef enum commondxCodeType "commondxCodeType":
|
|
422
|
+
COMMONDX_CODE_TYPE_LTO "COMMONDX_CODE_TYPE_LTO" = 0
|
|
423
|
+
COMMONDX_CODE_TYPE_PTX "COMMONDX_CODE_TYPE_PTX" = 1
|
|
424
|
+
|
|
425
|
+
ctypedef enum cusolverdxJob "cusolverdxJob":
|
|
426
|
+
CUSOLVERDX_JOB_NO_VECTORS "CUSOLVERDX_JOB_NO_VECTORS" = 0
|
|
427
|
+
CUSOLVERDX_JOB_ALL_VECTORS "CUSOLVERDX_JOB_ALL_VECTORS" = 1
|
|
428
|
+
CUSOLVERDX_JOB_MULTIPLY_VECTORS "CUSOLVERDX_JOB_MULTIPLY_VECTORS" = 2
|
|
429
|
+
CUSOLVERDX_JOB_OVERWRITE_VECTORS "CUSOLVERDX_JOB_OVERWRITE_VECTORS" = 3
|
|
430
|
+
|
|
431
|
+
# types
|
|
432
|
+
ctypedef long long int commondxCode 'commondxCode'
|
|
433
|
+
ctypedef long long int cublasdxDescriptor 'cublasdxDescriptor'
|
|
434
|
+
ctypedef long long int cublasdxTensor 'cublasdxTensor'
|
|
435
|
+
ctypedef long long int cublasdxDeviceFunction 'cublasdxDeviceFunction'
|
|
436
|
+
ctypedef long long int cublasdxPipeline 'cublasdxPipeline'
|
|
437
|
+
ctypedef long long int cufftdxDescriptor 'cufftdxDescriptor'
|
|
438
|
+
ctypedef long long int cusolverdxDescriptor 'cusolverdxDescriptor'
|
|
439
|
+
ctypedef long long int curanddxDescriptor 'curanddxDescriptor'
|
|
440
|
+
|
|
441
|
+
|
|
442
|
+
###############################################################################
|
|
443
|
+
# Functions
|
|
444
|
+
###############################################################################
|
|
445
|
+
|
|
446
|
+
cdef commondxStatusType commondxCreateCode(commondxCode* code) except?_COMMONDXSTATUSTYPE_INTERNAL_LOADING_ERROR nogil
|
|
447
|
+
cdef commondxStatusType commondxSetCodeOptionInt64(commondxCode code, commondxOption option, long long int value) except?_COMMONDXSTATUSTYPE_INTERNAL_LOADING_ERROR nogil
|
|
448
|
+
cdef commondxStatusType commondxSetCodeOptionInt64s(commondxCode code, commondxOption option, size_t count, long long int* values) except?_COMMONDXSTATUSTYPE_INTERNAL_LOADING_ERROR nogil
|
|
449
|
+
cdef commondxStatusType commondxSetCodeOptionStr(commondxCode code, commondxOption option, const char* value) except?_COMMONDXSTATUSTYPE_INTERNAL_LOADING_ERROR nogil
|
|
450
|
+
cdef commondxStatusType commondxGetCodeOptionInt64(commondxCode code, commondxOption option, long long int* value) except?_COMMONDXSTATUSTYPE_INTERNAL_LOADING_ERROR nogil
|
|
451
|
+
cdef commondxStatusType commondxGetCodeOptionsInt64s(commondxCode code, commondxOption option, size_t size, long long int* array) except?_COMMONDXSTATUSTYPE_INTERNAL_LOADING_ERROR nogil
|
|
452
|
+
cdef commondxStatusType commondxGetCodeLTOIRSize(commondxCode code, size_t* size) except?_COMMONDXSTATUSTYPE_INTERNAL_LOADING_ERROR nogil
|
|
453
|
+
cdef commondxStatusType commondxGetCodeLTOIR(commondxCode code, size_t size, void* out) except?_COMMONDXSTATUSTYPE_INTERNAL_LOADING_ERROR nogil
|
|
454
|
+
cdef commondxStatusType commondxGetCodeNumLTOIRs(commondxCode code, size_t* size) except?_COMMONDXSTATUSTYPE_INTERNAL_LOADING_ERROR nogil
|
|
455
|
+
cdef commondxStatusType commondxGetCodeLTOIRSizes(commondxCode code, size_t size, size_t* out) except?_COMMONDXSTATUSTYPE_INTERNAL_LOADING_ERROR nogil
|
|
456
|
+
cdef commondxStatusType commondxGetCodeLTOIRs(commondxCode code, size_t size, void** out) except?_COMMONDXSTATUSTYPE_INTERNAL_LOADING_ERROR nogil
|
|
457
|
+
cdef commondxStatusType commondxDestroyCode(commondxCode code) except?_COMMONDXSTATUSTYPE_INTERNAL_LOADING_ERROR nogil
|
|
458
|
+
cdef const char* commondxStatusToStr(commondxStatusType status) except?NULL nogil
|
|
459
|
+
cdef commondxStatusType commondxGetLastErrorStrSize(size_t* size) except?_COMMONDXSTATUSTYPE_INTERNAL_LOADING_ERROR nogil
|
|
460
|
+
cdef commondxStatusType commondxGetLastErrorStr(commondxStatusType* code, size_t size, char* value) except?_COMMONDXSTATUSTYPE_INTERNAL_LOADING_ERROR nogil
|
|
461
|
+
cdef commondxStatusType mathdxGetVersion(int* version) except?_COMMONDXSTATUSTYPE_INTERNAL_LOADING_ERROR nogil
|
|
462
|
+
cdef commondxStatusType mathdxGetVersionEx(int* major, int* minor, int* patch) except?_COMMONDXSTATUSTYPE_INTERNAL_LOADING_ERROR nogil
|
|
463
|
+
cdef commondxStatusType cublasdxCreateDescriptor(cublasdxDescriptor* handle) except?_COMMONDXSTATUSTYPE_INTERNAL_LOADING_ERROR nogil
|
|
464
|
+
cdef commondxStatusType cublasdxSetOptionStr(cublasdxDescriptor handle, commondxOption option, const char* value) except?_COMMONDXSTATUSTYPE_INTERNAL_LOADING_ERROR nogil
|
|
465
|
+
cdef commondxStatusType cublasdxSetOperatorInt64(cublasdxDescriptor handle, cublasdxOperatorType op, long long int value) except?_COMMONDXSTATUSTYPE_INTERNAL_LOADING_ERROR nogil
|
|
466
|
+
cdef commondxStatusType cublasdxSetOperatorInt64s(cublasdxDescriptor handle, cublasdxOperatorType op, size_t count, const long long int* array) except?_COMMONDXSTATUSTYPE_INTERNAL_LOADING_ERROR nogil
|
|
467
|
+
cdef commondxStatusType cublasdxCreateTensor(cublasdxDescriptor handle, cublasdxTensorType tensor_type, cublasdxTensor* tensor) except?_COMMONDXSTATUSTYPE_INTERNAL_LOADING_ERROR nogil
|
|
468
|
+
cdef commondxStatusType cublasdxCreateTensorStrided(cublasdxMemorySpace memory_space, commondxValueType value_type, void* ptr, long long int rank, long long int* shape, long long int* stride, cublasdxTensor* tensor) except?_COMMONDXSTATUSTYPE_INTERNAL_LOADING_ERROR nogil
|
|
469
|
+
cdef commondxStatusType cublasdxMakeTensorLike(cublasdxTensor input, commondxValueType value_type, cublasdxTensor* output) except?_COMMONDXSTATUSTYPE_INTERNAL_LOADING_ERROR nogil
|
|
470
|
+
cdef commondxStatusType cublasdxDestroyTensor(cublasdxTensor tensor) except?_COMMONDXSTATUSTYPE_INTERNAL_LOADING_ERROR nogil
|
|
471
|
+
cdef commondxStatusType cublasdxDestroyPipeline(cublasdxPipeline pipeline) except?_COMMONDXSTATUSTYPE_INTERNAL_LOADING_ERROR nogil
|
|
472
|
+
cdef commondxStatusType cublasdxCreateDevicePipeline(cublasdxDescriptor handle, cublasdxDevicePipelineType device_pipeline_type, long long int pipeline_depth, cublasdxBlockSizeStrategy block_size_strategy, cublasdxTensor tensor_a, cublasdxTensor tensor_b, cublasdxPipeline* device_pipeline) except?_COMMONDXSTATUSTYPE_INTERNAL_LOADING_ERROR nogil
|
|
473
|
+
cdef commondxStatusType cublasdxCreateTilePipeline(cublasdxDescriptor handle, cublasdxTilePipelineType tile_pipeline_type, cublasdxPipeline device_pipeline, cublasdxPipeline* tile_pipeline) except?_COMMONDXSTATUSTYPE_INTERNAL_LOADING_ERROR nogil
|
|
474
|
+
cdef commondxStatusType cublasdxSetTensorOptionInt64(cublasdxTensor tensor, cublasdxTensorOption option, long long int value) except?_COMMONDXSTATUSTYPE_INTERNAL_LOADING_ERROR nogil
|
|
475
|
+
cdef commondxStatusType cublasdxFinalizeTensors(size_t count, const cublasdxTensor* array) except?_COMMONDXSTATUSTYPE_INTERNAL_LOADING_ERROR nogil
|
|
476
|
+
cdef commondxStatusType cublasdxFinalizePipelines(size_t count, const cublasdxPipeline* array) except?_COMMONDXSTATUSTYPE_INTERNAL_LOADING_ERROR nogil
|
|
477
|
+
cdef commondxStatusType cublasdxFinalize(size_t countTensors, const cublasdxTensor* tensors, size_t countPipelines, const cublasdxPipeline* pipelines) except?_COMMONDXSTATUSTYPE_INTERNAL_LOADING_ERROR nogil
|
|
478
|
+
cdef commondxStatusType cublasdxGetTensorTraitInt64(cublasdxTensor tensor, cublasdxTensorTrait trait, long long int* value) except?_COMMONDXSTATUSTYPE_INTERNAL_LOADING_ERROR nogil
|
|
479
|
+
cdef commondxStatusType cublasdxGetPipelineTraitInt64(cublasdxPipeline pipeline, cublasdxPipelineTrait trait, long long int* value) except?_COMMONDXSTATUSTYPE_INTERNAL_LOADING_ERROR nogil
|
|
480
|
+
cdef commondxStatusType cublasdxGetPipelineTraitInt64s(cublasdxPipeline pipeline, cublasdxPipelineTrait trait, size_t count, long long int* array) except?_COMMONDXSTATUSTYPE_INTERNAL_LOADING_ERROR nogil
|
|
481
|
+
cdef commondxStatusType cublasdxGetTensorTraitStrSize(cublasdxTensor tensor, cublasdxTensorTrait trait, size_t* size) except?_COMMONDXSTATUSTYPE_INTERNAL_LOADING_ERROR nogil
|
|
482
|
+
cdef commondxStatusType cublasdxGetPipelineTraitStrSize(cublasdxPipeline pipeline, cublasdxPipelineTrait trait, size_t* size) except?_COMMONDXSTATUSTYPE_INTERNAL_LOADING_ERROR nogil
|
|
483
|
+
cdef commondxStatusType cublasdxGetTensorTraitStr(cublasdxTensor tensor, cublasdxTensorTrait trait, size_t size, char* value) except?_COMMONDXSTATUSTYPE_INTERNAL_LOADING_ERROR nogil
|
|
484
|
+
cdef commondxStatusType cublasdxGetPipelineTraitStr(cublasdxPipeline pipeline, cublasdxPipelineTrait trait, size_t size, char* value) except?_COMMONDXSTATUSTYPE_INTERNAL_LOADING_ERROR nogil
|
|
485
|
+
cdef commondxStatusType cublasdxCreateDeviceFunction(cublasdxDescriptor handle, cublasdxDeviceFunctionType device_function_type, size_t count, const cublasdxTensor* array, cublasdxDeviceFunction* device_function) except?_COMMONDXSTATUSTYPE_INTERNAL_LOADING_ERROR nogil
|
|
486
|
+
cdef commondxStatusType cublasdxDestroyDeviceFunction(cublasdxDeviceFunction device_function) except?_COMMONDXSTATUSTYPE_INTERNAL_LOADING_ERROR nogil
|
|
487
|
+
cdef commondxStatusType cublasdxCreateDeviceFunctionWithPipelines(cublasdxDescriptor handle, cublasdxDeviceFunctionType device_function_type, size_t tensor_count, const cublasdxTensor* tensors, size_t pipeline_count, const cublasdxPipeline* pipelines, cublasdxDeviceFunction* device_function) except?_COMMONDXSTATUSTYPE_INTERNAL_LOADING_ERROR nogil
|
|
488
|
+
cdef commondxStatusType cublasdxFinalizeDeviceFunctions(commondxCode code, size_t count, const cublasdxDeviceFunction* array) except?_COMMONDXSTATUSTYPE_INTERNAL_LOADING_ERROR nogil
|
|
489
|
+
cdef commondxStatusType cublasdxGetDeviceFunctionTraitStrSize(cublasdxDeviceFunction device_function, cublasdxDeviceFunctionTrait trait, size_t* size) except?_COMMONDXSTATUSTYPE_INTERNAL_LOADING_ERROR nogil
|
|
490
|
+
cdef commondxStatusType cublasdxGetDeviceFunctionTraitStr(cublasdxDeviceFunction device_function, cublasdxDeviceFunctionTrait trait, size_t size, char* value) except?_COMMONDXSTATUSTYPE_INTERNAL_LOADING_ERROR nogil
|
|
491
|
+
cdef commondxStatusType cublasdxGetLTOIRSize(cublasdxDescriptor handle, size_t* lto_size) except?_COMMONDXSTATUSTYPE_INTERNAL_LOADING_ERROR nogil
|
|
492
|
+
cdef commondxStatusType cublasdxGetLTOIR(cublasdxDescriptor handle, size_t size, void* lto) except?_COMMONDXSTATUSTYPE_INTERNAL_LOADING_ERROR nogil
|
|
493
|
+
cdef commondxStatusType cublasdxGetTraitStrSize(cublasdxDescriptor handle, cublasdxTraitType trait, size_t* size) except?_COMMONDXSTATUSTYPE_INTERNAL_LOADING_ERROR nogil
|
|
494
|
+
cdef commondxStatusType cublasdxGetTraitStr(cublasdxDescriptor handle, cublasdxTraitType trait, size_t size, char* value) except?_COMMONDXSTATUSTYPE_INTERNAL_LOADING_ERROR nogil
|
|
495
|
+
cdef commondxStatusType cublasdxGetTraitInt64(cublasdxDescriptor handle, cublasdxTraitType trait, long long int* value) except?_COMMONDXSTATUSTYPE_INTERNAL_LOADING_ERROR nogil
|
|
496
|
+
cdef commondxStatusType cublasdxGetTraitInt64s(cublasdxDescriptor handle, cublasdxTraitType trait, size_t count, long long int* array) except?_COMMONDXSTATUSTYPE_INTERNAL_LOADING_ERROR nogil
|
|
497
|
+
cdef const char* cublasdxOperatorTypeToStr(cublasdxOperatorType op) except?NULL nogil
|
|
498
|
+
cdef const char* cublasdxTraitTypeToStr(cublasdxTraitType trait) except?NULL nogil
|
|
499
|
+
cdef commondxStatusType cublasdxFinalizeCode(commondxCode code, cublasdxDescriptor handle) except?_COMMONDXSTATUSTYPE_INTERNAL_LOADING_ERROR nogil
|
|
500
|
+
cdef commondxStatusType cublasdxDestroyDescriptor(cublasdxDescriptor handle) except?_COMMONDXSTATUSTYPE_INTERNAL_LOADING_ERROR nogil
|
|
501
|
+
cdef commondxStatusType cufftdxCreateDescriptor(cufftdxDescriptor* handle) except?_COMMONDXSTATUSTYPE_INTERNAL_LOADING_ERROR nogil
|
|
502
|
+
cdef commondxStatusType cufftdxSetOptionStr(cufftdxDescriptor handle, commondxOption opt, const char* value) except?_COMMONDXSTATUSTYPE_INTERNAL_LOADING_ERROR nogil
|
|
503
|
+
cdef commondxStatusType cufftdxGetKnobInt64Size(cufftdxDescriptor handle, size_t num_knobs, cufftdxKnobType* knobs_ptr, size_t* size) except?_COMMONDXSTATUSTYPE_INTERNAL_LOADING_ERROR nogil
|
|
504
|
+
cdef commondxStatusType cufftdxGetKnobInt64s(cufftdxDescriptor handle, size_t num_knobs, cufftdxKnobType* knobs_ptr, size_t size, long long int* values) except?_COMMONDXSTATUSTYPE_INTERNAL_LOADING_ERROR nogil
|
|
505
|
+
cdef commondxStatusType cufftdxSetOperatorInt64(cufftdxDescriptor handle, cufftdxOperatorType op, long long int value) except?_COMMONDXSTATUSTYPE_INTERNAL_LOADING_ERROR nogil
|
|
506
|
+
cdef commondxStatusType cufftdxSetOperatorInt64s(cufftdxDescriptor handle, cufftdxOperatorType op, size_t count, const long long int* array) except?_COMMONDXSTATUSTYPE_INTERNAL_LOADING_ERROR nogil
|
|
507
|
+
cdef commondxStatusType cufftdxGetLTOIRSize(cufftdxDescriptor handle, size_t* lto_size) except?_COMMONDXSTATUSTYPE_INTERNAL_LOADING_ERROR nogil
|
|
508
|
+
cdef commondxStatusType cufftdxGetLTOIR(cufftdxDescriptor handle, size_t size, void* lto) except?_COMMONDXSTATUSTYPE_INTERNAL_LOADING_ERROR nogil
|
|
509
|
+
cdef commondxStatusType cufftdxGetTraitStrSize(cufftdxDescriptor handle, cufftdxTraitType trait, size_t* size) except?_COMMONDXSTATUSTYPE_INTERNAL_LOADING_ERROR nogil
|
|
510
|
+
cdef commondxStatusType cufftdxGetTraitStr(cufftdxDescriptor handle, cufftdxTraitType trait, size_t size, char* value) except?_COMMONDXSTATUSTYPE_INTERNAL_LOADING_ERROR nogil
|
|
511
|
+
cdef commondxStatusType cufftdxGetTraitInt64(cufftdxDescriptor handle, cufftdxTraitType trait, long long int* value) except?_COMMONDXSTATUSTYPE_INTERNAL_LOADING_ERROR nogil
|
|
512
|
+
cdef commondxStatusType cufftdxGetTraitInt64s(cufftdxDescriptor handle, cufftdxTraitType trait, size_t count, long long int* array) except?_COMMONDXSTATUSTYPE_INTERNAL_LOADING_ERROR nogil
|
|
513
|
+
cdef commondxStatusType cufftdxGetTraitCommondxDataType(cufftdxDescriptor handle, cufftdxTraitType trait, commondxValueType* value) except?_COMMONDXSTATUSTYPE_INTERNAL_LOADING_ERROR nogil
|
|
514
|
+
cdef commondxStatusType cufftdxFinalizeCode(commondxCode code, cufftdxDescriptor handle) except?_COMMONDXSTATUSTYPE_INTERNAL_LOADING_ERROR nogil
|
|
515
|
+
cdef commondxStatusType cufftdxDestroyDescriptor(cufftdxDescriptor handle) except?_COMMONDXSTATUSTYPE_INTERNAL_LOADING_ERROR nogil
|
|
516
|
+
cdef const char* cufftdxOperatorTypeToStr(cufftdxOperatorType op) except?NULL nogil
|
|
517
|
+
cdef const char* cufftdxTraitTypeToStr(cufftdxTraitType op) except?NULL nogil
|
|
518
|
+
cdef commondxStatusType cusolverdxCreateDescriptor(cusolverdxDescriptor* handle) except?_COMMONDXSTATUSTYPE_INTERNAL_LOADING_ERROR nogil
|
|
519
|
+
cdef commondxStatusType cusolverdxSetOptionStr(cusolverdxDescriptor handle, commondxOption opt, const char* value) except?_COMMONDXSTATUSTYPE_INTERNAL_LOADING_ERROR nogil
|
|
520
|
+
cdef commondxStatusType cusolverdxSetOperatorInt64(cusolverdxDescriptor handle, cusolverdxOperatorType op, long long int value) except?_COMMONDXSTATUSTYPE_INTERNAL_LOADING_ERROR nogil
|
|
521
|
+
cdef commondxStatusType cusolverdxSetOperatorInt64s(cusolverdxDescriptor handle, cusolverdxOperatorType op, size_t count, const long long int* array) except?_COMMONDXSTATUSTYPE_INTERNAL_LOADING_ERROR nogil
|
|
522
|
+
cdef commondxStatusType cusolverdxGetLTOIRSize(cusolverdxDescriptor handle, size_t* lto_size) except?_COMMONDXSTATUSTYPE_INTERNAL_LOADING_ERROR nogil
|
|
523
|
+
cdef commondxStatusType cusolverdxGetLTOIR(cusolverdxDescriptor handle, size_t size, void* lto) except?_COMMONDXSTATUSTYPE_INTERNAL_LOADING_ERROR nogil
|
|
524
|
+
cdef commondxStatusType cusolverdxGetUniversalFATBINSize(cusolverdxDescriptor handle, size_t* fatbin_size) except?_COMMONDXSTATUSTYPE_INTERNAL_LOADING_ERROR nogil
|
|
525
|
+
cdef commondxStatusType cusolverdxGetUniversalFATBIN(cusolverdxDescriptor handle, size_t fatbin_size, void* fatbin) except?_COMMONDXSTATUSTYPE_INTERNAL_LOADING_ERROR nogil
|
|
526
|
+
cdef commondxStatusType cusolverdxGetTraitStrSize(cusolverdxDescriptor handle, cusolverdxTraitType trait, size_t* size) except?_COMMONDXSTATUSTYPE_INTERNAL_LOADING_ERROR nogil
|
|
527
|
+
cdef commondxStatusType cusolverdxGetTraitStr(cusolverdxDescriptor handle, cusolverdxTraitType trait, size_t size, char* value) except?_COMMONDXSTATUSTYPE_INTERNAL_LOADING_ERROR nogil
|
|
528
|
+
cdef commondxStatusType cusolverdxGetTraitInt64(cusolverdxDescriptor handle, cusolverdxTraitType trait, long long int* value) except?_COMMONDXSTATUSTYPE_INTERNAL_LOADING_ERROR nogil
|
|
529
|
+
cdef commondxStatusType cusolverdxGetTraitInt64s(cusolverdxDescriptor handle, cusolverdxTraitType trait, size_t count, long long int* values) except?_COMMONDXSTATUSTYPE_INTERNAL_LOADING_ERROR nogil
|
|
530
|
+
cdef commondxStatusType cusolverdxFinalizeCode(commondxCode code, cusolverdxDescriptor handle) except?_COMMONDXSTATUSTYPE_INTERNAL_LOADING_ERROR nogil
|
|
531
|
+
cdef commondxStatusType cusolverdxDestroyDescriptor(cusolverdxDescriptor handle) except?_COMMONDXSTATUSTYPE_INTERNAL_LOADING_ERROR nogil
|
|
532
|
+
cdef const char* cusolverdxOperatorTypeToStr(cusolverdxOperatorType op) except?NULL nogil
|
|
533
|
+
cdef const char* cusolverdxTraitTypeToStr(cusolverdxTraitType trait) except?NULL nogil
|
|
534
|
+
cdef commondxStatusType curanddxGetVersion(int* major, int* minor, int* patch) except?_COMMONDXSTATUSTYPE_INTERNAL_LOADING_ERROR nogil
|
|
535
|
+
cdef commondxStatusType curanddxCreateDescriptor(curanddxDescriptor* handle) except?_COMMONDXSTATUSTYPE_INTERNAL_LOADING_ERROR nogil
|
|
536
|
+
cdef commondxStatusType curanddxSetOptionStr(curanddxDescriptor handle, commondxOption opt, const char* value) except?_COMMONDXSTATUSTYPE_INTERNAL_LOADING_ERROR nogil
|
|
537
|
+
cdef commondxStatusType curanddxSetOperatorInt64(curanddxDescriptor handle, curanddxOperatorType op, long long int value) except?_COMMONDXSTATUSTYPE_INTERNAL_LOADING_ERROR nogil
|
|
538
|
+
cdef commondxStatusType curanddxSetOperatorDoubles(curanddxDescriptor handle, curanddxOperatorType op, size_t count, double* values) except?_COMMONDXSTATUSTYPE_INTERNAL_LOADING_ERROR nogil
|
|
539
|
+
cdef commondxStatusType curanddxGetTraitStrSize(curanddxDescriptor handle, curanddxTraitType trait, size_t* size) except?_COMMONDXSTATUSTYPE_INTERNAL_LOADING_ERROR nogil
|
|
540
|
+
cdef commondxStatusType curanddxGetTraitStr(curanddxDescriptor handle, curanddxTraitType trait, size_t size, char* value) except?_COMMONDXSTATUSTYPE_INTERNAL_LOADING_ERROR nogil
|
|
541
|
+
cdef commondxStatusType curanddxGetTraitInt64(curanddxDescriptor handle, curanddxTraitType trait, long long int* value) except?_COMMONDXSTATUSTYPE_INTERNAL_LOADING_ERROR nogil
|
|
542
|
+
cdef commondxStatusType curanddxFinalizeCode(commondxCode code, curanddxDescriptor handle) except?_COMMONDXSTATUSTYPE_INTERNAL_LOADING_ERROR nogil
|
|
543
|
+
cdef commondxStatusType curanddxDestroyDescriptor(curanddxDescriptor handle) except?_COMMONDXSTATUSTYPE_INTERNAL_LOADING_ERROR nogil
|
|
544
|
+
cdef const char* curanddxOperatorTypeToStr(curanddxOperatorType op) except?NULL nogil
|
|
545
|
+
cdef const char* curanddxDistributionToStr(curanddxDistribution dist) except?NULL nogil
|
|
546
|
+
cdef const char* curanddxGeneratorToStr(curanddxGenerator generator) except?NULL nogil
|
|
547
|
+
cdef const char* curanddxGenerateMethodToStr(curanddxGenerateMethod generate_method) except?NULL nogil
|
|
548
|
+
cdef const char* curanddxNormalMethodToStr(curanddxNormalMethod normal_method) except?NULL nogil
|
|
549
|
+
cdef const char* curanddxTraitTypeToStr(curanddxTraitType trait) except?NULL nogil
|
|
550
|
+
cdef commondxStatusType commondxGetCodePTXSize(commondxCode code, size_t* size) except?_COMMONDXSTATUSTYPE_INTERNAL_LOADING_ERROR nogil
|
|
551
|
+
cdef commondxStatusType commondxGetCodePTX(commondxCode code, size_t size, void* out) except?_COMMONDXSTATUSTYPE_INTERNAL_LOADING_ERROR nogil
|
|
552
|
+
cdef commondxStatusType cusolverdxGetTraitCommondxDataTypes(cusolverdxDescriptor handle, cusolverdxTraitType trait, size_t count, commondxValueType* array) except?_COMMONDXSTATUSTYPE_INTERNAL_LOADING_ERROR nogil
|