The nvmath.bindings module is experimental and potentially subject to future changes.

cuSOLVERMp (nvmath.bindings.cusolverMp)#

For detailed documentation on the original C APIs, refer to the cuSOLVERMp documentation.

Enums and constants#

GridMapping(*values)

See cusolverMpGridMapping_t.

NewtonSchulzDescriptorAttribute(*values)

See cusolverMpNewtonSchulzDescriptorAttribute_t.

cuSOLVERMpError(status)

Functions#

buffer_deregister(intptr_t grid, intptr_t ptr)

See cusolverMpBufferDeregister.

buffer_register(intptr_t grid, intptr_t ptr, ...)

See cusolverMpBufferRegister.

create(int device_id, intptr_t stream)

See cusolverMpCreate.

create_device_grid(intptr_t handle, ...)

See cusolverMpCreateDeviceGrid.

create_matrix_desc(intptr_t grid, ...)

See cusolverMpCreateMatrixDesc.

destroy(intptr_t handle)

See cusolverMpDestroy.

destroy_grid(intptr_t grid)

See cusolverMpDestroyGrid.

destroy_matrix_desc(intptr_t desc)

See cusolverMpDestroyMatrixDesc.

gels(intptr_t handle, int trans, int64_t m, ...)

See cusolverMpGels.

gels_buffer_size(intptr_t handle, int trans, ...)

See cusolverMpGels_bufferSize.

geqrf(intptr_t handle, int64_t m, int64_t n, ...)

See cusolverMpGeqrf.

geqrf_buffer_size(intptr_t handle, ...)

See cusolverMpGeqrf_bufferSize.

get_emulation_strategy(intptr_t handle)

See cusolverMpGetEmulationStrategy.

get_math_mode(intptr_t handle)

See cusolverMpGetMathMode.

get_newton_schulz_descriptor_attribute_dtype(...)

Get the Python data type of the corresponding NewtonSchulzDescriptorAttribute attribute.

get_stream(intptr_t handle)

See cusolverMpGetStream.

get_version(intptr_t handle)

See cusolverMpGetVersion.

getrf(intptr_t handle, int64_t m, int64_t n, ...)

See cusolverMpGetrf.

getrf_buffer_size(intptr_t handle, ...)

See cusolverMpGetrf_bufferSize.

getrs(intptr_t handle, int trans, int64_t n, ...)

See cusolverMpGetrs.

getrs_buffer_size(intptr_t handle, ...)

See cusolverMpGetrs_bufferSize.

laset(intptr_t handle, int uplo, int64_t m, ...)

See cusolverMpLaset.

logger_force_disable()

See cusolverMpLoggerForceDisable.

logger_open_file(intptr_t log_file)

See cusolverMpLoggerOpenFile.

logger_set_file(intptr_t file)

See cusolverMpLoggerSetFile.

logger_set_level(int level)

See cusolverMpLoggerSetLevel.

logger_set_mask(int mask)

See cusolverMpLoggerSetMask.

matrix_gather_d2h(intptr_t handle, ...)

See cusolverMpMatrixGatherD2H.

matrix_scatter_h2d(intptr_t handle, ...)

See cusolverMpMatrixScatterH2D.

newton_schulz(intptr_t handle, ...)

See cusolverMpNewtonSchulz.

newton_schulz_buffer_size(intptr_t handle, ...)

See cusolverMpNewtonSchulz_bufferSize.

newton_schulz_descriptor_create()

See cusolverMpNewtonSchulzDescriptorCreate.

newton_schulz_descriptor_destroy(...)

See cusolverMpNewtonSchulzDescriptorDestroy.

newton_schulz_descriptor_get_attribute(...)

See cusolverMpNewtonSchulzDescriptorGetAttribute.

newton_schulz_descriptor_set_attribute(...)

See cusolverMpNewtonSchulzDescriptorSetAttribute.

numroc(int64_t n, int64_t nb, ...)

See cusolverMpNUMROC.

orgqr(intptr_t handle, int64_t m, int64_t n, ...)

See cusolverMpOrgqr.

orgqr_buffer_size(intptr_t handle, ...)

See cusolverMpOrgqr_bufferSize.

ormqr(intptr_t handle, int side, int trans, ...)

See cusolverMpOrmqr.

ormqr_buffer_size(intptr_t handle, int side, ...)

See cusolverMpOrmqr_bufferSize.

ormtr(intptr_t handle, int side, int uplo, ...)

See cusolverMpOrmtr.

ormtr_buffer_size(intptr_t handle, int side, ...)

See cusolverMpOrmtr_bufferSize.

potrf(intptr_t handle, int uplo, int64_t n, ...)

See cusolverMpPotrf.

potrf_buffer_size(intptr_t handle, int uplo, ...)

See cusolverMpPotrf_bufferSize.

potrs(intptr_t handle, int uplo, int64_t n, ...)

See cusolverMpPotrs.

potrs_buffer_size(intptr_t handle, int uplo, ...)

See cusolverMpPotrs_bufferSize.

set_emulation_strategy(intptr_t handle, ...)

See cusolverMpSetEmulationStrategy.

set_math_mode(intptr_t handle, int mode)

See cusolverMpSetMathMode.

set_stream(intptr_t handle, intptr_t stream)

See cusolverMpSetStream.

stedc(intptr_t handle, intptr_t compz, ...)

See cusolverMpStedc.

stedc_buffer_size(intptr_t handle, ...)

See cusolverMpStedc_bufferSize.

syevd(intptr_t handle, intptr_t compz, ...)

See cusolverMpSyevd.

syevd_buffer_size(intptr_t handle, ...)

See cusolverMpSyevd_bufferSize.

sygst(intptr_t handle, int ibtype, int uplo, ...)

See cusolverMpSygst.

sygst_buffer_size(intptr_t handle, ...)

See cusolverMpSygst_bufferSize.

sygvd(intptr_t handle, int ibtype, int jobz, ...)

See cusolverMpSygvd.

sygvd_buffer_size(intptr_t handle, ...)

See cusolverMpSygvd_bufferSize.

sytrd(intptr_t handle, int uplo, int64_t n, ...)

See cusolverMpSytrd.

sytrd_buffer_size(intptr_t handle, int uplo, ...)

See cusolverMpSytrd_bufferSize.