cuBLASMp (nvmath.bindings.cublasMp)#

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

Enums and constants#

EmulationStrategy(value)

See cublasMpEmulationStrategy_t.

GridLayout(value)

See cublasMpGridLayout_t.

MatmulAlgoType(value)

See cublasMpMatmulAlgoType_t.

MatmulDescriptorAttribute(value)

See cublasMpMatmulDescriptorAttribute_t.

MatmulEpilogue(value)

See cublasMpMatmulEpilogue_t.

MatmulMatrixScale(value)

See cublasMpMatmulMatrixScale_t.

Status(value)

See cublasMpStatus_t.

cuBLASMpError(status)

Functions#

create(intptr_t stream)

See cublasMpCreate.

destroy(intptr_t handle)

See cublasMpDestroy.

set_stream(intptr_t handle, intptr_t stream)

See cublasMpSetStream.

get_stream(intptr_t handle)

See cublasMpGetStream.

get_version()

See cublasMpGetVersion.

get_status_string(int status)

See cublasMpGetStatusString.

set_emulation_strategy(intptr_t handle, ...)

See cublasMpSetEmulationStrategy.

get_emulation_strategy(intptr_t handle)

See cublasMpGetEmulationStrategy.

grid_create(int64_t nprow, int64_t npcol, ...)

See cublasMpGridCreate.

grid_destroy(intptr_t grid)

See cublasMpGridDestroy.

malloc(intptr_t grid, size_t size)

See cublasMpMalloc.

free(intptr_t grid, intptr_t ptr)

See cublasMpFree.

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

See cublasMpBufferRegister.

buffer_deregister(intptr_t grid, intptr_t ptr)

See cublasMpBufferDeregister.

matrix_descriptor_create(int64_t m, ...)

See cublasMpMatrixDescriptorCreate.

matrix_descriptor_init(int64_t m, int64_t n, ...)

See cublasMpMatrixDescriptorInit.

matrix_descriptor_destroy(intptr_t desc)

See cublasMpMatrixDescriptorDestroy.

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

See cublasMpNumroc.

matmul_descriptor_create(int compute_type)

See cublasMpMatmulDescriptorCreate.

matmul_descriptor_init(intptr_t matmul_desc, ...)

See cublasMpMatmulDescriptorInit.

matmul_descriptor_destroy(intptr_t matmul_desc)

See cublasMpMatmulDescriptorDestroy.

matmul_descriptor_set_attribute(...)

See cublasMpMatmulDescriptorSetAttribute.

matmul_descriptor_get_attribute(...)

See cublasMpMatmulDescriptorGetAttribute.

matmul_buffer_size(intptr_t handle, ...)

See cublasMpMatmul_bufferSize.

matmul(intptr_t handle, ...)

See cublasMpMatmul.

geadd_buffer_size(intptr_t handle, ...)

See cublasMpGeadd_bufferSize.

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

See cublasMpGeadd.

gemm_buffer_size(intptr_t handle, ...)

See cublasMpGemm_bufferSize.

gemm(intptr_t handle, int trans_a, ...)

See cublasMpGemm.

gemr2d_buffer_size(intptr_t handle, ...)

See cublasMpGemr2D_bufferSize.

gemr2d(intptr_t handle, int64_t m, ...)

See cublasMpGemr2D.

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

See cublasMpSyrk_bufferSize.

syrk(intptr_t handle, int uplo, int trans, ...)

See cublasMpSyrk.

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

See cublasMpTradd_bufferSize.

tradd(intptr_t handle, int uplo, int trans, ...)

See cublasMpTradd.

trmr2d_buffer_size(intptr_t handle, ...)

See cublasMpTrmr2D_bufferSize.

trmr2d(intptr_t handle, int uplo, int diag, ...)

See cublasMpTrmr2D.

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

See cublasMpTrsm_bufferSize.

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

See cublasMpTrsm.