MatmulAlgoConfigAttribute#

enum nvmath.bindings.cublasLt.MatmulAlgoConfigAttribute(value)[source]#

See cublasLtMatmulAlgoConfigAttributes_t.

Member Type:: int

Valid values are as follows:

ID = <MatmulAlgoConfigAttribute.ID: 0>#

TILE_ID = <MatmulAlgoConfigAttribute.TILE_ID: 1>#

SPLITK_NUM = <MatmulAlgoConfigAttribute.SPLITK_NUM: 2>#

REDUCTION_SCHEME = <MatmulAlgoConfigAttribute.REDUCTION_SCHEME: 3>#

CTA_SWIZZLING = <MatmulAlgoConfigAttribute.CTA_SWIZZLING: 4>#

CUSTOM_OPTION = <MatmulAlgoConfigAttribute.CUSTOM_OPTION: 5>#

STAGES_ID = <MatmulAlgoConfigAttribute.STAGES_ID: 6>#

INNER_SHAPE_ID = <MatmulAlgoConfigAttribute.INNER_SHAPE_ID: 7>#

CLUSTER_SHAPE_ID = <MatmulAlgoConfigAttribute.CLUSTER_SHAPE_ID: 8>#