cuBLASLt (nvmath.bindings.cublaslt)

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

Enums and constants

MatmulTile(value[, names, module, qualname, ...])	See cublasLtMatmulTile_t.
MatmulStages(value[, names, module, ...])	See cublasLtMatmulStages_t.
PointerMode(value[, names, module, ...])	See cublasLtPointerMode_t.
PointerModeMask(value[, names, module, ...])	See cublasLtPointerModeMask_t.
Order(value[, names, module, qualname, ...])	See cublasLtOrder_t.
MatrixLayoutAttribute(value[, names, ...])	See cublasLtMatrixLayoutAttribute_t.
MatmulDescAttribute(value[, names, module, ...])	See cublasLtMatmulDescAttributes_t.
MatrixTransformDescAttribute(value[, names, ...])	See cublasLtMatrixTransformDescAttributes_t.
ReductionScheme(value[, names, module, ...])	See cublasLtReductionScheme_t.
Epilogue(value[, names, module, qualname, ...])	See cublasLtEpilogue_t.
MatmulSearch(value[, names, module, ...])	See cublasLtMatmulSearch_t.
MatmulPreferenceAttribute(value[, names, ...])	See cublasLtMatmulPreferenceAttributes_t.
MatmulAlgoCapAttribute(value[, names, ...])	See cublasLtMatmulAlgoCapAttributes_t.
MatmulAlgoConfigAttribute(value[, names, ...])	See cublasLtMatmulAlgoConfigAttributes_t.
ClusterShape(value[, names, module, ...])	See cublasLtClusterShape_t.
MatmulInnerShape(value[, names, module, ...])	See cublasLtMatmulInnerShape_t.
cuBLASLtError(status)

Functions

create()	See cublasLtCreate.
destroy(intptr_t light_handle)	See cublasLtDestroy.
get_version()	See cublasLtGetVersion.
get_cudart_version()	See cublasLtGetCudartVersion.
get_property(int type)	See cublasLtGetProperty.
matmul(intptr_t light_handle, ...)	See cublasLtMatmul.
matrix_transform(intptr_t light_handle, ...)	See cublasLtMatrixTransform.
matrix_layout_create(int type, ...)	See cublasLtMatrixLayoutCreate.
matrix_layout_destroy(intptr_t mat_layout)	See cublasLtMatrixLayoutDestroy.
get_matrix_layout_attribute_dtype(int attr)	Get the Python data type of the corresponding MatrixLayoutAttribute attribute.
matrix_layout_set_attribute(...)	See cublasLtMatrixLayoutSetAttribute.
matrix_layout_get_attribute(...)	See cublasLtMatrixLayoutGetAttribute.
matmul_desc_create(int compute_type, ...)	See cublasLtMatmulDescCreate.
matmul_desc_destroy(intptr_t matmul_desc)	See cublasLtMatmulDescDestroy.
get_matmul_desc_attribute_dtype(int attr)	Get the Python data type of the corresponding MatmulDescAttribute attribute.
matmul_desc_set_attribute(...)	See cublasLtMatmulDescSetAttribute.
matmul_desc_get_attribute(...)	See cublasLtMatmulDescGetAttribute.
matrix_transform_desc_create(int scale_type)	See cublasLtMatrixTransformDescCreate.
matrix_transform_desc_destroy(...)	See cublasLtMatrixTransformDescDestroy.
get_matrix_transform_desc_attribute_dtype(...)	Get the Python data type of the corresponding MatrixTransformDescAttribute attribute.
matrix_transform_desc_set_attribute(...)	See cublasLtMatrixTransformDescSetAttribute.
matrix_transform_desc_get_attribute(...)	See cublasLtMatrixTransformDescGetAttribute.
matmul_preference_create()	See cublasLtMatmulPreferenceCreate.
matmul_preference_destroy(intptr_t pref)	See cublasLtMatmulPreferenceDestroy.
get_matmul_preference_attribute_dtype(int attr)	Get the Python data type of the corresponding MatmulPreferenceAttribute attribute.
matmul_preference_set_attribute(...)	See cublasLtMatmulPreferenceSetAttribute.
matmul_preference_get_attribute(...)	See cublasLtMatmulPreferenceGetAttribute.
matmul_algo_get_heuristic(...)	See cublasLtMatmulAlgoGetHeuristic.
matmul_algo_init(intptr_t light_handle, ...)	See cublasLtMatmulAlgoInit.
matmul_algo_check(intptr_t light_handle, ...)	See cublasLtMatmulAlgoCheck.
get_matmul_algo_cap_attribute_dtype(int attr)	Get the Python data type of the corresponding MatmulAlgoCapAttribute attribute.
matmul_algo_cap_get_attribute(intptr_t algo, ...)	See cublasLtMatmulAlgoCapGetAttribute.
get_matmul_algo_config_attribute_dtype(int attr)	Get the Python data type of the corresponding MatmulAlgoConfigAttribute attribute.
matmul_algo_config_set_attribute(...)	See cublasLtMatmulAlgoConfigSetAttribute.
matmul_algo_config_get_attribute(...)	See cublasLtMatmulAlgoConfigGetAttribute.
logger_open_file(log_file)	See cublasLtLoggerOpenFile.
logger_set_level(int level)	See cublasLtLoggerSetLevel.
logger_set_mask(int mask)	See cublasLtLoggerSetMask.
logger_force_disable()	See cublasLtLoggerForceDisable.
get_status_name(int status)	See cublasLtGetStatusName.
get_status_string(int status)	See cublasLtGetStatusString.
heuristics_cache_get_capacity()	See cublasLtHeuristicsCacheGetCapacity.
heuristics_cache_set_capacity(size_t capacity)	See cublasLtHeuristicsCacheSetCapacity.
disable_cpu_instructions_set_mask(...)	See cublasLtDisableCpuInstructionsSetMask.
matmul_algo_get_ids(intptr_t light_handle, ...)