nvmath-python 1.0.0__cp314-cp314t-win_amd64.whl

nvmath/bindings/cufft.pyi

@@ -0,0 +1,301 @@
+# 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
+create: _cython_3_2_5.cython_function_or_method
+destroy: _cython_3_2_5.cython_function_or_method
+estimate1d: _cython_3_2_5.cython_function_or_method
+estimate2d: _cython_3_2_5.cython_function_or_method
+estimate3d: _cython_3_2_5.cython_function_or_method
+estimate_many: _cython_3_2_5.cython_function_or_method
+exec_c2c: _cython_3_2_5.cython_function_or_method
+exec_c2r: _cython_3_2_5.cython_function_or_method
+exec_d2z: _cython_3_2_5.cython_function_or_method
+exec_r2c: _cython_3_2_5.cython_function_or_method
+exec_z2d: _cython_3_2_5.cython_function_or_method
+exec_z2z: _cython_3_2_5.cython_function_or_method
+get_plan_property_int64: _cython_3_2_5.cython_function_or_method
+get_property: _cython_3_2_5.cython_function_or_method
+get_size: _cython_3_2_5.cython_function_or_method
+get_size1d: _cython_3_2_5.cython_function_or_method
+get_size2d: _cython_3_2_5.cython_function_or_method
+get_size3d: _cython_3_2_5.cython_function_or_method
+get_size_many: _cython_3_2_5.cython_function_or_method
+get_size_many64: _cython_3_2_5.cython_function_or_method
+get_version: _cython_3_2_5.cython_function_or_method
+make_plan1d: _cython_3_2_5.cython_function_or_method
+make_plan2d: _cython_3_2_5.cython_function_or_method
+make_plan3d: _cython_3_2_5.cython_function_or_method
+make_plan_many: _cython_3_2_5.cython_function_or_method
+make_plan_many64: _cython_3_2_5.cython_function_or_method
+plan1d: _cython_3_2_5.cython_function_or_method
+plan2d: _cython_3_2_5.cython_function_or_method
+plan3d: _cython_3_2_5.cython_function_or_method
+plan_many: _cython_3_2_5.cython_function_or_method
+reset_plan_property: _cython_3_2_5.cython_function_or_method
+set_auto_allocation: _cython_3_2_5.cython_function_or_method
+set_plan_property_int64: _cython_3_2_5.cython_function_or_method
+set_stream: _cython_3_2_5.cython_function_or_method
+set_work_area: _cython_3_2_5.cython_function_or_method
+xt_clear_callback: _cython_3_2_5.cython_function_or_method
+xt_exec: _cython_3_2_5.cython_function_or_method
+xt_exec_descriptor: _cython_3_2_5.cython_function_or_method
+xt_exec_descriptor_c2c: _cython_3_2_5.cython_function_or_method
+xt_exec_descriptor_c2r: _cython_3_2_5.cython_function_or_method
+xt_exec_descriptor_d2z: _cython_3_2_5.cython_function_or_method
+xt_exec_descriptor_r2c: _cython_3_2_5.cython_function_or_method
+xt_exec_descriptor_z2d: _cython_3_2_5.cython_function_or_method
+xt_exec_descriptor_z2z: _cython_3_2_5.cython_function_or_method
+xt_free: _cython_3_2_5.cython_function_or_method
+xt_get_size_many: _cython_3_2_5.cython_function_or_method
+xt_make_plan_many: _cython_3_2_5.cython_function_or_method
+xt_malloc: _cython_3_2_5.cython_function_or_method
+xt_memcpy: _cython_3_2_5.cython_function_or_method
+xt_query_plan: _cython_3_2_5.cython_function_or_method
+xt_set_callback_shared_size: _cython_3_2_5.cython_function_or_method
+xt_set_gpus: _cython_3_2_5.cython_function_or_method
+xt_set_jit_callback: _cython_3_2_5.cython_function_or_method
+xt_set_subformat_default: _cython_3_2_5.cython_function_or_method
+xt_set_work_area: _cython_3_2_5.cython_function_or_method
+xt_set_work_area_policy: _cython_3_2_5.cython_function_or_method
+
+class Compatibility(enum.IntEnum):
+    """
+    See `cufftCompatibility`.
+    """
+    __new__: ClassVar[Callable] = ...
+    FFTW_PADDING: ClassVar[Compatibility] = ...
+    _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 LibFormat(enum.IntEnum):
+    """
+    See `libFormat_t`.
+    """
+    __new__: ClassVar[Callable] = ...
+    CUFFT: ClassVar[LibFormat] = ...
+    UNDEFINED: ClassVar[LibFormat] = ...
+    _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 Property(enum.IntEnum):
+    """
+    See `cufftProperty`.
+    """
+    __new__: ClassVar[Callable] = ...
+    MAX_NUM_HOST_THREADS: ClassVar[Property] = ...
+    PATIENT_JIT: ClassVar[Property] = ...
+    _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 Result(enum.IntEnum):
+    """
+    See `cufftResult`.
+    """
+    __new__: ClassVar[Callable] = ...
+    ALLOC_FAILED: ClassVar[Result] = ...
+    EXEC_FAILED: ClassVar[Result] = ...
+    INCOMPLETE_PARAMETER_LIST: ClassVar[Result] = ...
+    INTERNAL_ERROR: ClassVar[Result] = ...
+    INVALID_DEVICE: ClassVar[Result] = ...
+    INVALID_PLAN: ClassVar[Result] = ...
+    INVALID_SIZE: ClassVar[Result] = ...
+    INVALID_TYPE: ClassVar[Result] = ...
+    INVALID_VALUE: ClassVar[Result] = ...
+    LICENSE_ERROR: ClassVar[Result] = ...
+    MISSING_DEPENDENCY: ClassVar[Result] = ...
+    NOT_IMPLEMENTED: ClassVar[Result] = ...
+    NOT_SUPPORTED: ClassVar[Result] = ...
+    NO_WORKSPACE: ClassVar[Result] = ...
+    NVJITLINK_FAILURE: ClassVar[Result] = ...
+    NVRTC_FAILURE: ClassVar[Result] = ...
+    NVSHMEM_FAILURE: ClassVar[Result] = ...
+    PARSE_ERROR: ClassVar[Result] = ...
+    SETUP_FAILED: ClassVar[Result] = ...
+    SUCCESS: ClassVar[Result] = ...
+    UNALIGNED_DATA: ClassVar[Result] = ...
+    _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 Type(enum.IntEnum):
+    """
+    See `cufftType`.
+    """
+    __new__: ClassVar[Callable] = ...
+    C2C: ClassVar[Type] = ...
+    C2R: ClassVar[Type] = ...
+    D2Z: ClassVar[Type] = ...
+    R2C: ClassVar[Type] = ...
+    Z2D: ClassVar[Type] = ...
+    Z2Z: ClassVar[Type] = ...
+    _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 XtCallbackType(enum.IntEnum):
+    """
+    See `cufftXtCallbackType`.
+    """
+    __new__: ClassVar[Callable] = ...
+    LD_COMPLEX: ClassVar[XtCallbackType] = ...
+    LD_COMPLEX_DOUBLE: ClassVar[XtCallbackType] = ...
+    LD_REAL: ClassVar[XtCallbackType] = ...
+    LD_REAL_DOUBLE: ClassVar[XtCallbackType] = ...
+    ST_COMPLEX: ClassVar[XtCallbackType] = ...
+    ST_COMPLEX_DOUBLE: ClassVar[XtCallbackType] = ...
+    ST_REAL: ClassVar[XtCallbackType] = ...
+    ST_REAL_DOUBLE: ClassVar[XtCallbackType] = ...
+    UNDEFINED: ClassVar[XtCallbackType] = ...
+    _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 XtCopyType(enum.IntEnum):
+    """
+    See `cufftXtCopyType`.
+    """
+    __new__: ClassVar[Callable] = ...
+    DEVICE_TO_DEVICE: ClassVar[XtCopyType] = ...
+    DEVICE_TO_HOST: ClassVar[XtCopyType] = ...
+    HOST_TO_DEVICE: ClassVar[XtCopyType] = ...
+    UNDEFINED: ClassVar[XtCopyType] = ...
+    _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 XtQueryType(enum.IntEnum):
+    """
+    See `cufftXtQueryType`.
+    """
+    __new__: ClassVar[Callable] = ...
+    QUERY_1D_FACTORS: ClassVar[XtQueryType] = ...
+    QUERY_UNDEFINED: ClassVar[XtQueryType] = ...
+    _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 XtSubFormat(enum.IntEnum):
+    """
+    See `cufftXtSubFormat`.
+    """
+    __new__: ClassVar[Callable] = ...
+    FORMAT_1D_INPUT_SHUFFLED: ClassVar[XtSubFormat] = ...
+    FORMAT_DISTRIBUTED_INPUT: ClassVar[XtSubFormat] = ...
+    FORMAT_DISTRIBUTED_OUTPUT: ClassVar[XtSubFormat] = ...
+    FORMAT_INPLACE: ClassVar[XtSubFormat] = ...
+    FORMAT_INPLACE_SHUFFLED: ClassVar[XtSubFormat] = ...
+    FORMAT_INPUT: ClassVar[XtSubFormat] = ...
+    FORMAT_OUTPUT: ClassVar[XtSubFormat] = ...
+    FORMAT_UNDEFINED: ClassVar[XtSubFormat] = ...
+    _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 XtWorkAreaPolicy(enum.IntEnum):
+    """
+    See `cufftXtWorkAreaPolicy`.
+    """
+    __new__: ClassVar[Callable] = ...
+    MINIMAL: ClassVar[XtWorkAreaPolicy] = ...
+    PERFORMANCE: ClassVar[XtWorkAreaPolicy] = ...
+    USER: ClassVar[XtWorkAreaPolicy] = ...
+    _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 cuFFTError(Exception):
+    def __init__(self, status) -> Any:
+        """cuFFTError.__init__(self, status)"""
+    def __reduce__(self) -> Any:
+        """cuFFTError.__reduce__(self)"""

