MatmulDescAttribute

enum nvmath.bindings.cublasLt.MatmulDescAttribute(value)

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>