nvmath-python 1.0.0__cp314-cp314t-win_amd64.whl

nvmath/bindings/cusparseLt.pxd

@@ -0,0 +1,99 @@
+# Copyright (c) 2024-2026, NVIDIA CORPORATION & AFFILIATES. ALL RIGHTS RESERVED.
+#
+# SPDX-License-Identifier: Apache-2.0
+#
+# This code was automatically generated across versions from 0.7.1 to 0.8.1, generator version 0.3.1.dev1565+g7fa82f8eb. Do not modify it directly.
+
+cimport cython
+
+from libc.stdint cimport intptr_t, int64_t, uint8_t, uint32_t, int32_t
+
+from .cycusparseLt cimport *
+from .cycusparseLt cimport cusparseComputeType
+from .cycusparse cimport cusparseStatus_t, cusparseOrder_t, cusparseOperation_t
+from .cusparse cimport LibraryPropertyType
+
+
+###############################################################################
+# Types
+###############################################################################
+
+ctypedef cusparseLtHandle_t Handle
+ctypedef cusparseLtMatDescriptor_t MatDescriptor
+ctypedef cusparseLtMatmulDescriptor_t MatmulDescriptor
+ctypedef cusparseLtMatmulAlgSelection_t MatmulAlgSelection
+ctypedef cusparseLtMatmulPlan_t MatmulPlan
+
+ctypedef cudaStream_t Stream
+ctypedef cudaDataType DataType
+
+
+###############################################################################
+# Enum
+###############################################################################
+
+ctypedef cusparseLtSparsity_t _Sparsity
+ctypedef cusparseLtMatDescAttribute_t _MatDescAttribute
+ctypedef cusparseComputeType _ComputeType
+ctypedef cusparseLtMatmulDescAttribute_t _MatmulDescAttribute
+ctypedef cusparseLtMatmulMatrixScale_t _MatmulMatrixScale
+ctypedef cusparseLtMatmulAlg_t _MatmulAlg
+ctypedef cusparseLtMatmulAlgAttribute_t _MatmulAlgAttribute
+ctypedef cusparseLtSplitKMode_t _SplitKMode
+ctypedef cusparseLtPruneAlg_t _PruneAlg
+
+
+###############################################################################
+# Functions
+###############################################################################
+
+cpdef str get_error_name(cusparseStatus_t status)
+cpdef str get_error_string(cusparseStatus_t status)
+cpdef _init(intptr_t handle)
+cpdef _destroy(intptr_t handle)
+cpdef int get_version(intptr_t handle) except? -1
+cpdef int get_property(int property_type) except? -1
+cpdef _dense_descriptor_init(intptr_t handle, intptr_t mat_descr, int64_t rows, int64_t cols, int64_t ld, uint32_t alignment, int value_type, int order)
+cpdef _structured_descriptor_init(intptr_t handle, intptr_t mat_descr, int64_t rows, int64_t cols, int64_t ld, uint32_t alignment, int value_type, int order, int sparsity)
+cpdef _mat_descriptor_destroy(intptr_t mat_descr)
+cpdef get_mat_desc_attribute_dtype(int attr)
+cpdef mat_desc_set_attribute(intptr_t handle, intptr_t matmul_descr, int mat_attribute, intptr_t data, size_t data_size)
+cpdef mat_desc_get_attribute(intptr_t handle, intptr_t matmul_descr, int mat_attribute, intptr_t data, size_t data_size)
+cpdef _matmul_descriptor_init(intptr_t handle, intptr_t matmul_descr, int op_a, int op_b, intptr_t mat_a, intptr_t mat_b, intptr_t mat_c, intptr_t mat_d, int compute_type)
+cpdef get_matmul_desc_attribute_dtype(int attr)
+cpdef matmul_desc_set_attribute(intptr_t handle, intptr_t matmul_descr, int matmul_attribute, intptr_t data, size_t data_size)
+cpdef matmul_desc_get_attribute(intptr_t handle, intptr_t matmul_descr, int matmul_attribute, intptr_t data, size_t data_size)
+cpdef _matmul_alg_selection_init(intptr_t handle, intptr_t alg_selection, intptr_t matmul_descr, int alg)
+cpdef get_matmul_alg_attribute_dtype(int attr)
+cpdef matmul_alg_set_attribute(intptr_t handle, intptr_t alg_selection, int attribute, intptr_t data, size_t data_size)
+cpdef matmul_alg_get_attribute(intptr_t handle, intptr_t alg_selection, int attribute, intptr_t data, size_t data_size)
+cpdef size_t matmul_get_workspace(intptr_t handle, intptr_t plan) except? -1
+cpdef _matmul_plan_init(intptr_t handle, intptr_t plan, intptr_t matmul_descr, intptr_t alg_selection)
+cpdef _matmul_plan_destroy(intptr_t plan)
+cpdef matmul(intptr_t handle, intptr_t plan, intptr_t alpha, intptr_t d_a, intptr_t d_b, intptr_t beta, intptr_t d_c, intptr_t d_d, intptr_t workspace, intptr_t streams, int32_t num_streams)
+cpdef matmul_search(intptr_t handle, intptr_t plan, intptr_t alpha, intptr_t d_a, intptr_t d_b, intptr_t beta, intptr_t d_c, intptr_t d_d, intptr_t workspace, intptr_t streams, int32_t num_streams)
+cpdef sp_mma_prune(intptr_t handle, intptr_t matmul_descr, intptr_t d_in, intptr_t d_out, int prune_alg, intptr_t stream)
+cpdef sp_mma_prune_check(intptr_t handle, intptr_t matmul_descr, intptr_t d_in, intptr_t valid, intptr_t stream)
+cpdef sp_mma_prune2(intptr_t handle, intptr_t sparse_mat_descr, int is_sparse_a, int op, intptr_t d_in, intptr_t d_out, int prune_alg, intptr_t stream)
+cpdef sp_mma_prune_check2(intptr_t handle, intptr_t sparse_mat_descr, int is_sparse_a, int op, intptr_t d_in, intptr_t d_valid, intptr_t stream)
+cpdef tuple sp_mma_compressed_size(intptr_t handle, intptr_t plan)
+cpdef sp_mma_compress(intptr_t handle, intptr_t plan, intptr_t d_dense, intptr_t d_compressed, intptr_t d_compressed_buffer, intptr_t stream)
+cpdef tuple sp_mma_compressed_size2(intptr_t handle, intptr_t sparse_mat_descr)
+cpdef sp_mma_compress2(intptr_t handle, intptr_t sparse_mat_descr, int is_sparse_a, int op, intptr_t d_dense, intptr_t d_compressed, intptr_t d_compressed_buffer, intptr_t stream)
+cpdef _matmul_alg_selection_destroy(intptr_t alg_selection)
+
+cpdef intptr_t init() except *
+cpdef void destroy(intptr_t handle) except *
+
+cpdef intptr_t dense_descriptor_init(intptr_t handle, int64_t rows, int64_t cols, int64_t ld, uint32_t alignment, int value_type, int order) except *
+cpdef intptr_t structured_descriptor_init(intptr_t handle, int64_t rows, int64_t cols, int64_t ld, uint32_t alignment, int value_type, int order, int sparsity) except *
+cpdef void mat_descriptor_destroy(intptr_t mat_descr) except *
+
+cpdef intptr_t matmul_descriptor_init(intptr_t handle, int op_a, int op_b, intptr_t mat_a, intptr_t mat_b, intptr_t mat_c, intptr_t mat_d, int compute_type) except *
+cpdef void matmul_descriptor_destroy(intptr_t matmul_descr) except *
+
+cpdef intptr_t matmul_alg_selection_init(intptr_t handle, intptr_t matmul_descr, int alg) except *
+cpdef void matmul_alg_selection_destroy(intptr_t alg_selection) except *
+
+cpdef intptr_t matmul_plan_init(intptr_t handle, intptr_t matmul_descr, intptr_t alg_selection) except *
+cpdef void matmul_plan_destroy(intptr_t plan) except *

