nvmath.bindings.cublas.ComputeType

enum nvmath.bindings.cublas.ComputeType(value)[source]

See cublasComputeType_t.

Member Type:

int

Valid values are as follows:

COMPUTE_16F = <ComputeType.COMPUTE_16F: 64>
COMPUTE_16F_PEDANTIC = <ComputeType.COMPUTE_16F_PEDANTIC: 65>
COMPUTE_32F = <ComputeType.COMPUTE_32F: 68>
COMPUTE_32F_PEDANTIC = <ComputeType.COMPUTE_32F_PEDANTIC: 69>
COMPUTE_32F_FAST_16F = <ComputeType.COMPUTE_32F_FAST_16F: 74>
COMPUTE_32F_FAST_16BF = <ComputeType.COMPUTE_32F_FAST_16BF: 75>
COMPUTE_32F_FAST_TF32 = <ComputeType.COMPUTE_32F_FAST_TF32: 77>
COMPUTE_64F = <ComputeType.COMPUTE_64F: 70>
COMPUTE_64F_PEDANTIC = <ComputeType.COMPUTE_64F_PEDANTIC: 71>
COMPUTE_32I = <ComputeType.COMPUTE_32I: 72>
COMPUTE_32I_PEDANTIC = <ComputeType.COMPUTE_32I_PEDANTIC: 73>