nvmath-python 0.5.0__cp313-cp313-win_amd64.whl

nvmath/bindings/cycufftMp.pxd

@@ -0,0 +1,333 @@
+# Copyright (c) 2025, 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. Do not modify it directly.
+# This layer exposes the C header to Cython as-is.
+
+###############################################################################
+# Types (structs, enums, ...)
+###############################################################################
+
+# enums
+ctypedef enum libFormat_t "libFormat_t":
+    LIB_FORMAT_CUFFT "LIB_FORMAT_CUFFT" = 0x0
+    LIB_FORMAT_UNDEFINED "LIB_FORMAT_UNDEFINED" = 0x1
+
+ctypedef enum cufftResult "cufftResult":
+    CUFFT_SUCCESS "CUFFT_SUCCESS" = 0x0
+    CUFFT_INVALID_PLAN "CUFFT_INVALID_PLAN" = 0x1
+    CUFFT_ALLOC_FAILED "CUFFT_ALLOC_FAILED" = 0x2
+    CUFFT_INVALID_TYPE "CUFFT_INVALID_TYPE" = 0x3
+    CUFFT_INVALID_VALUE "CUFFT_INVALID_VALUE" = 0x4
+    CUFFT_INTERNAL_ERROR "CUFFT_INTERNAL_ERROR" = 0x5
+    CUFFT_EXEC_FAILED "CUFFT_EXEC_FAILED" = 0x6
+    CUFFT_SETUP_FAILED "CUFFT_SETUP_FAILED" = 0x7
+    CUFFT_INVALID_SIZE "CUFFT_INVALID_SIZE" = 0x8
+    CUFFT_UNALIGNED_DATA "CUFFT_UNALIGNED_DATA" = 0x9
+    CUFFT_INCOMPLETE_PARAMETER_LIST "CUFFT_INCOMPLETE_PARAMETER_LIST" = 0xA
+    CUFFT_INVALID_DEVICE "CUFFT_INVALID_DEVICE" = 0xB
+    CUFFT_PARSE_ERROR "CUFFT_PARSE_ERROR" = 0xC
+    CUFFT_NO_WORKSPACE "CUFFT_NO_WORKSPACE" = 0xD
+    CUFFT_NOT_IMPLEMENTED "CUFFT_NOT_IMPLEMENTED" = 0xE
+    CUFFT_LICENSE_ERROR "CUFFT_LICENSE_ERROR" = 0x0F
+    CUFFT_NOT_SUPPORTED "CUFFT_NOT_SUPPORTED" = 0x10
+    _CUFFTRESULT_INTERNAL_LOADING_ERROR "_CUFFTRESULT_INTERNAL_LOADING_ERROR" = -42
+
+ctypedef enum cufftType "cufftType":
+    CUFFT_R2C "CUFFT_R2C" = 0x2a
+    CUFFT_C2R "CUFFT_C2R" = 0x2c
+    CUFFT_C2C "CUFFT_C2C" = 0x29
+    CUFFT_D2Z "CUFFT_D2Z" = 0x6a
+    CUFFT_Z2D "CUFFT_Z2D" = 0x6c
+    CUFFT_Z2Z "CUFFT_Z2Z" = 0x69
+
+ctypedef enum cufftCompatibility "cufftCompatibility":
+    CUFFT_COMPATIBILITY_FFTW_PADDING "CUFFT_COMPATIBILITY_FFTW_PADDING" = 0x01
+
+ctypedef enum cufftProperty "cufftProperty":
+    NVFFT_PLAN_PROPERTY_INT64_PATIENT_JIT "NVFFT_PLAN_PROPERTY_INT64_PATIENT_JIT" = 0x1
+    NVFFT_PLAN_PROPERTY_INT64_MAX_NUM_HOST_THREADS "NVFFT_PLAN_PROPERTY_INT64_MAX_NUM_HOST_THREADS" = 0x2
+
+ctypedef enum cufftXtSubFormat "cufftXtSubFormat":
+    CUFFT_XT_FORMAT_INPUT "CUFFT_XT_FORMAT_INPUT" = 0x00
+    CUFFT_XT_FORMAT_OUTPUT "CUFFT_XT_FORMAT_OUTPUT" = 0x01
+    CUFFT_XT_FORMAT_INPLACE "CUFFT_XT_FORMAT_INPLACE" = 0x02
+    CUFFT_XT_FORMAT_INPLACE_SHUFFLED "CUFFT_XT_FORMAT_INPLACE_SHUFFLED" = 0x03
+    CUFFT_XT_FORMAT_1D_INPUT_SHUFFLED "CUFFT_XT_FORMAT_1D_INPUT_SHUFFLED" = 0x04
+    CUFFT_XT_FORMAT_DISTRIBUTED_INPUT "CUFFT_XT_FORMAT_DISTRIBUTED_INPUT" = 0x05
+    CUFFT_XT_FORMAT_DISTRIBUTED_OUTPUT "CUFFT_XT_FORMAT_DISTRIBUTED_OUTPUT" = 0x06
+    CUFFT_FORMAT_UNDEFINED "CUFFT_FORMAT_UNDEFINED" = 0x07
+
+ctypedef enum cufftXtCopyType "cufftXtCopyType":
+    CUFFT_COPY_HOST_TO_DEVICE "CUFFT_COPY_HOST_TO_DEVICE" = 0x00
+    CUFFT_COPY_DEVICE_TO_HOST "CUFFT_COPY_DEVICE_TO_HOST" = 0x01
+    CUFFT_COPY_DEVICE_TO_DEVICE "CUFFT_COPY_DEVICE_TO_DEVICE" = 0x02
+    CUFFT_COPY_UNDEFINED "CUFFT_COPY_UNDEFINED" = 0x03
+
+ctypedef enum cufftXtQueryType "cufftXtQueryType":
+    CUFFT_QUERY_1D_FACTORS "CUFFT_QUERY_1D_FACTORS" = 0x00
+    CUFFT_QUERY_UNDEFINED "CUFFT_QUERY_UNDEFINED" = 0x01
+
+ctypedef enum cufftXtWorkAreaPolicy "cufftXtWorkAreaPolicy":
+    CUFFT_WORKAREA_MINIMAL "CUFFT_WORKAREA_MINIMAL" = 0
+    CUFFT_WORKAREA_USER "CUFFT_WORKAREA_USER" = 1
+    CUFFT_WORKAREA_PERFORMANCE "CUFFT_WORKAREA_PERFORMANCE" = 2
+
+ctypedef enum cufftXtCallbackType "cufftXtCallbackType":
+    CUFFT_CB_LD_COMPLEX "CUFFT_CB_LD_COMPLEX" = 0x0
+    CUFFT_CB_LD_COMPLEX_DOUBLE "CUFFT_CB_LD_COMPLEX_DOUBLE" = 0x1
+    CUFFT_CB_LD_REAL "CUFFT_CB_LD_REAL" = 0x2
+    CUFFT_CB_LD_REAL_DOUBLE "CUFFT_CB_LD_REAL_DOUBLE" = 0x3
+    CUFFT_CB_ST_COMPLEX "CUFFT_CB_ST_COMPLEX" = 0x4
+    CUFFT_CB_ST_COMPLEX_DOUBLE "CUFFT_CB_ST_COMPLEX_DOUBLE" = 0x5
+    CUFFT_CB_ST_REAL "CUFFT_CB_ST_REAL" = 0x6
+    CUFFT_CB_ST_REAL_DOUBLE "CUFFT_CB_ST_REAL_DOUBLE" = 0x7
+    CUFFT_CB_UNDEFINED "CUFFT_CB_UNDEFINED" = 0x8
+
+ctypedef enum cufftMpCommType "cufftMpCommType":
+    CUFFT_COMM_MPI "CUFFT_COMM_MPI" = 0x00
+    CUFFT_COMM_NONE "CUFFT_COMM_NONE" = 0x01
+
+
+# types
+cdef extern from *:
+    """
+    #include <driver_types.h>
+    #include <library_types.h>
+    #include <cuComplex.h>
+
+    #define MAX_CUDA_DESCRIPTOR_GPUS 64
+    #define CUFFT_FORWARD -1
+    #define CUFFT_INVERSE  1
+    """
+    ctypedef void* cudaStream_t 'cudaStream_t'
+    ctypedef int cudaDataType_t 'cudaDataType_t'
+    ctypedef int cudaDataType 'cudaDataType'
+    ctypedef int libraryPropertyType_t 'libraryPropertyType_t'
+    ctypedef int libraryPropertyType 'libraryPropertyType'
+
+    ctypedef struct cuComplex:
+        pass
+    ctypedef struct cuDoubleComplex:
+        pass
+
+    cdef const int MAX_CUDA_DESCRIPTOR_GPUS
+    cdef const int cuFFTFORWARD
+    cdef const int cuFFTINVERSE
+
+
+ctypedef float cufftReal 'cufftReal'
+ctypedef double cufftDoubleReal 'cufftDoubleReal'
+ctypedef cuComplex cufftComplex 'cufftComplex'
+ctypedef cuDoubleComplex cufftDoubleComplex 'cufftDoubleComplex'
+ctypedef int cufftHandle 'cufftHandle'
+ctypedef void* cufftReshapeHandle 'cufftReshapeHandle'
+ctypedef struct cufftXt1dFactors 'cufftXt1dFactors':
+    long long int size
+    long long int stringCount
+    long long int stringLength
+    long long int substringLength
+    long long int factor1
+    long long int factor2
+    long long int stringMask
+    long long int substringMask
+    long long int factor1Mask
+    long long int factor2Mask
+    int stringShift
+    int substringShift
+    int factor1Shift
+    int factor2Shift
+ctypedef cufftComplex (*cufftCallbackLoadC 'cufftCallbackLoadC')(
+    void* dataIn,
+    size_t offset,
+    void* callerInfo,
+    void* sharedPointer
+)
+ctypedef cufftDoubleComplex (*cufftCallbackLoadZ 'cufftCallbackLoadZ')(
+    void* dataIn,
+    size_t offset,
+    void* callerInfo,
+    void* sharedPointer
+)
+ctypedef cufftReal (*cufftCallbackLoadR 'cufftCallbackLoadR')(
+    void* dataIn,
+    size_t offset,
+    void* callerInfo,
+    void* sharedPointer
+)
+ctypedef cufftDoubleReal (*cufftCallbackLoadD 'cufftCallbackLoadD')(
+    void* dataIn,
+    size_t offset,
+    void* callerInfo,
+    void* sharedPointer
+)
+ctypedef cufftComplex (*cufftJITCallbackLoadC 'cufftJITCallbackLoadC')(
+    void* dataIn,
+    unsigned long long offset,
+    void* callerInfo,
+    void* sharedPointer
+)
+ctypedef cufftDoubleComplex (*cufftJITCallbackLoadZ 'cufftJITCallbackLoadZ')(
+    void* dataIn,
+    unsigned long long offset,
+    void* callerInfo,
+    void* sharedPointer
+)
+ctypedef cufftReal (*cufftJITCallbackLoadR 'cufftJITCallbackLoadR')(
+    void* dataIn,
+    unsigned long long offset,
+    void* callerInfo,
+    void* sharedPointer
+)
+ctypedef cufftDoubleReal (*cufftJITCallbackLoadD 'cufftJITCallbackLoadD')(
+    void* dataIn,
+    unsigned long long offset,
+    void* callerInfo,
+    void* sharedPointer
+)
+ctypedef void (*cufftCallbackStoreR 'cufftCallbackStoreR')(
+    void* dataOut,
+    size_t offset,
+    cufftReal element,
+    void* callerInfo,
+    void* sharedPointer
+)
+ctypedef void (*cufftJITCallbackStoreR 'cufftJITCallbackStoreR')(
+    void* dataOut,
+    unsigned long long offset,
+    cufftReal element,
+    void* callerInfo,
+    void* sharedPointer
+)
+ctypedef void (*cufftCallbackStoreD 'cufftCallbackStoreD')(
+    void* dataOut,
+    size_t offset,
+    cufftDoubleReal element,
+    void* callerInfo,
+    void* sharedPointer
+)
+ctypedef void (*cufftJITCallbackStoreD 'cufftJITCallbackStoreD')(
+    void* dataOut,
+    unsigned long long offset,
+    cufftDoubleReal element,
+    void* callerInfo,
+    void* sharedPointer
+)
+ctypedef void (*cufftCallbackStoreC 'cufftCallbackStoreC')(
+    void* dataOut,
+    size_t offset,
+    cufftComplex element,
+    void* callerInfo,
+    void* sharedPointer
+)
+ctypedef void (*cufftJITCallbackStoreC 'cufftJITCallbackStoreC')(
+    void* dataOut,
+    unsigned long long offset,
+    cufftComplex element,
+    void* callerInfo,
+    void* sharedPointer
+)
+ctypedef void (*cufftCallbackStoreZ 'cufftCallbackStoreZ')(
+    void* dataOut,
+    size_t offset,
+    cufftDoubleComplex element,
+    void* callerInfo,
+    void* sharedPointer
+)
+ctypedef void (*cufftJITCallbackStoreZ 'cufftJITCallbackStoreZ')(
+    void* dataOut,
+    unsigned long long offset,
+    cufftDoubleComplex element,
+    void* callerInfo,
+    void* sharedPointer
+)
+
+ctypedef struct cudaXtDesc_t 'cudaXtDesc_t':
+    int version "version"
+    int nGPUs "nGPUs"
+    int GPUs "GPUs" [MAX_CUDA_DESCRIPTOR_GPUS]
+    void* data "data" [MAX_CUDA_DESCRIPTOR_GPUS]
+    size_t size "size" [MAX_CUDA_DESCRIPTOR_GPUS]
+    void* cudaXtState "cudaXtState"
+
+ctypedef cudaXtDesc_t cudaXtDesc 'cudaXtDesc'
+
+ctypedef struct cudaLibXtDesc_t 'cudaLibXtDesc_t':
+    int version "version"
+    cudaXtDesc* descriptor "descriptor"
+    libFormat_t library "library"
+    int subFormat "subFormat"
+    void* libDescriptor "libDescriptor"
+
+ctypedef cudaLibXtDesc_t cudaLibXtDesc 'cudaLibXtDesc'
+
+
+###############################################################################
+# Functions
+###############################################################################
+
+cdef cufftResult cufftPlan1d(cufftHandle* plan, int nx, cufftType type, int batch) except?_CUFFTRESULT_INTERNAL_LOADING_ERROR nogil
+cdef cufftResult cufftPlan2d(cufftHandle* plan, int nx, int ny, cufftType type) except?_CUFFTRESULT_INTERNAL_LOADING_ERROR nogil
+cdef cufftResult cufftPlan3d(cufftHandle* plan, int nx, int ny, int nz, cufftType type) except?_CUFFTRESULT_INTERNAL_LOADING_ERROR nogil
+cdef cufftResult cufftPlanMany(cufftHandle* plan, int rank, int* n, int* inembed, int istride, int idist, int* onembed, int ostride, int odist, cufftType type, int batch) except?_CUFFTRESULT_INTERNAL_LOADING_ERROR nogil
+cdef cufftResult cufftMakePlan1d(cufftHandle plan, int nx, cufftType type, int batch, size_t* workSize) except?_CUFFTRESULT_INTERNAL_LOADING_ERROR nogil
+cdef cufftResult cufftMakePlan2d(cufftHandle plan, int nx, int ny, cufftType type, size_t* workSize) except?_CUFFTRESULT_INTERNAL_LOADING_ERROR nogil
+cdef cufftResult cufftMakePlan3d(cufftHandle plan, int nx, int ny, int nz, cufftType type, size_t* workSize) except?_CUFFTRESULT_INTERNAL_LOADING_ERROR nogil
+cdef cufftResult cufftMakePlanMany(cufftHandle plan, int rank, int* n, int* inembed, int istride, int idist, int* onembed, int ostride, int odist, cufftType type, int batch, size_t* workSize) except?_CUFFTRESULT_INTERNAL_LOADING_ERROR nogil
+cdef cufftResult cufftMakePlanMany64(cufftHandle plan, int rank, long long int* n, long long int* inembed, long long int istride, long long int idist, long long int* onembed, long long int ostride, long long int odist, cufftType type, long long int batch, size_t* workSize) except?_CUFFTRESULT_INTERNAL_LOADING_ERROR nogil
+cdef cufftResult cufftGetSizeMany64(cufftHandle plan, int rank, long long int* n, long long int* inembed, long long int istride, long long int idist, long long int* onembed, long long int ostride, long long int odist, cufftType type, long long int batch, size_t* workSize) except?_CUFFTRESULT_INTERNAL_LOADING_ERROR nogil
+cdef cufftResult cufftEstimate1d(int nx, cufftType type, int batch, size_t* workSize) except?_CUFFTRESULT_INTERNAL_LOADING_ERROR nogil
+cdef cufftResult cufftEstimate2d(int nx, int ny, cufftType type, size_t* workSize) except?_CUFFTRESULT_INTERNAL_LOADING_ERROR nogil
+cdef cufftResult cufftEstimate3d(int nx, int ny, int nz, cufftType type, size_t* workSize) except?_CUFFTRESULT_INTERNAL_LOADING_ERROR nogil
+cdef cufftResult cufftEstimateMany(int rank, int* n, int* inembed, int istride, int idist, int* onembed, int ostride, int odist, cufftType type, int batch, size_t* workSize) except?_CUFFTRESULT_INTERNAL_LOADING_ERROR nogil
+cdef cufftResult cufftCreate(cufftHandle* handle) except?_CUFFTRESULT_INTERNAL_LOADING_ERROR nogil
+cdef cufftResult cufftGetSize1d(cufftHandle handle, int nx, cufftType type, int batch, size_t* workSize) except?_CUFFTRESULT_INTERNAL_LOADING_ERROR nogil
+cdef cufftResult cufftGetSize2d(cufftHandle handle, int nx, int ny, cufftType type, size_t* workSize) except?_CUFFTRESULT_INTERNAL_LOADING_ERROR nogil
+cdef cufftResult cufftGetSize3d(cufftHandle handle, int nx, int ny, int nz, cufftType type, size_t* workSize) except?_CUFFTRESULT_INTERNAL_LOADING_ERROR nogil
+cdef cufftResult cufftGetSizeMany(cufftHandle handle, int rank, int* n, int* inembed, int istride, int idist, int* onembed, int ostride, int odist, cufftType type, int batch, size_t* workArea) except?_CUFFTRESULT_INTERNAL_LOADING_ERROR nogil
+cdef cufftResult cufftGetSize(cufftHandle handle, size_t* workSize) except?_CUFFTRESULT_INTERNAL_LOADING_ERROR nogil
+cdef cufftResult cufftSetWorkArea(cufftHandle plan, void* workArea) except?_CUFFTRESULT_INTERNAL_LOADING_ERROR nogil
+cdef cufftResult cufftSetAutoAllocation(cufftHandle plan, int autoAllocate) except?_CUFFTRESULT_INTERNAL_LOADING_ERROR nogil
+cdef cufftResult cufftExecC2C(cufftHandle plan, cufftComplex* idata, cufftComplex* odata, int direction) except?_CUFFTRESULT_INTERNAL_LOADING_ERROR nogil
+cdef cufftResult cufftExecR2C(cufftHandle plan, cufftReal* idata, cufftComplex* odata) except?_CUFFTRESULT_INTERNAL_LOADING_ERROR nogil
+cdef cufftResult cufftExecC2R(cufftHandle plan, cufftComplex* idata, cufftReal* odata) except?_CUFFTRESULT_INTERNAL_LOADING_ERROR nogil
+cdef cufftResult cufftExecZ2Z(cufftHandle plan, cufftDoubleComplex* idata, cufftDoubleComplex* odata, int direction) except?_CUFFTRESULT_INTERNAL_LOADING_ERROR nogil
+cdef cufftResult cufftExecD2Z(cufftHandle plan, cufftDoubleReal* idata, cufftDoubleComplex* odata) except?_CUFFTRESULT_INTERNAL_LOADING_ERROR nogil
+cdef cufftResult cufftExecZ2D(cufftHandle plan, cufftDoubleComplex* idata, cufftDoubleReal* odata) except?_CUFFTRESULT_INTERNAL_LOADING_ERROR nogil
+cdef cufftResult cufftSetStream(cufftHandle plan, cudaStream_t stream) except?_CUFFTRESULT_INTERNAL_LOADING_ERROR nogil
+cdef cufftResult cufftDestroy(cufftHandle plan) except?_CUFFTRESULT_INTERNAL_LOADING_ERROR nogil
+cdef cufftResult cufftGetVersion(int* version) except?_CUFFTRESULT_INTERNAL_LOADING_ERROR nogil
+cdef cufftResult cufftGetProperty(libraryPropertyType type, int* value) except?_CUFFTRESULT_INTERNAL_LOADING_ERROR nogil
+cdef cufftResult cufftSetPlanPropertyInt64(cufftHandle plan, cufftProperty property, const long long int inputValueInt) except?_CUFFTRESULT_INTERNAL_LOADING_ERROR nogil
+cdef cufftResult cufftGetPlanPropertyInt64(cufftHandle plan, cufftProperty property, long long int* returnPtrValue) except?_CUFFTRESULT_INTERNAL_LOADING_ERROR nogil
+cdef cufftResult cufftResetPlanProperty(cufftHandle plan, cufftProperty property) except?_CUFFTRESULT_INTERNAL_LOADING_ERROR nogil
+cdef cufftResult cufftXtSetGPUs(cufftHandle handle, int nGPUs, int* whichGPUs) except?_CUFFTRESULT_INTERNAL_LOADING_ERROR nogil
+cdef cufftResult cufftXtMalloc(cufftHandle plan, cudaLibXtDesc** descriptor, cufftXtSubFormat format) except?_CUFFTRESULT_INTERNAL_LOADING_ERROR nogil
+cdef cufftResult cufftXtMemcpy(cufftHandle plan, void* dstPointer, void* srcPointer, cufftXtCopyType type) except?_CUFFTRESULT_INTERNAL_LOADING_ERROR nogil
+cdef cufftResult cufftXtFree(cudaLibXtDesc* descriptor) except?_CUFFTRESULT_INTERNAL_LOADING_ERROR nogil
+cdef cufftResult cufftXtSetWorkArea(cufftHandle plan, void** workArea) except?_CUFFTRESULT_INTERNAL_LOADING_ERROR nogil
+cdef cufftResult cufftXtExecDescriptorC2C(cufftHandle plan, cudaLibXtDesc* input, cudaLibXtDesc* output, int direction) except?_CUFFTRESULT_INTERNAL_LOADING_ERROR nogil
+cdef cufftResult cufftXtExecDescriptorR2C(cufftHandle plan, cudaLibXtDesc* input, cudaLibXtDesc* output) except?_CUFFTRESULT_INTERNAL_LOADING_ERROR nogil
+cdef cufftResult cufftXtExecDescriptorC2R(cufftHandle plan, cudaLibXtDesc* input, cudaLibXtDesc* output) except?_CUFFTRESULT_INTERNAL_LOADING_ERROR nogil
+cdef cufftResult cufftXtExecDescriptorZ2Z(cufftHandle plan, cudaLibXtDesc* input, cudaLibXtDesc* output, int direction) except?_CUFFTRESULT_INTERNAL_LOADING_ERROR nogil
+cdef cufftResult cufftXtExecDescriptorD2Z(cufftHandle plan, cudaLibXtDesc* input, cudaLibXtDesc* output) except?_CUFFTRESULT_INTERNAL_LOADING_ERROR nogil
+cdef cufftResult cufftXtExecDescriptorZ2D(cufftHandle plan, cudaLibXtDesc* input, cudaLibXtDesc* output) except?_CUFFTRESULT_INTERNAL_LOADING_ERROR nogil
+cdef cufftResult cufftXtQueryPlan(cufftHandle plan, void* queryStruct, cufftXtQueryType queryType) except?_CUFFTRESULT_INTERNAL_LOADING_ERROR nogil
+cdef cufftResult cufftXtClearCallback(cufftHandle plan, cufftXtCallbackType cbType) except?_CUFFTRESULT_INTERNAL_LOADING_ERROR nogil
+cdef cufftResult cufftXtSetCallbackSharedSize(cufftHandle plan, cufftXtCallbackType cbType, size_t sharedSize) except?_CUFFTRESULT_INTERNAL_LOADING_ERROR nogil
+cdef cufftResult cufftXtMakePlanMany(cufftHandle plan, int rank, long long int* n, long long int* inembed, long long int istride, long long int idist, cudaDataType inputtype, long long int* onembed, long long int ostride, long long int odist, cudaDataType outputtype, long long int batch, size_t* workSize, cudaDataType executiontype) except?_CUFFTRESULT_INTERNAL_LOADING_ERROR nogil
+cdef cufftResult cufftXtGetSizeMany(cufftHandle plan, int rank, long long int* n, long long int* inembed, long long int istride, long long int idist, cudaDataType inputtype, long long int* onembed, long long int ostride, long long int odist, cudaDataType outputtype, long long int batch, size_t* workSize, cudaDataType executiontype) except?_CUFFTRESULT_INTERNAL_LOADING_ERROR nogil
+cdef cufftResult cufftXtExec(cufftHandle plan, void* input, void* output, int direction) except?_CUFFTRESULT_INTERNAL_LOADING_ERROR nogil
+cdef cufftResult cufftXtExecDescriptor(cufftHandle plan, cudaLibXtDesc* input, cudaLibXtDesc* output, int direction) except?_CUFFTRESULT_INTERNAL_LOADING_ERROR nogil
+cdef cufftResult cufftXtSetWorkAreaPolicy(cufftHandle plan, cufftXtWorkAreaPolicy policy, size_t* workSize) except?_CUFFTRESULT_INTERNAL_LOADING_ERROR nogil
+cdef cufftResult cufftMpAttachComm(cufftHandle plan, cufftMpCommType comm_type, void* comm_handle) except?_CUFFTRESULT_INTERNAL_LOADING_ERROR nogil
+cdef cufftResult cufftXtSetDistribution(cufftHandle plan, int rank, const long long int* lower_input, const long long int* upper_input, const long long int* lower_output, const long long int* upper_output, const long long int* strides_input, const long long int* strides_output) except?_CUFFTRESULT_INTERNAL_LOADING_ERROR nogil
+cdef cufftResult cufftXtSetSubformatDefault(cufftHandle plan, cufftXtSubFormat subformat_forward, cufftXtSubFormat subformat_inverse) except?_CUFFTRESULT_INTERNAL_LOADING_ERROR nogil
+cdef cufftResult cufftMpCreateReshape(cufftReshapeHandle* handle) except?_CUFFTRESULT_INTERNAL_LOADING_ERROR nogil
+cdef cufftResult cufftMpAttachReshapeComm(cufftReshapeHandle handle, cufftMpCommType comm_type, void* comm_handle) except?_CUFFTRESULT_INTERNAL_LOADING_ERROR nogil
+cdef cufftResult cufftMpGetReshapeSize(cufftReshapeHandle handle, size_t* workspace_size) except?_CUFFTRESULT_INTERNAL_LOADING_ERROR nogil
+cdef cufftResult __cufftMpMakeReshape_11_2(cufftReshapeHandle handle, size_t element_size, int rank, const long long int* lower_input, const long long int* upper_input, const long long int* lower_output, const long long int* upper_output, const long long int* strides_input, const long long int* strides_output) except?_CUFFTRESULT_INTERNAL_LOADING_ERROR nogil
+cdef cufftResult cufftMpExecReshapeAsync(cufftReshapeHandle handle, void* data_out, const void* data_in, void* workspace, cudaStream_t stream) except?_CUFFTRESULT_INTERNAL_LOADING_ERROR nogil
+cdef cufftResult cufftMpDestroyReshape(cufftReshapeHandle handle) except?_CUFFTRESULT_INTERNAL_LOADING_ERROR nogil
+cdef cufftResult __cufftMpMakeReshape_11_4(cufftReshapeHandle handle, size_t element_size, int rank, const long long int* lower_input, const long long int* upper_input, const long long int* strides_input, const long long int* lower_output, const long long int* upper_output, const long long int* strides_output, void* comm_handle, cufftMpCommType comm_type) except?_CUFFTRESULT_INTERNAL_LOADING_ERROR nogil