nvmath/bindings/cusparseLt.pyi

@@ -0,0 +1,252 @@
+# Copyright (c) 2026, NVIDIA CORPORATION & AFFILIATES. ALL RIGHTS RESERVED.
+#
+# SPDX-License-Identifier: Apache-2.0
+
+import _cython_3_2_5
+import enum
+from typing import Any, Callable, ClassVar
+
+__pyx_capi__: dict
+__test__: dict
+check_status: _cython_3_2_5.cython_function_or_method
+dense_descriptor_init: _cython_3_2_5.cython_function_or_method
+destroy: _cython_3_2_5.cython_function_or_method
+get_error_name: _cython_3_2_5.cython_function_or_method
+get_error_string: _cython_3_2_5.cython_function_or_method
+get_mat_desc_attribute_dtype: _cython_3_2_5.cython_function_or_method
+get_matmul_alg_attribute_dtype: _cython_3_2_5.cython_function_or_method
+get_matmul_desc_attribute_dtype: _cython_3_2_5.cython_function_or_method
+get_property: _cython_3_2_5.cython_function_or_method
+get_version: _cython_3_2_5.cython_function_or_method
+init: _cython_3_2_5.cython_function_or_method
+mat_desc_get_attribute: _cython_3_2_5.cython_function_or_method
+mat_desc_set_attribute: _cython_3_2_5.cython_function_or_method
+mat_descriptor_destroy: _cython_3_2_5.cython_function_or_method
+matmul: _cython_3_2_5.cython_function_or_method
+matmul_alg_get_attribute: _cython_3_2_5.cython_function_or_method
+matmul_alg_selection_destroy: _cython_3_2_5.cython_function_or_method
+matmul_alg_selection_init: _cython_3_2_5.cython_function_or_method
+matmul_alg_set_attribute: _cython_3_2_5.cython_function_or_method
+matmul_desc_get_attribute: _cython_3_2_5.cython_function_or_method
+matmul_desc_set_attribute: _cython_3_2_5.cython_function_or_method
+matmul_descriptor_destroy: _cython_3_2_5.cython_function_or_method
+matmul_descriptor_init: _cython_3_2_5.cython_function_or_method
+matmul_get_workspace: _cython_3_2_5.cython_function_or_method
+matmul_plan_destroy: _cython_3_2_5.cython_function_or_method
+matmul_plan_init: _cython_3_2_5.cython_function_or_method
+matmul_search: _cython_3_2_5.cython_function_or_method
+sp_mma_compress: _cython_3_2_5.cython_function_or_method
+sp_mma_compress2: _cython_3_2_5.cython_function_or_method
+sp_mma_compressed_size: _cython_3_2_5.cython_function_or_method
+sp_mma_compressed_size2: _cython_3_2_5.cython_function_or_method
+sp_mma_prune: _cython_3_2_5.cython_function_or_method
+sp_mma_prune2: _cython_3_2_5.cython_function_or_method
+sp_mma_prune_check: _cython_3_2_5.cython_function_or_method
+sp_mma_prune_check2: _cython_3_2_5.cython_function_or_method
+structured_descriptor_init: _cython_3_2_5.cython_function_or_method
+
+class ComputeType(enum.IntEnum):
+    """
+    See `cusparseComputeType`.
+    """
+    __new__: ClassVar[Callable] = ...
+    COMPUTE_16F: ClassVar[ComputeType] = ...
+    COMPUTE_32F: ClassVar[ComputeType] = ...
+    COMPUTE_32I: ClassVar[ComputeType] = ...
+    _generate_next_value_: ClassVar[Callable] = ...
+    _hashable_values_: ClassVar[list] = ...
+    _member_map_: ClassVar[dict] = ...
+    _member_names_: ClassVar[list] = ...
+    _member_type_: ClassVar[type[int]] = ...
+    _unhashable_values_: ClassVar[list] = ...
+    _unhashable_values_map_: ClassVar[dict] = ...
+    _use_args_: ClassVar[bool] = ...
+    _value2member_map_: ClassVar[dict] = ...
+    def __format__(self, *args, **kwargs) -> str:
+        """Convert to a string according to format_spec."""
+
+class MatDescAttribute(enum.IntEnum):
+    """
+    See `cusparseLtMatDescAttribute_t`.
+    """
+    __new__: ClassVar[Callable] = ...
+    BATCH_STRIDE: ClassVar[MatDescAttribute] = ...
+    NUM_BATCHES: ClassVar[MatDescAttribute] = ...
+    _generate_next_value_: ClassVar[Callable] = ...
+    _hashable_values_: ClassVar[list] = ...
+    _member_map_: ClassVar[dict] = ...
+    _member_names_: ClassVar[list] = ...
+    _member_type_: ClassVar[type[int]] = ...
+    _unhashable_values_: ClassVar[list] = ...
+    _unhashable_values_map_: ClassVar[dict] = ...
+    _use_args_: ClassVar[bool] = ...
+    _value2member_map_: ClassVar[dict] = ...
+    def __format__(self, *args, **kwargs) -> str:
+        """Convert to a string according to format_spec."""
+
+class MatmulAlg(enum.IntEnum):
+    """
+    See `cusparseLtMatmulAlg_t`.
+    """
+    __new__: ClassVar[Callable] = ...
+    DEFAULT: ClassVar[MatmulAlg] = ...
+    _generate_next_value_: ClassVar[Callable] = ...
+    _hashable_values_: ClassVar[list] = ...
+    _member_map_: ClassVar[dict] = ...
+    _member_names_: ClassVar[list] = ...
+    _member_type_: ClassVar[type[int]] = ...
+    _unhashable_values_: ClassVar[list] = ...
+    _unhashable_values_map_: ClassVar[dict] = ...
+    _use_args_: ClassVar[bool] = ...
+    _value2member_map_: ClassVar[dict] = ...
+    def __format__(self, *args, **kwargs) -> str:
+        """Convert to a string according to format_spec."""
+
+class MatmulAlgAttribute(enum.IntEnum):
+    """
+    See `cusparseLtMatmulAlgAttribute_t`.
+    """
+    __new__: ClassVar[Callable] = ...
+    CONFIG_ID: ClassVar[MatmulAlgAttribute] = ...
+    CONFIG_MAX_ID: ClassVar[MatmulAlgAttribute] = ...
+    SEARCH_ITERATIONS: ClassVar[MatmulAlgAttribute] = ...
+    SPLIT_K: ClassVar[MatmulAlgAttribute] = ...
+    SPLIT_K_BUFFERS: ClassVar[MatmulAlgAttribute] = ...
+    SPLIT_K_MODE: ClassVar[MatmulAlgAttribute] = ...
+    _generate_next_value_: ClassVar[Callable] = ...
+    _hashable_values_: ClassVar[list] = ...
+    _member_map_: ClassVar[dict] = ...
+    _member_names_: ClassVar[list] = ...
+    _member_type_: ClassVar[type[int]] = ...
+    _unhashable_values_: ClassVar[list] = ...
+    _unhashable_values_map_: ClassVar[dict] = ...
+    _use_args_: ClassVar[bool] = ...
+    _value2member_map_: ClassVar[dict] = ...
+    def __format__(self, *args, **kwargs) -> str:
+        """Convert to a string according to format_spec."""
+
+class MatmulDescAttribute(enum.IntEnum):
+    """
+    See `cusparseLtMatmulDescAttribute_t`.
+    """
+    __new__: ClassVar[Callable] = ...
+    ACTIVATION_GELU: ClassVar[MatmulDescAttribute] = ...
+    ACTIVATION_GELU_SCALING: ClassVar[MatmulDescAttribute] = ...
+    ACTIVATION_RELU: ClassVar[MatmulDescAttribute] = ...
+    ACTIVATION_RELU_THRESHOLD: ClassVar[MatmulDescAttribute] = ...
+    ACTIVATION_RELU_UPPERBOUND: ClassVar[MatmulDescAttribute] = ...
+    ALPHA_VECTOR_SCALING: ClassVar[MatmulDescAttribute] = ...
+    A_SCALE_MODE: ClassVar[MatmulDescAttribute] = ...
+    A_SCALE_POINTER: ClassVar[MatmulDescAttribute] = ...
+    BETA_VECTOR_SCALING: ClassVar[MatmulDescAttribute] = ...
+    BIAS_POINTER: ClassVar[MatmulDescAttribute] = ...
+    BIAS_STRIDE: ClassVar[MatmulDescAttribute] = ...
+    B_SCALE_MODE: ClassVar[MatmulDescAttribute] = ...
+    B_SCALE_POINTER: ClassVar[MatmulDescAttribute] = ...
+    C_SCALE_MODE: ClassVar[MatmulDescAttribute] = ...
+    C_SCALE_POINTER: ClassVar[MatmulDescAttribute] = ...
+    D_OUT_SCALE_MODE: ClassVar[MatmulDescAttribute] = ...
+    D_OUT_SCALE_POINTER: ClassVar[MatmulDescAttribute] = ...
+    D_SCALE_MODE: ClassVar[MatmulDescAttribute] = ...
+    D_SCALE_POINTER: ClassVar[MatmulDescAttribute] = ...
+    SPARSE_MAT_POINTER: ClassVar[MatmulDescAttribute] = ...
+    _generate_next_value_: ClassVar[Callable] = ...
+    _hashable_values_: ClassVar[list] = ...
+    _member_map_: ClassVar[dict] = ...
+    _member_names_: ClassVar[list] = ...
+    _member_type_: ClassVar[type[int]] = ...
+    _unhashable_values_: ClassVar[list] = ...
+    _unhashable_values_map_: ClassVar[dict] = ...
+    _use_args_: ClassVar[bool] = ...
+    _value2member_map_: ClassVar[dict] = ...
+    def __format__(self, *args, **kwargs) -> str:
+        """Convert to a string according to format_spec."""
+
+class MatmulMatrixScale(enum.IntEnum):
+    """
+    See `cusparseLtMatmulMatrixScale_t`.
+    """
+    __new__: ClassVar[Callable] = ...
+    NONE: ClassVar[MatmulMatrixScale] = ...
+    SCALAR_32F: ClassVar[MatmulMatrixScale] = ...
+    VEC32_UE4M3: ClassVar[MatmulMatrixScale] = ...
+    VEC64_UE8M0: ClassVar[MatmulMatrixScale] = ...
+    _generate_next_value_: ClassVar[Callable] = ...
+    _hashable_values_: ClassVar[list] = ...
+    _member_map_: ClassVar[dict] = ...
+    _member_names_: ClassVar[list] = ...
+    _member_type_: ClassVar[type[int]] = ...
+    _unhashable_values_: ClassVar[list] = ...
+    _unhashable_values_map_: ClassVar[dict] = ...
+    _use_args_: ClassVar[bool] = ...
+    _value2member_map_: ClassVar[dict] = ...
+    def __format__(self, *args, **kwargs) -> str:
+        """Convert to a string according to format_spec."""
+
+class PruneAlg(enum.IntEnum):
+    """
+    See `cusparseLtPruneAlg_t`.
+    """
+    __new__: ClassVar[Callable] = ...
+    SPMMA_STRIP: ClassVar[PruneAlg] = ...
+    SPMMA_TILE: ClassVar[PruneAlg] = ...
+    _generate_next_value_: ClassVar[Callable] = ...
+    _hashable_values_: ClassVar[list] = ...
+    _member_map_: ClassVar[dict] = ...
+    _member_names_: ClassVar[list] = ...
+    _member_type_: ClassVar[type[int]] = ...
+    _unhashable_values_: ClassVar[list] = ...
+    _unhashable_values_map_: ClassVar[dict] = ...
+    _use_args_: ClassVar[bool] = ...
+    _value2member_map_: ClassVar[dict] = ...
+    def __format__(self, *args, **kwargs) -> str:
+        """Convert to a string according to format_spec."""
+
+class Sparsity(enum.IntEnum):
+    """
+    See `cusparseLtSparsity_t`.
+    """
+    __new__: ClassVar[Callable] = ...
+    SPARSITY_50_PERCENT: ClassVar[Sparsity] = ...
+    _generate_next_value_: ClassVar[Callable] = ...
+    _hashable_values_: ClassVar[list] = ...
+    _member_map_: ClassVar[dict] = ...
+    _member_names_: ClassVar[list] = ...
+    _member_type_: ClassVar[type[int]] = ...
+    _unhashable_values_: ClassVar[list] = ...
+    _unhashable_values_map_: ClassVar[dict] = ...
+    _use_args_: ClassVar[bool] = ...
+    _value2member_map_: ClassVar[dict] = ...
+    def __format__(self, *args, **kwargs) -> str:
+        """Convert to a string according to format_spec."""
+
+class SplitKMode(enum.IntEnum):
+    """
+    See `cusparseLtSplitKMode_t`.
+    """
+    __new__: ClassVar[Callable] = ...
+    DATAPARALLEL: ClassVar[SplitKMode] = ...
+    HEURISTIC: ClassVar[SplitKMode] = ...
+    INVALID_MODE: ClassVar[SplitKMode] = ...
+    ONE_KERNEL: ClassVar[SplitKMode] = ...
+    SPLITK: ClassVar[SplitKMode] = ...
+    STREAMK: ClassVar[SplitKMode] = ...
+    TWO_KERNELS: ClassVar[SplitKMode] = ...
+    _generate_next_value_: ClassVar[Callable] = ...
+    _hashable_values_: ClassVar[list] = ...
+    _member_map_: ClassVar[dict] = ...
+    _member_names_: ClassVar[list] = ...
+    _member_type_: ClassVar[type[int]] = ...
+    _unhashable_values_: ClassVar[list] = ...
+    _unhashable_values_map_: ClassVar[dict] = ...
+    _use_args_: ClassVar[bool] = ...
+    _value2member_map_: ClassVar[dict] = ...
+    def __format__(self, *args, **kwargs) -> str:
+        """Convert to a string according to format_spec."""
+
+class cuSPARSELtError(Exception):
+    """cuSPARSELtError(status)"""
+    def __init__(self, status) -> Any:
+        """Initialize self.  See help(type(self)) for accurate signature."""
+    def __reduce__(self) -> Any:
+        """cuSPARSELtError.__reduce__(self)"""

