MatmulPreferenceAttribute#

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

See cublasLtMatmulPreferenceAttributes_t.

Member Type:

int

Valid values are as follows:

SEARCH_MODE = <MatmulPreferenceAttribute.SEARCH_MODE: 0>#
MAX_WORKSPACE_BYTES = <MatmulPreferenceAttribute.MAX_WORKSPACE_BYTES: 1>#
REDUCTION_SCHEME_MASK = <MatmulPreferenceAttribute.REDUCTION_SCHEME_MASK: 3>#
MIN_ALIGNMENT_A_BYTES = <MatmulPreferenceAttribute.MIN_ALIGNMENT_A_BYTES: 5>#
MIN_ALIGNMENT_B_BYTES = <MatmulPreferenceAttribute.MIN_ALIGNMENT_B_BYTES: 6>#
MIN_ALIGNMENT_C_BYTES = <MatmulPreferenceAttribute.MIN_ALIGNMENT_C_BYTES: 7>#
MIN_ALIGNMENT_D_BYTES = <MatmulPreferenceAttribute.MIN_ALIGNMENT_D_BYTES: 8>#
MAX_WAVES_COUNT = <MatmulPreferenceAttribute.MAX_WAVES_COUNT: 9>#
IMPL_MASK = <MatmulPreferenceAttribute.IMPL_MASK: 12>#
MATH_MODE_MASK = <MatmulPreferenceAttribute.MATH_MODE_MASK: 2>#
GAUSSIAN_MODE_MASK = <MatmulPreferenceAttribute.GAUSSIAN_MODE_MASK: 4>#
POINTER_MODE_MASK = <MatmulPreferenceAttribute.POINTER_MODE_MASK: 10>#
EPILOGUE_MASK = <MatmulPreferenceAttribute.EPILOGUE_MASK: 11>#
SM_COUNT_TARGET = <MatmulPreferenceAttribute.SM_COUNT_TARGET: 13>#