nvmath.bindings.cublasLt.MatmulDescAttribute

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

See cublasLtMatmulDescAttributes_t.

Member Type:

int

Valid values are as follows:

COMPUTE_TYPE = <MatmulDescAttribute.COMPUTE_TYPE: 0>
SCALE_TYPE = <MatmulDescAttribute.SCALE_TYPE: 1>
POINTER_MODE = <MatmulDescAttribute.POINTER_MODE: 2>
TRANSA = <MatmulDescAttribute.TRANSA: 3>
TRANSB = <MatmulDescAttribute.TRANSB: 4>
TRANSC = <MatmulDescAttribute.TRANSC: 5>
FILL_MODE = <MatmulDescAttribute.FILL_MODE: 6>
EPILOGUE = <MatmulDescAttribute.EPILOGUE: 7>
BIAS_POINTER = <MatmulDescAttribute.BIAS_POINTER: 8>
BIAS_BATCH_STRIDE = <MatmulDescAttribute.BIAS_BATCH_STRIDE: 10>
EPILOGUE_AUX_POINTER = <MatmulDescAttribute.EPILOGUE_AUX_POINTER: 11>
EPILOGUE_AUX_LD = <MatmulDescAttribute.EPILOGUE_AUX_LD: 12>
EPILOGUE_AUX_BATCH_STRIDE = <MatmulDescAttribute.EPILOGUE_AUX_BATCH_STRIDE: 13>
ALPHA_VECTOR_BATCH_STRIDE = <MatmulDescAttribute.ALPHA_VECTOR_BATCH_STRIDE: 14>
SM_COUNT_TARGET = <MatmulDescAttribute.SM_COUNT_TARGET: 15>
A_SCALE_POINTER = <MatmulDescAttribute.A_SCALE_POINTER: 17>
B_SCALE_POINTER = <MatmulDescAttribute.B_SCALE_POINTER: 18>
C_SCALE_POINTER = <MatmulDescAttribute.C_SCALE_POINTER: 19>
D_SCALE_POINTER = <MatmulDescAttribute.D_SCALE_POINTER: 20>
AMAX_D_POINTER = <MatmulDescAttribute.AMAX_D_POINTER: 21>
EPILOGUE_AUX_DATA_TYPE = <MatmulDescAttribute.EPILOGUE_AUX_DATA_TYPE: 22>
EPILOGUE_AUX_SCALE_POINTER = <MatmulDescAttribute.EPILOGUE_AUX_SCALE_POINTER: 23>
EPILOGUE_AUX_AMAX_POINTER = <MatmulDescAttribute.EPILOGUE_AUX_AMAX_POINTER: 24>
FAST_ACCUM = <MatmulDescAttribute.FAST_ACCUM: 25>
BIAS_DATA_TYPE = <MatmulDescAttribute.BIAS_DATA_TYPE: 26>
ATOMIC_SYNC_NUM_CHUNKS_D_ROWS = <MatmulDescAttribute.ATOMIC_SYNC_NUM_CHUNKS_D_ROWS: 27>
ATOMIC_SYNC_NUM_CHUNKS_D_COLS = <MatmulDescAttribute.ATOMIC_SYNC_NUM_CHUNKS_D_COLS: 28>
ATOMIC_SYNC_IN_COUNTERS_POINTER = <MatmulDescAttribute.ATOMIC_SYNC_IN_COUNTERS_POINTER: 29>
ATOMIC_SYNC_OUT_COUNTERS_POINTER = <MatmulDescAttribute.ATOMIC_SYNC_OUT_COUNTERS_POINTER: 30>