@@ -951,7 +951,7 @@ LogicalResult NVVM::StMatrixOp::verify() {
951951 return success ();
952952}
953953
954- FailureOr<int > getAllowedSizeK (NVVM::WGMMATypes typeA) {
954+ static FailureOr<int > getAllowedSizeK (NVVM::WGMMATypes typeA) {
955955 if (typeA == NVVM::WGMMATypes::tf32)
956956 return 8 ;
957957 if (typeA == NVVM::WGMMATypes::f16 || typeA == NVVM::WGMMATypes::bf16 )
@@ -965,9 +965,9 @@ FailureOr<int> getAllowedSizeK(NVVM::WGMMATypes typeA) {
965965 return failure ();
966966}
967967
968- LogicalResult isAllowedWGMMADataType (NVVM::WGMMATypes typeD,
969- NVVM::WGMMATypes typeA,
970- NVVM::WGMMATypes typeB) {
968+ static LogicalResult isAllowedWGMMADataType (NVVM::WGMMATypes typeD,
969+ NVVM::WGMMATypes typeA,
970+ NVVM::WGMMATypes typeB) {
971971 switch (typeA) {
972972 case NVVM::WGMMATypes::f16 :
973973 if ((typeD == NVVM::WGMMATypes::f32 || typeD == NVVM::WGMMATypes::f16 ) &&
@@ -1007,7 +1007,7 @@ LogicalResult isAllowedWGMMADataType(NVVM::WGMMATypes typeD,
10071007 return failure ();
10081008}
10091009
1010- LogicalResult isAllowedSizeN (int sizeN, NVVM::WGMMATypes typeA) {
1010+ static LogicalResult isAllowedSizeN (int sizeN, NVVM::WGMMATypes typeA) {
10111011 SmallVector<int > allowedN = {8 , 16 , 24 , 32 , 40 , 48 , 56 , 64 ,
10121012 72 , 80 , 88 , 96 , 104 , 112 , 120 , 128 ,
10131013 136 , 144 , 152 , 160 , 168 , 176 , 184 , 192 ,
0 commit comments