nvmath/bindings/cycurand.pxd

@@ -0,0 +1,146 @@
+# Copyright (c) 2024-2025, NVIDIA CORPORATION & AFFILIATES. ALL RIGHTS RESERVED.
+#
+# SPDX-License-Identifier: Apache-2.0
+#
+# This code was automatically generated across versions from 11.0.3 to 12.8.0. Do not modify it directly.
+# This layer exposes the C header to Cython as-is.
+
+from libc.stdint cimport int64_t
+from libc.stdio cimport FILE
+
+
+###############################################################################
+# Types (structs, enums, ...)
+###############################################################################
+
+# enums
+ctypedef enum curandStatus "curandStatus":
+    CURAND_STATUS_SUCCESS "CURAND_STATUS_SUCCESS" = 0
+    CURAND_STATUS_VERSION_MISMATCH "CURAND_STATUS_VERSION_MISMATCH" = 100
+    CURAND_STATUS_NOT_INITIALIZED "CURAND_STATUS_NOT_INITIALIZED" = 101
+    CURAND_STATUS_ALLOCATION_FAILED "CURAND_STATUS_ALLOCATION_FAILED" = 102
+    CURAND_STATUS_TYPE_ERROR "CURAND_STATUS_TYPE_ERROR" = 103
+    CURAND_STATUS_OUT_OF_RANGE "CURAND_STATUS_OUT_OF_RANGE" = 104
+    CURAND_STATUS_LENGTH_NOT_MULTIPLE "CURAND_STATUS_LENGTH_NOT_MULTIPLE" = 105
+    CURAND_STATUS_DOUBLE_PRECISION_REQUIRED "CURAND_STATUS_DOUBLE_PRECISION_REQUIRED" = 106
+    CURAND_STATUS_LAUNCH_FAILURE "CURAND_STATUS_LAUNCH_FAILURE" = 201
+    CURAND_STATUS_PREEXISTING_FAILURE "CURAND_STATUS_PREEXISTING_FAILURE" = 202
+    CURAND_STATUS_INITIALIZATION_FAILED "CURAND_STATUS_INITIALIZATION_FAILED" = 203
+    CURAND_STATUS_ARCH_MISMATCH "CURAND_STATUS_ARCH_MISMATCH" = 204
+    CURAND_STATUS_INTERNAL_ERROR "CURAND_STATUS_INTERNAL_ERROR" = 999
+    _CURANDSTATUS_T_INTERNAL_LOADING_ERROR "_CURANDSTATUS_T_INTERNAL_LOADING_ERROR" = -42
+
+ctypedef curandStatus curandStatus_t "curandStatus_t"
+
+ctypedef enum curandRngType "curandRngType":
+    CURAND_RNG_TEST "CURAND_RNG_TEST" = 0
+    CURAND_RNG_PSEUDO_DEFAULT "CURAND_RNG_PSEUDO_DEFAULT" = 100
+    CURAND_RNG_PSEUDO_XORWOW "CURAND_RNG_PSEUDO_XORWOW" = 101
+    CURAND_RNG_PSEUDO_MRG32K3A "CURAND_RNG_PSEUDO_MRG32K3A" = 121
+    CURAND_RNG_PSEUDO_MTGP32 "CURAND_RNG_PSEUDO_MTGP32" = 141
+    CURAND_RNG_PSEUDO_MT19937 "CURAND_RNG_PSEUDO_MT19937" = 142
+    CURAND_RNG_PSEUDO_PHILOX4_32_10 "CURAND_RNG_PSEUDO_PHILOX4_32_10" = 161
+    CURAND_RNG_QUASI_DEFAULT "CURAND_RNG_QUASI_DEFAULT" = 200
+    CURAND_RNG_QUASI_SOBOL32 "CURAND_RNG_QUASI_SOBOL32" = 201
+    CURAND_RNG_QUASI_SCRAMBLED_SOBOL32 "CURAND_RNG_QUASI_SCRAMBLED_SOBOL32" = 202
+    CURAND_RNG_QUASI_SOBOL64 "CURAND_RNG_QUASI_SOBOL64" = 203
+    CURAND_RNG_QUASI_SCRAMBLED_SOBOL64 "CURAND_RNG_QUASI_SCRAMBLED_SOBOL64" = 204
+
+ctypedef curandRngType curandRngType_t "curandRngType_t"
+
+ctypedef enum curandOrdering "curandOrdering":
+    CURAND_ORDERING_PSEUDO_BEST "CURAND_ORDERING_PSEUDO_BEST" = 100
+    CURAND_ORDERING_PSEUDO_DEFAULT "CURAND_ORDERING_PSEUDO_DEFAULT" = 101
+    CURAND_ORDERING_PSEUDO_SEEDED "CURAND_ORDERING_PSEUDO_SEEDED" = 102
+    CURAND_ORDERING_PSEUDO_LEGACY "CURAND_ORDERING_PSEUDO_LEGACY" = 103
+    CURAND_ORDERING_PSEUDO_DYNAMIC "CURAND_ORDERING_PSEUDO_DYNAMIC" = 104
+    CURAND_ORDERING_QUASI_DEFAULT "CURAND_ORDERING_QUASI_DEFAULT" = 201
+
+ctypedef curandOrdering curandOrdering_t "curandOrdering_t"
+
+ctypedef enum curandDirectionVectorSet "curandDirectionVectorSet":
+    CURAND_DIRECTION_VECTORS_32_JOEKUO6 "CURAND_DIRECTION_VECTORS_32_JOEKUO6" = 101
+    CURAND_SCRAMBLED_DIRECTION_VECTORS_32_JOEKUO6 "CURAND_SCRAMBLED_DIRECTION_VECTORS_32_JOEKUO6" = 102
+    CURAND_DIRECTION_VECTORS_64_JOEKUO6 "CURAND_DIRECTION_VECTORS_64_JOEKUO6" = 103
+    CURAND_SCRAMBLED_DIRECTION_VECTORS_64_JOEKUO6 "CURAND_SCRAMBLED_DIRECTION_VECTORS_64_JOEKUO6" = 104
+
+ctypedef curandDirectionVectorSet curandDirectionVectorSet_t "curandDirectionVectorSet_t"
+
+ctypedef enum curandMethod "curandMethod":
+    CURAND_CHOOSE_BEST "CURAND_CHOOSE_BEST" = 0
+    CURAND_ITR "CURAND_ITR" = 1
+    CURAND_KNUTH "CURAND_KNUTH" = 2
+    CURAND_HITR "CURAND_HITR" = 3
+    CURAND_M1 "CURAND_M1" = 4
+    CURAND_M2 "CURAND_M2" = 5
+    CURAND_BINARY_SEARCH "CURAND_BINARY_SEARCH" = 6
+    CURAND_DISCRETE_GAUSS "CURAND_DISCRETE_GAUSS" = 7
+    CURAND_REJECTION "CURAND_REJECTION" = 8
+    CURAND_DEVICE_API "CURAND_DEVICE_API" = 9
+    CURAND_FAST_REJECTION "CURAND_FAST_REJECTION" = 10
+    CURAND_3RD "CURAND_3RD" = 11
+    CURAND_DEFINITION "CURAND_DEFINITION" = 12
+    CURAND_POISSON "CURAND_POISSON" = 13
+
+ctypedef curandMethod curandMethod_t "curandMethod_t"
+
+
+# types
+cdef extern from *:
+    """
+    #include <driver_types.h>
+    #include <library_types.h>
+    #include <cuComplex.h>
+    """
+    ctypedef void* cudaStream_t 'cudaStream_t'
+    ctypedef int cudaDataType_t 'cudaDataType_t'
+    ctypedef int cudaDataType 'cudaDataType'
+    ctypedef int libraryPropertyType_t 'libraryPropertyType_t'
+    ctypedef int libraryPropertyType 'libraryPropertyType'
+
+
+ctypedef void* curandGenerator_t 'curandGenerator_t'
+ctypedef void* curandDistribution_t 'curandDistribution_t'
+ctypedef void* curandDistributionShift_t 'curandDistributionShift_t'
+ctypedef void* curandDistributionM2Shift_t 'curandDistributionM2Shift_t'
+ctypedef void* curandHistogramM2_t 'curandHistogramM2_t'
+ctypedef void* curandHistogramM2K_t 'curandHistogramM2K_t'
+ctypedef void* curandHistogramM2V_t 'curandHistogramM2V_t'
+ctypedef void* curandDiscreteDistribution_t 'curandDiscreteDistribution_t'
+ctypedef unsigned int curandDirectionVectors32_t[32]
+ctypedef unsigned long long curandDirectionVectors64_t[64]
+
+
+###############################################################################
+# Functions
+###############################################################################
+
+cdef curandStatus_t curandCreateGenerator(curandGenerator_t* generator, curandRngType_t rng_type) except?_CURANDSTATUS_T_INTERNAL_LOADING_ERROR nogil
+cdef curandStatus_t curandCreateGeneratorHost(curandGenerator_t* generator, curandRngType_t rng_type) except?_CURANDSTATUS_T_INTERNAL_LOADING_ERROR nogil
+cdef curandStatus_t curandDestroyGenerator(curandGenerator_t generator) except?_CURANDSTATUS_T_INTERNAL_LOADING_ERROR nogil
+cdef curandStatus_t curandGetVersion(int* version) except?_CURANDSTATUS_T_INTERNAL_LOADING_ERROR nogil
+cdef curandStatus_t curandGetProperty(libraryPropertyType type, int* value) except?_CURANDSTATUS_T_INTERNAL_LOADING_ERROR nogil
+cdef curandStatus_t curandSetStream(curandGenerator_t generator, cudaStream_t stream) except?_CURANDSTATUS_T_INTERNAL_LOADING_ERROR nogil
+cdef curandStatus_t curandSetPseudoRandomGeneratorSeed(curandGenerator_t generator, unsigned long long seed) except?_CURANDSTATUS_T_INTERNAL_LOADING_ERROR nogil
+cdef curandStatus_t curandSetGeneratorOffset(curandGenerator_t generator, unsigned long long offset) except?_CURANDSTATUS_T_INTERNAL_LOADING_ERROR nogil
+cdef curandStatus_t curandSetGeneratorOrdering(curandGenerator_t generator, curandOrdering_t order) except?_CURANDSTATUS_T_INTERNAL_LOADING_ERROR nogil
+cdef curandStatus_t curandSetQuasiRandomGeneratorDimensions(curandGenerator_t generator, unsigned int num_dimensions) except?_CURANDSTATUS_T_INTERNAL_LOADING_ERROR nogil
+cdef curandStatus_t curandGenerate(curandGenerator_t generator, unsigned int* outputPtr, size_t num) except?_CURANDSTATUS_T_INTERNAL_LOADING_ERROR nogil
+cdef curandStatus_t curandGenerateLongLong(curandGenerator_t generator, unsigned long long* outputPtr, size_t num) except?_CURANDSTATUS_T_INTERNAL_LOADING_ERROR nogil
+cdef curandStatus_t curandGenerateUniform(curandGenerator_t generator, float* outputPtr, size_t num) except?_CURANDSTATUS_T_INTERNAL_LOADING_ERROR nogil
+cdef curandStatus_t curandGenerateUniformDouble(curandGenerator_t generator, double* outputPtr, size_t num) except?_CURANDSTATUS_T_INTERNAL_LOADING_ERROR nogil
+cdef curandStatus_t curandGenerateNormal(curandGenerator_t generator, float* outputPtr, size_t n, float mean, float stddev) except?_CURANDSTATUS_T_INTERNAL_LOADING_ERROR nogil
+cdef curandStatus_t curandGenerateNormalDouble(curandGenerator_t generator, double* outputPtr, size_t n, double mean, double stddev) except?_CURANDSTATUS_T_INTERNAL_LOADING_ERROR nogil
+cdef curandStatus_t curandGenerateLogNormal(curandGenerator_t generator, float* outputPtr, size_t n, float mean, float stddev) except?_CURANDSTATUS_T_INTERNAL_LOADING_ERROR nogil
+cdef curandStatus_t curandGenerateLogNormalDouble(curandGenerator_t generator, double* outputPtr, size_t n, double mean, double stddev) except?_CURANDSTATUS_T_INTERNAL_LOADING_ERROR nogil
+cdef curandStatus_t curandCreatePoissonDistribution(double lambda_, curandDiscreteDistribution_t* discrete_distribution) except?_CURANDSTATUS_T_INTERNAL_LOADING_ERROR nogil
+cdef curandStatus_t curandDestroyDistribution(curandDiscreteDistribution_t discrete_distribution) except?_CURANDSTATUS_T_INTERNAL_LOADING_ERROR nogil
+cdef curandStatus_t curandGeneratePoisson(curandGenerator_t generator, unsigned int* outputPtr, size_t n, double lambda_) except?_CURANDSTATUS_T_INTERNAL_LOADING_ERROR nogil
+cdef curandStatus_t curandGeneratePoissonMethod(curandGenerator_t generator, unsigned int* outputPtr, size_t n, double lambda_, curandMethod_t method) except?_CURANDSTATUS_T_INTERNAL_LOADING_ERROR nogil
+cdef curandStatus_t curandGenerateBinomial(curandGenerator_t generator, unsigned int* outputPtr, size_t num, unsigned int n, double p) except?_CURANDSTATUS_T_INTERNAL_LOADING_ERROR nogil
+cdef curandStatus_t curandGenerateBinomialMethod(curandGenerator_t generator, unsigned int* outputPtr, size_t num, unsigned int n, double p, curandMethod_t method) except?_CURANDSTATUS_T_INTERNAL_LOADING_ERROR nogil
+cdef curandStatus_t curandGenerateSeeds(curandGenerator_t generator) except?_CURANDSTATUS_T_INTERNAL_LOADING_ERROR nogil
+cdef curandStatus_t curandGetDirectionVectors32(curandDirectionVectors32_t* vectors[], curandDirectionVectorSet_t set) except?_CURANDSTATUS_T_INTERNAL_LOADING_ERROR nogil
+cdef curandStatus_t curandGetScrambleConstants32(unsigned int** constants) except?_CURANDSTATUS_T_INTERNAL_LOADING_ERROR nogil
+cdef curandStatus_t curandGetDirectionVectors64(curandDirectionVectors64_t* vectors[], curandDirectionVectorSet_t set) except?_CURANDSTATUS_T_INTERNAL_LOADING_ERROR nogil
+cdef curandStatus_t curandGetScrambleConstants64(unsigned long long** constants) except?_CURANDSTATUS_T_INTERNAL_LOADING_ERROR nogil

