cuSPARSELt (nvmath.bindings.cusparseLt)

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

Enums and constants

ComputeType(value)	See cusparseComputeType.
MatDescAttribute(value)	See cusparseLtMatDescAttribute_t.
MatmulAlg(value)	See cusparseLtMatmulAlg_t.
MatmulAlgAttribute(value)	See cusparseLtMatmulAlgAttribute_t.
MatmulDescAttribute(value)	See cusparseLtMatmulDescAttribute_t.
MatmulMatrixScale(value)	See cusparseLtMatmulMatrixScale_t.
PruneAlg(value)	See cusparseLtPruneAlg_t.
Sparsity(value)	See cusparseLtSparsity_t.
SplitKMode(value)	See cusparseLtSplitKMode_t.
cuSPARSELtError(status)

Functions

check_status(int status)
dense_descriptor_init(intptr_t handle, ...)
destroy(intptr_t handle_ptr)
get_error_name(cusparseStatus_t status)	See cusparseLtGetErrorName.
get_error_string(cusparseStatus_t status)	See cusparseLtGetErrorString.
get_mat_desc_attribute_dtype(int attr)	Get the Python data type of the corresponding MatDescAttribute attribute.
get_matmul_alg_attribute_dtype(int attr)	Get the Python data type of the corresponding MatmulAlgAttribute attribute.
get_matmul_desc_attribute_dtype(int attr)	Get the Python data type of the corresponding MatmulDescAttribute attribute.
get_property(int property_type)	See cusparseLtGetProperty.
get_version(intptr_t handle)	See cusparseLtGetVersion.
init()
mat_desc_get_attribute(intptr_t handle, ...)	See cusparseLtMatDescGetAttribute.
mat_desc_set_attribute(intptr_t handle, ...)	See cusparseLtMatDescSetAttribute.
mat_descriptor_destroy(intptr_t mat_descr_ptr)
matmul(intptr_t handle, intptr_t plan, ...)	See cusparseLtMatmul.
matmul_alg_get_attribute(intptr_t handle, ...)	See cusparseLtMatmulAlgGetAttribute.
matmul_alg_selection_destroy(...)
matmul_alg_selection_init(intptr_t handle, ...)
matmul_alg_set_attribute(intptr_t handle, ...)	See cusparseLtMatmulAlgSetAttribute.
matmul_desc_get_attribute(intptr_t handle, ...)	See cusparseLtMatmulDescGetAttribute.
matmul_desc_set_attribute(intptr_t handle, ...)	See cusparseLtMatmulDescSetAttribute.
matmul_descriptor_destroy(...)
matmul_descriptor_init(intptr_t handle, ...)
matmul_get_workspace(intptr_t handle, ...)	See cusparseLtMatmulGetWorkspace.
matmul_plan_destroy(intptr_t plan_ptr)
matmul_plan_init(intptr_t handle, ...)
matmul_search(intptr_t handle, ...)	See cusparseLtMatmulSearch.
sp_mma_compress(intptr_t handle, ...)	See cusparseLtSpMMACompress.
sp_mma_compress2(intptr_t handle, ...)	See cusparseLtSpMMACompress2.
sp_mma_compressed_size(intptr_t handle, ...)	See cusparseLtSpMMACompressedSize.
sp_mma_compressed_size2(intptr_t handle, ...)	See cusparseLtSpMMACompressedSize2.
sp_mma_prune(intptr_t handle, ...)	See cusparseLtSpMMAPrune.
sp_mma_prune2(intptr_t handle, ...)	See cusparseLtSpMMAPrune2.
sp_mma_prune_check(intptr_t handle, ...)	See cusparseLtSpMMAPruneCheck.
sp_mma_prune_check2(intptr_t handle, ...)	See cusparseLtSpMMAPruneCheck2.
structured_descriptor_init(intptr_t handle, ...)