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