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