cuBLASMp (nvmath.bindings.cublasMp)#

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

Enums and constants#

ComputeType(value[, names, module, ...])

See cublasComputeType_t.

cuBLASMpError(status)

GridLayout(value[, names, module, qualname, ...])

See cublasMpGridLayout_t.

MatmulAlgoType(value[, names, module, ...])

See cublasMpMatmulAlgoType_t.

MatmulDescriptorAttribute(value[, names, ...])

See cublasMpMatmulDescriptorAttribute_t.

MatmulEpilogue(value[, names, module, ...])

See cublasMpMatmulEpilogue_t.

MatmulMatrixScale(value[, names, module, ...])

See cublasMpMatmulMatrixScale_t.

Operation(value[, names, module, qualname, ...])

See cublasOperation_t.

Status(value[, names, module, qualname, ...])

See cublasMpStatus_t.

Functions#

create(intptr_t stream)

See cublasMpCreate.

destroy(intptr_t handle)

See cublasMpDestroy.

stream_set(intptr_t handle, intptr_t stream)

See cublasMpStreamSet.

get_version()

See cublasMpGetVersion.

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

See cublasMpGridCreate.

grid_destroy(intptr_t grid)

See cublasMpGridDestroy.

matrix_descriptor_create(int64_t m, ...)

See cublasMpMatrixDescriptorCreate.

matrix_descriptor_destroy(intptr_t desc)

See cublasMpMatrixDescriptorDestroy.

matmul_descriptor_create(int compute_type)

See cublasMpMatmulDescriptorCreate.

matmul_descriptor_destroy(intptr_t matmul_desc)

See cublasMpMatmulDescriptorDestroy.

matmul_descriptor_attribute_set(...)

See cublasMpMatmulDescriptorAttributeSet.

matmul_descriptor_attribute_get(...)

See cublasMpMatmulDescriptorAttributeGet.

matmul_buffer_size(intptr_t handle, ...)

See cublasMpMatmul_bufferSize.

matmul(intptr_t handle, ...)

See cublasMpMatmul.

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

See cublasMpNumroc.