nvmath-python 1.0.0__cp314-cp314t-win_amd64.whl

nvmath/bindings/cudss.pyi

@@ -0,0 +1,443 @@
+# 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
+config_create: _cython_3_2_5.cython_function_or_method
+config_destroy: _cython_3_2_5.cython_function_or_method
+config_get: _cython_3_2_5.cython_function_or_method
+config_set: _cython_3_2_5.cython_function_or_method
+create: _cython_3_2_5.cython_function_or_method
+create_mg: _cython_3_2_5.cython_function_or_method
+data_create: _cython_3_2_5.cython_function_or_method
+data_destroy: _cython_3_2_5.cython_function_or_method
+data_get: _cython_3_2_5.cython_function_or_method
+data_set: _cython_3_2_5.cython_function_or_method
+destroy: _cython_3_2_5.cython_function_or_method
+execute: _cython_3_2_5.cython_function_or_method
+get_config_param_dtype: _cython_3_2_5.cython_function_or_method
+get_data_param_dtype: _cython_3_2_5.cython_function_or_method
+get_device_mem_handler: _cython_3_2_5.cython_function_or_method
+get_property: _cython_3_2_5.cython_function_or_method
+logger_force_disable: _cython_3_2_5.cython_function_or_method
+logger_open_file: _cython_3_2_5.cython_function_or_method
+logger_set_level: _cython_3_2_5.cython_function_or_method
+logger_set_mask: _cython_3_2_5.cython_function_or_method
+matrix_create_batch_csr: _cython_3_2_5.cython_function_or_method
+matrix_create_batch_dn: _cython_3_2_5.cython_function_or_method
+matrix_create_csr: _cython_3_2_5.cython_function_or_method
+matrix_create_dn: _cython_3_2_5.cython_function_or_method
+matrix_destroy: _cython_3_2_5.cython_function_or_method
+matrix_get_batch_csr: _cython_3_2_5.cython_function_or_method
+matrix_get_batch_dn: _cython_3_2_5.cython_function_or_method
+matrix_get_csr: _cython_3_2_5.cython_function_or_method
+matrix_get_distribution_row1d: _cython_3_2_5.cython_function_or_method
+matrix_get_dn: _cython_3_2_5.cython_function_or_method
+matrix_get_format: _cython_3_2_5.cython_function_or_method
+matrix_set_batch_csr_pointers: _cython_3_2_5.cython_function_or_method
+matrix_set_batch_values: _cython_3_2_5.cython_function_or_method
+matrix_set_csr_pointers: _cython_3_2_5.cython_function_or_method
+matrix_set_distribution_row1d: _cython_3_2_5.cython_function_or_method
+matrix_set_values: _cython_3_2_5.cython_function_or_method
+set_async_workspace_allocator: _cython_3_2_5.cython_function_or_method
+set_comm_layer: _cython_3_2_5.cython_function_or_method
+set_device_mem_handler: _cython_3_2_5.cython_function_or_method
+set_mg_streams: _cython_3_2_5.cython_function_or_method
+set_stream: _cython_3_2_5.cython_function_or_method
+set_threading_layer: _cython_3_2_5.cython_function_or_method
+
+class ConfigParam(enum.IntEnum):
+    """
+    See `cudssConfigParam_t`.
+    """
+    __new__: ClassVar[Callable] = ...
+    DETERMINISTIC_MODE: ClassVar[ConfigParam] = ...
+    DEVICE_COUNT: ClassVar[ConfigParam] = ...
+    DEVICE_INDICES: ClassVar[ConfigParam] = ...
+    FACTORIZATION_ALG: ClassVar[ConfigParam] = ...
+    HOST_NTHREADS: ClassVar[ConfigParam] = ...
+    HYBRID_DEVICE_MEMORY_LIMIT: ClassVar[ConfigParam] = ...
+    HYBRID_EXECUTE_MODE: ClassVar[ConfigParam] = ...
+    HYBRID_MEMORY_MODE: ClassVar[ConfigParam] = ...
+    IR_N_STEPS: ClassVar[ConfigParam] = ...
+    IR_TOL: ClassVar[ConfigParam] = ...
+    MATCHING_ALG: ClassVar[ConfigParam] = ...
+    MAX_LU_NNZ: ClassVar[ConfigParam] = ...
+    ND_NLEVELS: ClassVar[ConfigParam] = ...
+    ND_UBFACTOR: ClassVar[ConfigParam] = ...
+    PIVOT_EPSILON: ClassVar[ConfigParam] = ...
+    PIVOT_EPSILON_ALG: ClassVar[ConfigParam] = ...
+    PIVOT_THRESHOLD: ClassVar[ConfigParam] = ...
+    PIVOT_TYPE: ClassVar[ConfigParam] = ...
+    REORDERING_ALG: ClassVar[ConfigParam] = ...
+    SCHUR_MODE: ClassVar[ConfigParam] = ...
+    SOLVE_ALG: ClassVar[ConfigParam] = ...
+    SOLVE_MODE: ClassVar[ConfigParam] = ...
+    UBATCH_INDEX: ClassVar[ConfigParam] = ...
+    UBATCH_SIZE: ClassVar[ConfigParam] = ...
+    USE_CUDA_REGISTER_MEMORY: ClassVar[ConfigParam] = ...
+    USE_SUPERPANELS: ClassVar[ConfigParam] = ...
+    _generate_next_value_: ClassVar[Callable] = ...
+    _member_map_: ClassVar[dict] = ...
+    _member_names_: ClassVar[list] = ...
+    _member_type_: ClassVar[type[int]] = ...
+    _unhashable_values_: ClassVar[list] = ...
+    _use_args_: ClassVar[bool] = ...
+    _value2member_map_: ClassVar[dict] = ...
+    def __format__(self, *args, **kwargs) -> str:
+        """Convert to a string according to format_spec."""
+
+class DataParam(enum.IntEnum):
+    """
+    See `cudssDataParam_t`.
+    """
+    __new__: ClassVar[Callable] = ...
+    COMM_DEVICE: ClassVar[DataParam] = ...
+    COMM_HOST: ClassVar[DataParam] = ...
+    DIAG: ClassVar[DataParam] = ...
+    FLOPS: ClassVar[DataParam] = ...
+    HYBRID_DEVICE_MEMORY_MIN: ClassVar[DataParam] = ...
+    INERTIA: ClassVar[DataParam] = ...
+    INFO: ClassVar[DataParam] = ...
+    IR_N_STEPS: ClassVar[DataParam] = ...
+    LU_NNZ: ClassVar[DataParam] = ...
+    MEMORY_ESTIMATES: ClassVar[DataParam] = ...
+    ND_PARTITION_TREE: ClassVar[DataParam] = ...
+    NPIVOTS: ClassVar[DataParam] = ...
+    NSUPERPANELS: ClassVar[DataParam] = ...
+    PERM_COL: ClassVar[DataParam] = ...
+    PERM_MATCHING: ClassVar[DataParam] = ...
+    PERM_REORDER_COL: ClassVar[DataParam] = ...
+    PERM_REORDER_ROW: ClassVar[DataParam] = ...
+    PERM_ROW: ClassVar[DataParam] = ...
+    SCALE_COL: ClassVar[DataParam] = ...
+    SCALE_ROW: ClassVar[DataParam] = ...
+    SCHUR_MATRIX: ClassVar[DataParam] = ...
+    SCHUR_SHAPE: ClassVar[DataParam] = ...
+    UBATCH_MASK: ClassVar[DataParam] = ...
+    USER_HOST_INTERRUPT: ClassVar[DataParam] = ...
+    USER_ND_PARTITION_TREE: ClassVar[DataParam] = ...
+    USER_PERM: ClassVar[DataParam] = ...
+    USER_SCHUR_INDICES: ClassVar[DataParam] = ...
+    _generate_next_value_: ClassVar[Callable] = ...
+    _member_map_: ClassVar[dict] = ...
+    _member_names_: ClassVar[list] = ...
+    _member_type_: ClassVar[type[int]] = ...
+    _unhashable_values_: ClassVar[list] = ...
+    _use_args_: ClassVar[bool] = ...
+    _value2member_map_: ClassVar[dict] = ...
+    def __format__(self, *args, **kwargs) -> str:
+        """Convert to a string according to format_spec."""
+
+class DataType(enum.IntEnum):
+    """
+    See `cudssDataType_t`.
+    """
+    __new__: ClassVar[Callable] = ...
+    C_32F: ClassVar[DataType] = ...
+    C_64F: ClassVar[DataType] = ...
+    R_32F: ClassVar[DataType] = ...
+    R_32I: ClassVar[DataType] = ...
+    R_64F: ClassVar[DataType] = ...
+    R_64F_64F: ClassVar[DataType] = ...
+    R_64I: ClassVar[DataType] = ...
+    UNSET: ClassVar[DataType] = ...
+    _generate_next_value_: ClassVar[Callable] = ...
+    _member_map_: ClassVar[dict] = ...
+    _member_names_: ClassVar[list] = ...
+    _member_type_: ClassVar[type[int]] = ...
+    _unhashable_values_: ClassVar[list] = ...
+    _use_args_: ClassVar[bool] = ...
+    _value2member_map_: ClassVar[dict] = ...
+    def __format__(self, *args, **kwargs) -> str:
+        """Convert to a string according to format_spec."""
+
+class FactorizationAlg(enum.IntEnum):
+    """
+    See `cudssFactorizationAlg_t`.
+    """
+    __new__: ClassVar[Callable] = ...
+    DEFAULT: ClassVar[FactorizationAlg] = ...
+    GENERAL: ClassVar[FactorizationAlg] = ...
+    MULTIBLOCK: ClassVar[FactorizationAlg] = ...
+    _generate_next_value_: ClassVar[Callable] = ...
+    _member_map_: ClassVar[dict] = ...
+    _member_names_: ClassVar[list] = ...
+    _member_type_: ClassVar[type[int]] = ...
+    _unhashable_values_: ClassVar[list] = ...
+    _use_args_: ClassVar[bool] = ...
+    _value2member_map_: ClassVar[dict] = ...
+    def __format__(self, *args, **kwargs) -> str:
+        """Convert to a string according to format_spec."""
+
+class IndexBase(enum.IntEnum):
+    """
+    See `cudssIndexBase_t`.
+    """
+    __new__: ClassVar[Callable] = ...
+    ONE: ClassVar[IndexBase] = ...
+    ZERO: ClassVar[IndexBase] = ...
+    _generate_next_value_: ClassVar[Callable] = ...
+    _member_map_: ClassVar[dict] = ...
+    _member_names_: ClassVar[list] = ...
+    _member_type_: ClassVar[type[int]] = ...
+    _unhashable_values_: ClassVar[list] = ...
+    _use_args_: ClassVar[bool] = ...
+    _value2member_map_: ClassVar[dict] = ...
+    def __format__(self, *args, **kwargs) -> str:
+        """Convert to a string according to format_spec."""
+
+class Layout(enum.IntEnum):
+    """
+    See `cudssLayout_t`.
+    """
+    __new__: ClassVar[Callable] = ...
+    COL_MAJOR: ClassVar[Layout] = ...
+    ROW_MAJOR: ClassVar[Layout] = ...
+    _generate_next_value_: ClassVar[Callable] = ...
+    _member_map_: ClassVar[dict] = ...
+    _member_names_: ClassVar[list] = ...
+    _member_type_: ClassVar[type[int]] = ...
+    _unhashable_values_: ClassVar[list] = ...
+    _use_args_: ClassVar[bool] = ...
+    _value2member_map_: ClassVar[dict] = ...
+    def __format__(self, *args, **kwargs) -> str:
+        """Convert to a string according to format_spec."""
+
+class MatchingAlg(enum.IntEnum):
+    """
+    See `cudssMatchingAlg_t`.
+    """
+    __new__: ClassVar[Callable] = ...
+    AUTO: ClassVar[MatchingAlg] = ...
+    MAX_DIAG_COUNT: ClassVar[MatchingAlg] = ...
+    MAX_DIAG_PRODUCT: ClassVar[MatchingAlg] = ...
+    MAX_DIAG_SUM: ClassVar[MatchingAlg] = ...
+    MAX_MIN_DIAG: ClassVar[MatchingAlg] = ...
+    MAX_MIN_DIAG_ALT: ClassVar[MatchingAlg] = ...
+    NONE: ClassVar[MatchingAlg] = ...
+    _generate_next_value_: ClassVar[Callable] = ...
+    _member_map_: ClassVar[dict] = ...
+    _member_names_: ClassVar[list] = ...
+    _member_type_: ClassVar[type[int]] = ...
+    _unhashable_values_: ClassVar[list] = ...
+    _use_args_: ClassVar[bool] = ...
+    _value2member_map_: ClassVar[dict] = ...
+    def __format__(self, *args, **kwargs) -> str:
+        """Convert to a string according to format_spec."""
+
+class MatrixFormat(enum.IntEnum):
+    """
+    See `cudssMatrixFormat_t`.
+    """
+    __new__: ClassVar[Callable] = ...
+    BATCH: ClassVar[MatrixFormat] = ...
+    CSR: ClassVar[MatrixFormat] = ...
+    DENSE: ClassVar[MatrixFormat] = ...
+    DISTRIBUTED: ClassVar[MatrixFormat] = ...
+    _generate_next_value_: ClassVar[Callable] = ...
+    _member_map_: ClassVar[dict] = ...
+    _member_names_: ClassVar[list] = ...
+    _member_type_: ClassVar[type[int]] = ...
+    _unhashable_values_: ClassVar[list] = ...
+    _use_args_: ClassVar[bool] = ...
+    _value2member_map_: ClassVar[dict] = ...
+    def __format__(self, *args, **kwargs) -> str:
+        """Convert to a string according to format_spec."""
+
+class MatrixType(enum.IntEnum):
+    """
+    See `cudssMatrixType_t`.
+    """
+    __new__: ClassVar[Callable] = ...
+    GENERAL: ClassVar[MatrixType] = ...
+    HERMITIAN: ClassVar[MatrixType] = ...
+    HPD: ClassVar[MatrixType] = ...
+    SPD: ClassVar[MatrixType] = ...
+    SYMMETRIC: ClassVar[MatrixType] = ...
+    _generate_next_value_: ClassVar[Callable] = ...
+    _member_map_: ClassVar[dict] = ...
+    _member_names_: ClassVar[list] = ...
+    _member_type_: ClassVar[type[int]] = ...
+    _unhashable_values_: ClassVar[list] = ...
+    _use_args_: ClassVar[bool] = ...
+    _value2member_map_: ClassVar[dict] = ...
+    def __format__(self, *args, **kwargs) -> str:
+        """Convert to a string according to format_spec."""
+
+class MatrixViewType(enum.IntEnum):
+    """
+    See `cudssMatrixViewType_t`.
+    """
+    __new__: ClassVar[Callable] = ...
+    FULL: ClassVar[MatrixViewType] = ...
+    LOWER: ClassVar[MatrixViewType] = ...
+    UPPER: ClassVar[MatrixViewType] = ...
+    _generate_next_value_: ClassVar[Callable] = ...
+    _member_map_: ClassVar[dict] = ...
+    _member_names_: ClassVar[list] = ...
+    _member_type_: ClassVar[type[int]] = ...
+    _unhashable_values_: ClassVar[list] = ...
+    _use_args_: ClassVar[bool] = ...
+    _value2member_map_: ClassVar[dict] = ...
+    def __format__(self, *args, **kwargs) -> str:
+        """Convert to a string according to format_spec."""
+
+class OpType(enum.IntEnum):
+    """
+    See `cudssOpType_t`.
+    """
+    __new__: ClassVar[Callable] = ...
+    MAX: ClassVar[OpType] = ...
+    MIN: ClassVar[OpType] = ...
+    SUM: ClassVar[OpType] = ...
+    _generate_next_value_: ClassVar[Callable] = ...
+    _member_map_: ClassVar[dict] = ...
+    _member_names_: ClassVar[list] = ...
+    _member_type_: ClassVar[type[int]] = ...
+    _unhashable_values_: ClassVar[list] = ...
+    _use_args_: ClassVar[bool] = ...
+    _value2member_map_: ClassVar[dict] = ...
+    def __format__(self, *args, **kwargs) -> str:
+        """Convert to a string according to format_spec."""
+
+class Phase(enum.IntEnum):
+    """
+    See `cudssPhase_t`.
+    """
+    __new__: ClassVar[Callable] = ...
+    ANALYSIS: ClassVar[Phase] = ...
+    FACTORIZATION: ClassVar[Phase] = ...
+    REFACTORIZATION: ClassVar[Phase] = ...
+    REORDERING: ClassVar[Phase] = ...
+    SOLVE: ClassVar[Phase] = ...
+    SOLVE_BWD: ClassVar[Phase] = ...
+    SOLVE_BWD_PERM: ClassVar[Phase] = ...
+    SOLVE_DIAG: ClassVar[Phase] = ...
+    SOLVE_FWD: ClassVar[Phase] = ...
+    SOLVE_FWD_PERM: ClassVar[Phase] = ...
+    SOLVE_REFINEMENT: ClassVar[Phase] = ...
+    SYMBOLIC_FACTORIZATION: ClassVar[Phase] = ...
+    _generate_next_value_: ClassVar[Callable] = ...
+    _member_map_: ClassVar[dict] = ...
+    _member_names_: ClassVar[list] = ...
+    _member_type_: ClassVar[type[int]] = ...
+    _unhashable_values_: ClassVar[list] = ...
+    _use_args_: ClassVar[bool] = ...
+    _value2member_map_: ClassVar[dict] = ...
+    def __format__(self, *args, **kwargs) -> str:
+        """Convert to a string according to format_spec."""
+
+class PivotEpsilonAlg(enum.IntEnum):
+    """
+    See `cudssPivotEpsilonAlg_t`.
+    """
+    __new__: ClassVar[Callable] = ...
+    DEFAULT: ClassVar[PivotEpsilonAlg] = ...
+    SCALED: ClassVar[PivotEpsilonAlg] = ...
+    STATIC: ClassVar[PivotEpsilonAlg] = ...
+    _generate_next_value_: ClassVar[Callable] = ...
+    _member_map_: ClassVar[dict] = ...
+    _member_names_: ClassVar[list] = ...
+    _member_type_: ClassVar[type[int]] = ...
+    _unhashable_values_: ClassVar[list] = ...
+    _use_args_: ClassVar[bool] = ...
+    _value2member_map_: ClassVar[dict] = ...
+    def __format__(self, *args, **kwargs) -> str:
+        """Convert to a string according to format_spec."""
+
+class PivotType(enum.IntEnum):
+    """
+    See `cudssPivotType_t`.
+    """
+    __new__: ClassVar[Callable] = ...
+    PIVOT_AUTO: ClassVar[PivotType] = ...
+    PIVOT_BUNCH_KAUFMAN: ClassVar[PivotType] = ...
+    PIVOT_DIAGONAL: ClassVar[PivotType] = ...
+    PIVOT_GLOBAL_COL: ClassVar[PivotType] = ...
+    PIVOT_GLOBAL_ROW: ClassVar[PivotType] = ...
+    PIVOT_LOCAL_BLOCK: ClassVar[PivotType] = ...
+    PIVOT_NONE: ClassVar[PivotType] = ...
+    _generate_next_value_: ClassVar[Callable] = ...
+    _member_map_: ClassVar[dict] = ...
+    _member_names_: ClassVar[list] = ...
+    _member_type_: ClassVar[type[int]] = ...
+    _unhashable_values_: ClassVar[list] = ...
+    _use_args_: ClassVar[bool] = ...
+    _value2member_map_: ClassVar[dict] = ...
+    def __format__(self, *args, **kwargs) -> str:
+        """Convert to a string according to format_spec."""
+
+class ReorderingAlg(enum.IntEnum):
+    """
+    See `cudssReorderingAlg_t`.
+    """
+    __new__: ClassVar[Callable] = ...
+    AMD: ClassVar[ReorderingAlg] = ...
+    BTF_COLAMD: ClassVar[ReorderingAlg] = ...
+    COLAMD: ClassVar[ReorderingAlg] = ...
+    DEFAULT: ClassVar[ReorderingAlg] = ...
+    NESTED_DISSECTION: ClassVar[ReorderingAlg] = ...
+    NONE: ClassVar[ReorderingAlg] = ...
+    _generate_next_value_: ClassVar[Callable] = ...
+    _member_map_: ClassVar[dict] = ...
+    _member_names_: ClassVar[list] = ...
+    _member_type_: ClassVar[type[int]] = ...
+    _unhashable_values_: ClassVar[list] = ...
+    _use_args_: ClassVar[bool] = ...
+    _value2member_map_: ClassVar[dict] = ...
+    def __format__(self, *args, **kwargs) -> str:
+        """Convert to a string according to format_spec."""
+
+class SolveAlg(enum.IntEnum):
+    """
+    See `cudssSolveAlg_t`.
+    """
+    __new__: ClassVar[Callable] = ...
+    DEFAULT: ClassVar[SolveAlg] = ...
+    GENERAL: ClassVar[SolveAlg] = ...
+    _generate_next_value_: ClassVar[Callable] = ...
+    _member_map_: ClassVar[dict] = ...
+    _member_names_: ClassVar[list] = ...
+    _member_type_: ClassVar[type[int]] = ...
+    _unhashable_values_: ClassVar[list] = ...
+    _use_args_: ClassVar[bool] = ...
+    _value2member_map_: ClassVar[dict] = ...
+    def __format__(self, *args, **kwargs) -> str:
+        """Convert to a string according to format_spec."""
+
+class Status(enum.IntEnum):
+    """
+    See `cudssStatus_t`.
+    """
+    __new__: ClassVar[Callable] = ...
+    ALLOC_FAILED: ClassVar[Status] = ...
+    EXECUTION_FAILED: ClassVar[Status] = ...
+    INTERNAL_ERROR: ClassVar[Status] = ...
+    INVALID_VALUE: ClassVar[Status] = ...
+    IR_FAILED: ClassVar[Status] = ...
+    NOT_INITIALIZED: ClassVar[Status] = ...
+    NOT_SUPPORTED: ClassVar[Status] = ...
+    SUCCESS: ClassVar[Status] = ...
+    _generate_next_value_: ClassVar[Callable] = ...
+    _member_map_: ClassVar[dict] = ...
+    _member_names_: ClassVar[list] = ...
+    _member_type_: ClassVar[type[int]] = ...
+    _unhashable_values_: ClassVar[list] = ...
+    _use_args_: ClassVar[bool] = ...
+    _value2member_map_: ClassVar[dict] = ...
+    def __format__(self, *args, **kwargs) -> str:
+        """Convert to a string according to format_spec."""
+
+class cuDSSError(Exception):
+    """cuDSSError(status)"""
+    def __init__(self, status) -> Any:
+        """Initialize self.  See help(type(self)) for accurate signature."""
+    def __reduce__(self) -> Any:
+        """cuDSSError.__reduce__(self)"""

