nvmath.bindings.cublasLt.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>