nvmath/bindings/cycusolver.pxd

@@ -0,0 +1,143 @@
+# Copyright (c) 2024-2025, NVIDIA CORPORATION & AFFILIATES. ALL RIGHTS RESERVED.
+#
+# SPDX-License-Identifier: Apache-2.0
+#
+# This code was automatically generated across versions from 11.0.3 to 12.8.0. Do not modify it directly.
+# This layer exposes the C header to Cython as-is.
+
+from libc.stdint cimport int64_t
+
+from .cycublas cimport (cublasFillMode_t, cublasSideMode_t, cublasOperation_t,
+                        cublasDiagType_t,)
+
+
+###############################################################################
+# Types (structs, enums, ...)
+###############################################################################
+
+# enums
+ctypedef enum cusolverStatus_t "cusolverStatus_t":
+    CUSOLVER_STATUS_SUCCESS "CUSOLVER_STATUS_SUCCESS" = 0
+    CUSOLVER_STATUS_NOT_INITIALIZED "CUSOLVER_STATUS_NOT_INITIALIZED" = 1
+    CUSOLVER_STATUS_ALLOC_FAILED "CUSOLVER_STATUS_ALLOC_FAILED" = 2
+    CUSOLVER_STATUS_INVALID_VALUE "CUSOLVER_STATUS_INVALID_VALUE" = 3
+    CUSOLVER_STATUS_ARCH_MISMATCH "CUSOLVER_STATUS_ARCH_MISMATCH" = 4
+    CUSOLVER_STATUS_MAPPING_ERROR "CUSOLVER_STATUS_MAPPING_ERROR" = 5
+    CUSOLVER_STATUS_EXECUTION_FAILED "CUSOLVER_STATUS_EXECUTION_FAILED" = 6
+    CUSOLVER_STATUS_INTERNAL_ERROR "CUSOLVER_STATUS_INTERNAL_ERROR" = 7
+    CUSOLVER_STATUS_MATRIX_TYPE_NOT_SUPPORTED "CUSOLVER_STATUS_MATRIX_TYPE_NOT_SUPPORTED" = 8
+    CUSOLVER_STATUS_NOT_SUPPORTED "CUSOLVER_STATUS_NOT_SUPPORTED" = 9
+    CUSOLVER_STATUS_ZERO_PIVOT "CUSOLVER_STATUS_ZERO_PIVOT" = 10
+    CUSOLVER_STATUS_INVALID_LICENSE "CUSOLVER_STATUS_INVALID_LICENSE" = 11
+    CUSOLVER_STATUS_IRS_PARAMS_NOT_INITIALIZED "CUSOLVER_STATUS_IRS_PARAMS_NOT_INITIALIZED" = 12
+    CUSOLVER_STATUS_IRS_PARAMS_INVALID "CUSOLVER_STATUS_IRS_PARAMS_INVALID" = 13
+    CUSOLVER_STATUS_IRS_PARAMS_INVALID_PREC "CUSOLVER_STATUS_IRS_PARAMS_INVALID_PREC" = 14
+    CUSOLVER_STATUS_IRS_PARAMS_INVALID_REFINE "CUSOLVER_STATUS_IRS_PARAMS_INVALID_REFINE" = 15
+    CUSOLVER_STATUS_IRS_PARAMS_INVALID_MAXITER "CUSOLVER_STATUS_IRS_PARAMS_INVALID_MAXITER" = 16
+    CUSOLVER_STATUS_IRS_INTERNAL_ERROR "CUSOLVER_STATUS_IRS_INTERNAL_ERROR" = 20
+    CUSOLVER_STATUS_IRS_NOT_SUPPORTED "CUSOLVER_STATUS_IRS_NOT_SUPPORTED" = 21
+    CUSOLVER_STATUS_IRS_OUT_OF_RANGE "CUSOLVER_STATUS_IRS_OUT_OF_RANGE" = 22
+    CUSOLVER_STATUS_IRS_NRHS_NOT_SUPPORTED_FOR_REFINE_GMRES "CUSOLVER_STATUS_IRS_NRHS_NOT_SUPPORTED_FOR_REFINE_GMRES" = 23
+    CUSOLVER_STATUS_IRS_INFOS_NOT_INITIALIZED "CUSOLVER_STATUS_IRS_INFOS_NOT_INITIALIZED" = 25
+    CUSOLVER_STATUS_IRS_INFOS_NOT_DESTROYED "CUSOLVER_STATUS_IRS_INFOS_NOT_DESTROYED" = 26
+    CUSOLVER_STATUS_IRS_MATRIX_SINGULAR "CUSOLVER_STATUS_IRS_MATRIX_SINGULAR" = 30
+    CUSOLVER_STATUS_INVALID_WORKSPACE "CUSOLVER_STATUS_INVALID_WORKSPACE" = 31
+    _CUSOLVERSTATUS_T_INTERNAL_LOADING_ERROR "_CUSOLVERSTATUS_T_INTERNAL_LOADING_ERROR" = -42
+
+ctypedef enum cusolverEigType_t "cusolverEigType_t":
+    CUSOLVER_EIG_TYPE_1 "CUSOLVER_EIG_TYPE_1" = 1
+    CUSOLVER_EIG_TYPE_2 "CUSOLVER_EIG_TYPE_2" = 2
+    CUSOLVER_EIG_TYPE_3 "CUSOLVER_EIG_TYPE_3" = 3
+
+ctypedef enum cusolverEigMode_t "cusolverEigMode_t":
+    CUSOLVER_EIG_MODE_NOVECTOR "CUSOLVER_EIG_MODE_NOVECTOR" = 0
+    CUSOLVER_EIG_MODE_VECTOR "CUSOLVER_EIG_MODE_VECTOR" = 1
+
+ctypedef enum cusolverEigRange_t "cusolverEigRange_t":
+    CUSOLVER_EIG_RANGE_ALL "CUSOLVER_EIG_RANGE_ALL" = 1001
+    CUSOLVER_EIG_RANGE_I "CUSOLVER_EIG_RANGE_I" = 1002
+    CUSOLVER_EIG_RANGE_V "CUSOLVER_EIG_RANGE_V" = 1003
+
+ctypedef enum cusolverNorm_t "cusolverNorm_t":
+    CUSOLVER_INF_NORM "CUSOLVER_INF_NORM" = 104
+    CUSOLVER_MAX_NORM "CUSOLVER_MAX_NORM" = 105
+    CUSOLVER_ONE_NORM "CUSOLVER_ONE_NORM" = 106
+    CUSOLVER_FRO_NORM "CUSOLVER_FRO_NORM" = 107
+
+ctypedef enum cusolverIRSRefinement_t "cusolverIRSRefinement_t":
+    CUSOLVER_IRS_REFINE_NOT_SET "CUSOLVER_IRS_REFINE_NOT_SET" = 1100
+    CUSOLVER_IRS_REFINE_NONE "CUSOLVER_IRS_REFINE_NONE" = 1101
+    CUSOLVER_IRS_REFINE_CLASSICAL "CUSOLVER_IRS_REFINE_CLASSICAL" = 1102
+    CUSOLVER_IRS_REFINE_CLASSICAL_GMRES "CUSOLVER_IRS_REFINE_CLASSICAL_GMRES" = 1103
+    CUSOLVER_IRS_REFINE_GMRES "CUSOLVER_IRS_REFINE_GMRES" = 1104
+    CUSOLVER_IRS_REFINE_GMRES_GMRES "CUSOLVER_IRS_REFINE_GMRES_GMRES" = 1105
+    CUSOLVER_IRS_REFINE_GMRES_NOPCOND "CUSOLVER_IRS_REFINE_GMRES_NOPCOND" = 1106
+    CUSOLVER_PREC_DD "CUSOLVER_PREC_DD" = 1150
+    CUSOLVER_PREC_SS "CUSOLVER_PREC_SS" = 1151
+    CUSOLVER_PREC_SHT "CUSOLVER_PREC_SHT" = 1152
+
+ctypedef enum cusolverPrecType_t "cusolverPrecType_t":
+    CUSOLVER_R_8I "CUSOLVER_R_8I" = 1201
+    CUSOLVER_R_8U "CUSOLVER_R_8U" = 1202
+    CUSOLVER_R_64F "CUSOLVER_R_64F" = 1203
+    CUSOLVER_R_32F "CUSOLVER_R_32F" = 1204
+    CUSOLVER_R_16F "CUSOLVER_R_16F" = 1205
+    CUSOLVER_R_16BF "CUSOLVER_R_16BF" = 1206
+    CUSOLVER_R_TF32 "CUSOLVER_R_TF32" = 1207
+    CUSOLVER_R_AP "CUSOLVER_R_AP" = 1208
+    CUSOLVER_C_8I "CUSOLVER_C_8I" = 1211
+    CUSOLVER_C_8U "CUSOLVER_C_8U" = 1212
+    CUSOLVER_C_64F "CUSOLVER_C_64F" = 1213
+    CUSOLVER_C_32F "CUSOLVER_C_32F" = 1214
+    CUSOLVER_C_16F "CUSOLVER_C_16F" = 1215
+    CUSOLVER_C_16BF "CUSOLVER_C_16BF" = 1216
+    CUSOLVER_C_TF32 "CUSOLVER_C_TF32" = 1217
+    CUSOLVER_C_AP "CUSOLVER_C_AP" = 1218
+
+ctypedef enum cusolverAlgMode_t "cusolverAlgMode_t":
+    CUSOLVER_ALG_0 "CUSOLVER_ALG_0" = 0
+    CUSOLVER_ALG_1 "CUSOLVER_ALG_1" = 1
+    CUSOLVER_ALG_2 "CUSOLVER_ALG_2" = 2
+
+ctypedef enum cusolverStorevMode_t "cusolverStorevMode_t":
+    CUBLAS_STOREV_COLUMNWISE "CUBLAS_STOREV_COLUMNWISE" = 0
+    CUBLAS_STOREV_ROWWISE "CUBLAS_STOREV_ROWWISE" = 1
+
+ctypedef enum cusolverDirectMode_t "cusolverDirectMode_t":
+    CUBLAS_DIRECT_FORWARD "CUBLAS_DIRECT_FORWARD" = 0
+    CUBLAS_DIRECT_BACKWARD "CUBLAS_DIRECT_BACKWARD" = 1
+
+ctypedef enum cusolverDeterministicMode_t "cusolverDeterministicMode_t":
+    CUSOLVER_DETERMINISTIC_RESULTS "CUSOLVER_DETERMINISTIC_RESULTS" = 1
+    CUSOLVER_ALLOW_NON_DETERMINISTIC_RESULTS "CUSOLVER_ALLOW_NON_DETERMINISTIC_RESULTS" = 2
+
+
+# types
+cdef extern from *:
+    """
+    #include <driver_types.h>
+    #include <library_types.h>
+    #include <cuComplex.h>
+    """
+    ctypedef void* cudaStream_t 'cudaStream_t'
+    ctypedef int cudaDataType_t 'cudaDataType_t'
+    ctypedef int cudaDataType 'cudaDataType'
+    ctypedef int libraryPropertyType_t 'libraryPropertyType_t'
+    ctypedef int libraryPropertyType 'libraryPropertyType'
+
+    ctypedef struct cuComplex:
+        pass
+    ctypedef struct cuDoubleComplex:
+        pass
+
+
+
+ctypedef int cusolver_int_t 'cusolver_int_t'
+
+
+###############################################################################
+# Functions
+###############################################################################
+
+cdef cusolverStatus_t cusolverGetProperty(libraryPropertyType type, int* value) except?_CUSOLVERSTATUS_T_INTERNAL_LOADING_ERROR nogil
+cdef cusolverStatus_t cusolverGetVersion(int* version) except?_CUSOLVERSTATUS_T_INTERNAL_LOADING_ERROR nogil