nvmath.bindings.cublasLt.MatmulAlgoCapAttribute

enum nvmath.bindings.cublasLt.MatmulAlgoCapAttribute(value)[source]

See cublasLtMatmulAlgoCapAttributes_t.

Member Type:

int

Valid values are as follows:

SPLITK_SUPPORT = <MatmulAlgoCapAttribute.SPLITK_SUPPORT: 0>
REDUCTION_SCHEME_MASK = <MatmulAlgoCapAttribute.REDUCTION_SCHEME_MASK: 1>
CTA_SWIZZLING_SUPPORT = <MatmulAlgoCapAttribute.CTA_SWIZZLING_SUPPORT: 2>
STRIDED_BATCH_SUPPORT = <MatmulAlgoCapAttribute.STRIDED_BATCH_SUPPORT: 3>
OUT_OF_PLACE_RESULT_SUPPORT = <MatmulAlgoCapAttribute.OUT_OF_PLACE_RESULT_SUPPORT: 4>
UPLO_SUPPORT = <MatmulAlgoCapAttribute.UPLO_SUPPORT: 5>
TILE_IDS = <MatmulAlgoCapAttribute.TILE_IDS: 6>
CUSTOM_OPTION_MAX = <MatmulAlgoCapAttribute.CUSTOM_OPTION_MAX: 7>
CUSTOM_MEMORY_ORDER = <MatmulAlgoCapAttribute.CUSTOM_MEMORY_ORDER: 10>
POINTER_MODE_MASK = <MatmulAlgoCapAttribute.POINTER_MODE_MASK: 11>
EPILOGUE_MASK = <MatmulAlgoCapAttribute.EPILOGUE_MASK: 12>
STAGES_IDS = <MatmulAlgoCapAttribute.STAGES_IDS: 13>
LD_NEGATIVE = <MatmulAlgoCapAttribute.LD_NEGATIVE: 14>
NUMERICAL_IMPL_FLAGS = <MatmulAlgoCapAttribute.NUMERICAL_IMPL_FLAGS: 15>
MIN_ALIGNMENT_A_BYTES = <MatmulAlgoCapAttribute.MIN_ALIGNMENT_A_BYTES: 16>
MIN_ALIGNMENT_B_BYTES = <MatmulAlgoCapAttribute.MIN_ALIGNMENT_B_BYTES: 17>
MIN_ALIGNMENT_C_BYTES = <MatmulAlgoCapAttribute.MIN_ALIGNMENT_C_BYTES: 18>
MIN_ALIGNMENT_D_BYTES = <MatmulAlgoCapAttribute.MIN_ALIGNMENT_D_BYTES: 19>
ATOMIC_SYNC = <MatmulAlgoCapAttribute.ATOMIC_SYNC: 20>
MATHMODE_IMPL = <MatmulAlgoCapAttribute.MATHMODE_IMPL: 8>
GAUSSIAN_IMPL = <MatmulAlgoCapAttribute.GAUSSIAN_IMPL: 9>