nvmath.bindings.cublasLt.MatmulStages

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

See cublasLtMatmulStages_t.

Member Type:

int

Valid values are as follows:

STAGES_UNDEFINED = <MatmulStages.STAGES_UNDEFINED: 0>
STAGES_16x1 = <MatmulStages.STAGES_16x1: 1>
STAGES_16x2 = <MatmulStages.STAGES_16x2: 2>
STAGES_16x3 = <MatmulStages.STAGES_16x3: 3>
STAGES_16x4 = <MatmulStages.STAGES_16x4: 4>
STAGES_16x5 = <MatmulStages.STAGES_16x5: 5>
STAGES_16x6 = <MatmulStages.STAGES_16x6: 6>
STAGES_32x1 = <MatmulStages.STAGES_32x1: 7>
STAGES_32x2 = <MatmulStages.STAGES_32x2: 8>
STAGES_32x3 = <MatmulStages.STAGES_32x3: 9>
STAGES_32x4 = <MatmulStages.STAGES_32x4: 10>
STAGES_32x5 = <MatmulStages.STAGES_32x5: 11>
STAGES_32x6 = <MatmulStages.STAGES_32x6: 12>
STAGES_64x1 = <MatmulStages.STAGES_64x1: 13>
STAGES_64x2 = <MatmulStages.STAGES_64x2: 14>
STAGES_64x3 = <MatmulStages.STAGES_64x3: 15>
STAGES_64x4 = <MatmulStages.STAGES_64x4: 16>
STAGES_64x5 = <MatmulStages.STAGES_64x5: 17>
STAGES_64x6 = <MatmulStages.STAGES_64x6: 18>
STAGES_128x1 = <MatmulStages.STAGES_128x1: 19>
STAGES_128x2 = <MatmulStages.STAGES_128x2: 20>
STAGES_128x3 = <MatmulStages.STAGES_128x3: 21>
STAGES_128x4 = <MatmulStages.STAGES_128x4: 22>
STAGES_128x5 = <MatmulStages.STAGES_128x5: 23>
STAGES_128x6 = <MatmulStages.STAGES_128x6: 24>
STAGES_32x10 = <MatmulStages.STAGES_32x10: 25>
STAGES_8x4 = <MatmulStages.STAGES_8x4: 26>
STAGES_16x10 = <MatmulStages.STAGES_16x10: 27>
STAGES_8x5 = <MatmulStages.STAGES_8x5: 28>
STAGES_8x3 = <MatmulStages.STAGES_8x3: 31>
STAGES_8xAUTO = <MatmulStages.STAGES_8xAUTO: 32>
STAGES_16xAUTO = <MatmulStages.STAGES_16xAUTO: 33>
STAGES_32xAUTO = <MatmulStages.STAGES_32xAUTO: 34>
STAGES_64xAUTO = <MatmulStages.STAGES_64xAUTO: 35>
STAGES_128xAUTO = <MatmulStages.STAGES_128xAUTO: 36>
STAGES_16x80 = <MatmulStages.STAGES_16x80: 29>
STAGES_64x80 = <MatmulStages.STAGES_64x80: 30>