MatmulInnerShape

enum nvmath.linalg.advanced.MatmulInnerShape(value)

请参阅 cublasLtMatmulInnerShape_t。

成员类型:

int

有效值如下

UNDEFINED = <MatmulInnerShape.UNDEFINED: 0>

MMA884 = <MatmulInnerShape.MMA884: 1>

MMA1684 = <MatmulInnerShape.MMA1684: 2>

MMA1688 = <MatmulInnerShape.MMA1688: 3>

MMA16816 = <MatmulInnerShape.MMA16816: 4>