MatmulDescriptorAttribute#

enum nvmath.bindings.cublasMp.MatmulDescriptorAttribute(value)[source]#

See cublasMpMatmulDescriptorAttribute_t.

Member Type:: int

Valid values are as follows:

TRANSA = <MatmulDescriptorAttribute.TRANSA: 0>#

TRANSB = <MatmulDescriptorAttribute.TRANSB: 1>#

COMPUTE_TYPE = <MatmulDescriptorAttribute.COMPUTE_TYPE: 2>#

ALGO_TYPE = <MatmulDescriptorAttribute.ALGO_TYPE: 3>#

COMMUNICATION_SM_COUNT = <MatmulDescriptorAttribute.COMMUNICATION_SM_COUNT: 4>#

EPILOGUE = <MatmulDescriptorAttribute.EPILOGUE: 5>#

BIAS_POINTER = <MatmulDescriptorAttribute.BIAS_POINTER: 6>#

BIAS_BATCH_STRIDE = <MatmulDescriptorAttribute.BIAS_BATCH_STRIDE: 7>#

BIAS_DATA_TYPE = <MatmulDescriptorAttribute.BIAS_DATA_TYPE: 8>#

EPILOGUE_AUX_POINTER = <MatmulDescriptorAttribute.EPILOGUE_AUX_POINTER: 9>#

EPILOGUE_AUX_LD = <MatmulDescriptorAttribute.EPILOGUE_AUX_LD: 10>#

EPILOGUE_AUX_BATCH_STRIDE = <MatmulDescriptorAttribute.EPILOGUE_AUX_BATCH_STRIDE: 11>#

EPILOGUE_AUX_DATA_TYPE = <MatmulDescriptorAttribute.EPILOGUE_AUX_DATA_TYPE: 12>#

EPILOGUE_AUX_SCALE_POINTER = <MatmulDescriptorAttribute.EPILOGUE_AUX_SCALE_POINTER: 13>#

EPILOGUE_AUX_AMAX_POINTER = <MatmulDescriptorAttribute.EPILOGUE_AUX_AMAX_POINTER: 14>#

EPILOGUE_AUX_SCALE_MODE = <MatmulDescriptorAttribute.EPILOGUE_AUX_SCALE_MODE: 15>#

A_SCALE_POINTER = <MatmulDescriptorAttribute.A_SCALE_POINTER: 16>#

A_SCALE_MODE = <MatmulDescriptorAttribute.A_SCALE_MODE: 17>#

B_SCALE_POINTER = <MatmulDescriptorAttribute.B_SCALE_POINTER: 18>#

B_SCALE_MODE = <MatmulDescriptorAttribute.B_SCALE_MODE: 19>#

C_SCALE_POINTER = <MatmulDescriptorAttribute.C_SCALE_POINTER: 20>#

C_SCALE_MODE = <MatmulDescriptorAttribute.C_SCALE_MODE: 21>#

D_SCALE_POINTER = <MatmulDescriptorAttribute.D_SCALE_POINTER: 22>#

D_SCALE_MODE = <MatmulDescriptorAttribute.D_SCALE_MODE: 23>#

AMAX_D_POINTER = <MatmulDescriptorAttribute.AMAX_D_POINTER: 24>#