nvmath/bindings/cufft.pxd

@@ -0,0 +1,118 @@
+# Copyright (c) 2024-2026, NVIDIA CORPORATION & AFFILIATES. ALL RIGHTS RESERVED.
+#
+# SPDX-License-Identifier: Apache-2.0
+#
+# This code was automatically generated across versions from 11.0.3 to 13.2.0, generator version 0.3.1.dev1301+g7215ac36e. Do not modify it directly.
+
+from libc.stdint cimport intptr_t
+
+from .cycufft cimport *
+
+
+###############################################################################
+# Types
+###############################################################################
+
+ctypedef cufftXt1dFactors Xt1dFactors
+ctypedef cufftCallbackLoadC CallbackLoadC
+ctypedef cufftCallbackLoadZ CallbackLoadZ
+ctypedef cufftCallbackLoadR CallbackLoadR
+ctypedef cufftCallbackLoadD CallbackLoadD
+ctypedef cufftJITCallbackLoadC JITCallbackLoadC
+ctypedef cufftJITCallbackLoadZ JITCallbackLoadZ
+ctypedef cufftJITCallbackLoadR JITCallbackLoadR
+ctypedef cufftJITCallbackLoadD JITCallbackLoadD
+ctypedef cufftCallbackStoreR CallbackStoreR
+ctypedef cufftJITCallbackStoreR JITCallbackStoreR
+ctypedef cufftCallbackStoreD CallbackStoreD
+ctypedef cufftJITCallbackStoreD JITCallbackStoreD
+ctypedef cufftCallbackStoreC CallbackStoreC
+ctypedef cufftJITCallbackStoreC JITCallbackStoreC
+ctypedef cufftCallbackStoreZ CallbackStoreZ
+ctypedef cufftJITCallbackStoreZ JITCallbackStoreZ
+
+ctypedef cudaStream_t Stream
+ctypedef cudaDataType DataType
+ctypedef libraryPropertyType_t LibraryPropertyType
+
+
+###############################################################################
+# Enum
+###############################################################################
+
+ctypedef libFormat_t _LibFormat
+ctypedef cufftResult _Result
+ctypedef cufftType _Type
+ctypedef cufftCompatibility _Compatibility
+ctypedef cufftXtSubFormat _XtSubFormat
+ctypedef cufftXtCopyType _XtCopyType
+ctypedef cufftXtQueryType _XtQueryType
+ctypedef cufftXtWorkAreaPolicy _XtWorkAreaPolicy
+ctypedef cufftXtCallbackType _XtCallbackType
+ctypedef cufftProperty _Property
+
+
+###############################################################################
+# Functions
+###############################################################################
+
+cpdef plan1d(intptr_t plan, int nx, int type, int batch)
+cpdef plan2d(intptr_t plan, int nx, int ny, int type)
+cpdef plan3d(intptr_t plan, int nx, int ny, int nz, int type)
+cpdef plan_many(intptr_t plan, int rank, n, inembed, int istride, int idist, onembed, int ostride, int odist, int type, int batch)
+cpdef size_t make_plan1d(int plan, int nx, int type, int batch) except? -1
+cpdef size_t make_plan2d(int plan, int nx, int ny, int type) except? -1
+cpdef size_t make_plan3d(int plan, int nx, int ny, int nz, int type) except? -1
+cpdef size_t make_plan_many(int plan, int rank, n, inembed, int istride, int idist, onembed, int ostride, int odist, int type, int batch) except? -1
+cpdef size_t make_plan_many64(int plan, int rank, n, inembed, long long int istride, long long int idist, onembed, long long int ostride, long long int odist, int type, long long int batch) except? -1
+cpdef size_t get_size_many64(int plan, int rank, n, inembed, long long int istride, long long int idist, onembed, long long int ostride, long long int odist, int type, long long int batch) except? -1
+cpdef size_t estimate1d(int nx, int type, int batch) except? -1
+cpdef size_t estimate2d(int nx, int ny, int type) except? -1
+cpdef size_t estimate3d(int nx, int ny, int nz, int type) except? -1
+cpdef size_t estimate_many(int rank, n, inembed, int istride, int idist, onembed, int ostride, int odist, int type, int batch) except? -1
+cpdef int create() except? -1
+cpdef size_t get_size1d(int handle, int nx, int type, int batch) except? -1
+cpdef size_t get_size2d(int handle, int nx, int ny, int type) except? -1
+cpdef size_t get_size3d(int handle, int nx, int ny, int nz, int type) except? -1
+cpdef size_t get_size_many(int handle, int rank, intptr_t n, intptr_t inembed, int istride, int idist, intptr_t onembed, int ostride, int odist, int type, int batch) except? -1
+cpdef size_t get_size(int handle) except? -1
+cpdef set_work_area(int plan, intptr_t work_area)
+cpdef set_auto_allocation(int plan, int auto_allocate)
+cpdef exec_c2c(int plan, intptr_t idata, intptr_t odata, int direction)
+cpdef exec_r2c(int plan, intptr_t idata, intptr_t odata)
+cpdef exec_c2r(int plan, intptr_t idata, intptr_t odata)
+cpdef exec_z2z(int plan, intptr_t idata, intptr_t odata, int direction)
+cpdef exec_d2z(int plan, intptr_t idata, intptr_t odata)
+cpdef exec_z2d(int plan, intptr_t idata, intptr_t odata)
+cpdef set_stream(int plan, intptr_t stream)
+cpdef destroy(int plan)
+cpdef int get_version() except? -1
+cpdef int get_property(int type) except? -1
+cpdef xt_set_gpus(int handle, int n_gpus, which_gpus)
+cpdef intptr_t xt_malloc(int plan, int format) except? -1
+cpdef xt_memcpy(int plan, intptr_t dst_pointer, intptr_t src_pointer, int type)
+cpdef xt_free(intptr_t descriptor)
+cpdef xt_set_work_area(int plan, intptr_t work_area)
+cpdef xt_exec_descriptor_c2c(int plan, intptr_t input, intptr_t output, int direction)
+cpdef xt_exec_descriptor_r2c(int plan, intptr_t input, intptr_t output)
+cpdef xt_exec_descriptor_c2r(int plan, intptr_t input, intptr_t output)
+cpdef xt_exec_descriptor_z2z(int plan, intptr_t input, intptr_t output, int direction)
+cpdef xt_exec_descriptor_d2z(int plan, intptr_t input, intptr_t output)
+cpdef xt_exec_descriptor_z2d(int plan, intptr_t input, intptr_t output)
+cpdef xt_query_plan(int plan, intptr_t query_struct, int query_type)
+cpdef xt_clear_callback(int plan, int cb_type)
+cpdef xt_set_callback_shared_size(int plan, int cb_type, size_t shared_size)
+cpdef size_t xt_make_plan_many(int plan, int rank, n, inembed, long long int istride, long long int idist, int inputtype, onembed, long long int ostride, long long int odist, int outputtype, long long int batch, int executiontype) except? 0
+cpdef size_t xt_get_size_many(int plan, int rank, n, inembed, long long int istride, long long int idist, int inputtype, onembed, long long int ostride, long long int odist, int outputtype, long long int batch, int executiontype) except? 0
+cpdef xt_exec(int plan, intptr_t input, intptr_t output, int direction)
+cpdef xt_exec_descriptor(int plan, intptr_t input, intptr_t output, int direction)
+cpdef xt_set_work_area_policy(int plan, int policy, intptr_t work_size)
+cpdef _xt_set_jit_callback(int plan, intptr_t lto_callback_symbol_name, lto_callback_fatbin, size_t lto_callback_fatbin_size, int type, caller_info)
+cpdef xt_set_subformat_default(int plan, int subformat_forward, int subformat_inverse)
+cpdef set_plan_property_int64(int plan, int property, long long int input_value_int)
+cpdef long long int get_plan_property_int64(int plan, int property) except? -1
+cpdef reset_plan_property(int plan, int property)
+cpdef _xt_set_jit_callback_12_7(int plan, intptr_t lto_callback_symbol_name, lto_callback_fatbin, size_t lto_callback_fatbin_size, int type, caller_info)
+
+# wrapper calling the correct function based on the CTK version
+cpdef xt_set_jit_callback(int plan, intptr_t lto_callback_symbol_name, lto_callback_fatbin, size_t lto_callback_fatbin_size, int type, caller_info)