nvmath/bindings/cutensor.pxd

@@ -0,0 +1,98 @@
+# Copyright (c) 2025-2026, NVIDIA CORPORATION & AFFILIATES. ALL RIGHTS RESERVED.
+#
+# SPDX-License-Identifier: Apache-2.0
+#
+# This code was automatically generated with version 2.5.0, generator version 0.3.1.dev1393+g0a20dc9d7. Do not modify it directly.
+
+cimport cython
+
+from libc.stdint cimport intptr_t
+
+from .cycutensor cimport *
+
+
+###############################################################################
+# Types
+###############################################################################
+
+ctypedef cutensorComputeDescriptor_t ComputeDescriptor
+ctypedef cutensorOperationDescriptor_t OperationDescriptor
+ctypedef cutensorPlan_t Plan
+ctypedef cutensorPlanPreference_t PlanPreference
+ctypedef cutensorHandle_t Handle
+ctypedef cutensorTensorDescriptor_t TensorDescriptor
+ctypedef cutensorBlockSparseTensorDescriptor_t BlockSparseTensorDescriptor
+ctypedef cutensorLoggerCallback_t LoggerCallback
+
+ctypedef cudaStream_t Stream
+ctypedef cudaDataType DataType
+ctypedef libraryPropertyType_t LibraryPropertyType
+
+
+###############################################################################
+# Enum
+###############################################################################
+
+ctypedef cutensorOperator_t _Operator
+ctypedef cutensorStatus_t _Status
+ctypedef cutensorAlgo_t _Algo
+ctypedef cutensorWorksizePreference_t _WorksizePreference
+ctypedef cutensorOperationDescriptorAttribute_t _OperationDescriptorAttribute
+ctypedef cutensorPlanPreferenceAttribute_t _PlanPreferenceAttribute
+ctypedef cutensorAutotuneMode_t _AutotuneMode
+ctypedef cutensorJitMode_t _JitMode
+ctypedef cutensorCacheMode_t _CacheMode
+ctypedef cutensorPlanAttribute_t _PlanAttribute
+
+
+###############################################################################
+# Functions
+###############################################################################
+
+cpdef intptr_t create() except? 0
+cpdef destroy(intptr_t handle)
+cpdef handle_resize_plan_cache(intptr_t handle, uint32_t num_entries)
+cpdef handle_write_plan_cache_to_file(intptr_t handle, filename)
+cpdef uint32_t handle_read_plan_cache_from_file(intptr_t handle, filename) except? -1
+cpdef write_kernel_cache_to_file(intptr_t handle, filename)
+cpdef read_kernel_cache_from_file(intptr_t handle, filename)
+cpdef intptr_t create_tensor_descriptor(intptr_t handle, uint32_t num_modes, extent, stride, int data_type, uint32_t alignment_requirement) except? 0
+cpdef destroy_tensor_descriptor(intptr_t desc)
+cpdef intptr_t create_elementwise_trinary(intptr_t handle, intptr_t desc_a, mode_a, int op_a, intptr_t desc_b, mode_b, int op_b, intptr_t desc_c, mode_c, int op_c, intptr_t desc_d, mode_d, int op_ab, int op_abc, intptr_t desc_compute) except? 0
+cpdef elementwise_trinary_execute(intptr_t handle, intptr_t plan, intptr_t alpha, intptr_t a, intptr_t beta, intptr_t b, intptr_t gamma, intptr_t c, intptr_t d, intptr_t stream)
+cpdef intptr_t create_elementwise_binary(intptr_t handle, intptr_t desc_a, mode_a, int op_a, intptr_t desc_c, mode_c, int op_c, intptr_t desc_d, mode_d, int op_ac, intptr_t desc_compute) except? 0
+cpdef elementwise_binary_execute(intptr_t handle, intptr_t plan, intptr_t alpha, intptr_t a, intptr_t gamma, intptr_t c, intptr_t d, intptr_t stream)
+cpdef intptr_t create_permutation(intptr_t handle, intptr_t desc_a, mode_a, int op_a, intptr_t desc_b, mode_b, intptr_t desc_compute) except? 0
+cpdef permute(intptr_t handle, intptr_t plan, intptr_t alpha, intptr_t a, intptr_t b, intptr_t stream)
+cpdef intptr_t create_contraction(intptr_t handle, intptr_t desc_a, mode_a, int op_a, intptr_t desc_b, mode_b, int op_b, intptr_t desc_c, mode_c, int op_c, intptr_t desc_d, mode_d, intptr_t desc_compute) except? 0
+cpdef destroy_operation_descriptor(intptr_t desc)
+cpdef get_operation_descriptor_attribute_dtype(int attr)
+cpdef operation_descriptor_set_attribute(intptr_t handle, intptr_t desc, int attr, intptr_t buf, size_t size_in_bytes)
+cpdef operation_descriptor_get_attribute(intptr_t handle, intptr_t desc, int attr, intptr_t buf, size_t size_in_bytes)
+cpdef intptr_t create_plan_preference(intptr_t handle, int algo, int jit_mode) except? 0
+cpdef destroy_plan_preference(intptr_t pref)
+cpdef get_plan_preference_attribute_dtype(int attr)
+cpdef plan_preference_get_attribute(intptr_t handle, intptr_t pref, int attr, intptr_t buf, size_t size_in_bytes)
+cpdef plan_preference_set_attribute(intptr_t handle, intptr_t pref, int attr, intptr_t buf, size_t size_in_bytes)
+cpdef get_plan_attribute_dtype(int attr)
+cpdef plan_get_attribute(intptr_t handle, intptr_t plan, int attr, intptr_t buf, size_t size_in_bytes)
+cpdef uint64_t estimate_workspace_size(intptr_t handle, intptr_t desc, intptr_t plan_pref, int workspace_pref) except? -1
+cpdef intptr_t create_plan(intptr_t handle, intptr_t desc, intptr_t pref, uint64_t workspace_size_limit) except? 0
+cpdef destroy_plan(intptr_t plan)
+cpdef contract(intptr_t handle, intptr_t plan, intptr_t alpha, intptr_t a, intptr_t b, intptr_t beta, intptr_t c, intptr_t d, intptr_t workspace, uint64_t workspace_size, intptr_t stream)
+cpdef intptr_t create_reduction(intptr_t handle, intptr_t desc_a, mode_a, int op_a, intptr_t desc_c, mode_c, int op_c, intptr_t desc_d, mode_d, int op_reduce, intptr_t desc_compute) except? 0
+cpdef reduce(intptr_t handle, intptr_t plan, intptr_t alpha, intptr_t a, intptr_t beta, intptr_t c, intptr_t d, intptr_t workspace, uint64_t workspace_size, intptr_t stream)
+cpdef intptr_t create_contraction_trinary(intptr_t handle, intptr_t desc_a, mode_a, int op_a, intptr_t desc_b, mode_b, int op_b, intptr_t desc_c, mode_c, int op_c, intptr_t desc_d, mode_d, int op_d, intptr_t desc_e, mode_e, intptr_t desc_compute) except? 0
+cpdef contract_trinary(intptr_t handle, intptr_t plan, intptr_t alpha, intptr_t a, intptr_t b, intptr_t c, intptr_t beta, intptr_t d, intptr_t e, intptr_t workspace, uint64_t workspace_size, intptr_t stream)
+cpdef intptr_t create_block_sparse_tensor_descriptor(intptr_t handle, uint32_t num_modes, uint64_t num_non_zero_blocks, num_sections_per_mode, extent, non_zero_coordinates, stride, int data_type) except? 0
+cpdef destroy_block_sparse_tensor_descriptor(intptr_t desc)
+cpdef intptr_t create_block_sparse_contraction(intptr_t handle, intptr_t desc_a, mode_a, int op_a, intptr_t desc_b, mode_b, int op_b, intptr_t desc_c, mode_c, int op_c, intptr_t desc_d, mode_d, intptr_t desc_compute) except? 0
+cpdef block_sparse_contract(intptr_t handle, intptr_t plan, intptr_t alpha, a, b, intptr_t beta, c, d, intptr_t workspace, uint64_t workspace_size, intptr_t stream)
+cpdef str get_error_string(int error)
+cpdef size_t get_version() except? 0
+cpdef size_t get_cudart_version() except? 0
+cpdef logger_set_file(intptr_t file)
+cpdef logger_open_file(log_file)
+cpdef logger_set_level(int32_t level)
+cpdef logger_set_mask(int32_t mask)
+cpdef logger_force_disable()