nvmath/bindings/cufftMp.pxd

@@ -0,0 +1,124 @@
+# Copyright (c) 2025-2026, NVIDIA CORPORATION & AFFILIATES. ALL RIGHTS RESERVED.
+#
+# SPDX-License-Identifier: Apache-2.0
+#
+# This code was automatically generated across versions from 11.2.6 to 11.4.0, generator version 0.3.1.dev1303+g031f1197f. Do not modify it directly.
+
+from libc.stdint cimport intptr_t
+
+from .cycufftMp cimport *
+
+
+###############################################################################
+# Types
+###############################################################################
+
+ctypedef cufftReshapeHandle ReshapeHandle
+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 cufftProperty _Property
+ctypedef cufftXtSubFormat _XtSubFormat
+ctypedef cufftXtCopyType _XtCopyType
+ctypedef cufftXtQueryType _XtQueryType
+ctypedef cufftXtWorkAreaPolicy _XtWorkAreaPolicy
+ctypedef cufftXtCallbackType _XtCallbackType
+ctypedef cufftMpCommType _MpCommType
+
+
+###############################################################################
+# 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 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_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 attach_comm(int plan, int comm_type, intptr_t comm_handle)
+cpdef xt_set_distribution(int plan, int rank, lower_input, upper_input, lower_output, upper_output, strides_input, strides_output)
+cpdef xt_set_subformat_default(int plan, int subformat_forward, int subformat_inverse)
+cpdef intptr_t create_reshape() except? -1
+cpdef attach_reshape_comm(intptr_t handle, int comm_type, intptr_t comm_handle)
+cpdef size_t get_reshape_size(intptr_t handle) except? -1
+cpdef make_reshape_11_2(intptr_t handle, size_t element_size, int rank, lower_input, upper_input, lower_output, upper_output, strides_input, strides_output)
+cpdef exec_reshape_async(intptr_t handle, intptr_t data_out, intptr_t data_in, intptr_t workspace, intptr_t stream)
+cpdef destroy_reshape(intptr_t handle)
+cpdef make_reshape_11_4(intptr_t handle, size_t element_size, int rank, lower_input, upper_input, strides_input, lower_output, upper_output, strides_output, intptr_t comm_handle, int comm_type)