nvmath.linalg.advanced.MatmulInnerShape

enum nvmath.linalg.advanced.MatmulInnerShape(value)[source]

See cublasLtMatmulInnerShape_t.

Member Type:

int

Valid values are as follows:

UNDEFINED = <MatmulInnerShape.UNDEFINED: 0>
MMA884 = <MatmulInnerShape.MMA884: 1>
MMA1684 = <MatmulInnerShape.MMA1684: 2>
MMA1688 = <MatmulInnerShape.MMA1688: 3>
MMA16816 = <MatmulInnerShape.MMA16816: 4>