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>#