Skip to content

Commit 71b0b00

Browse files
Use f16 quick tuning list in case of bf16 type.
1 parent f8a6235 commit 71b0b00

File tree

1 file changed

+6
-2
lines changed

1 file changed

+6
-2
lines changed

mlir/lib/Dialect/Rock/Tuning/GridwiseGemmParams.cpp

Lines changed: 6 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -80,10 +80,14 @@ class ParamLookupTable {
8080
llvm::raw_string_ostream os(dataTypeStr);
8181
os << dataType;
8282

83-
if (dataTypeStr.find("f8E") != std::string::npos)
83+
if (dataTypeStr == "bf16") {
84+
dataTypeStr = "f16";
85+
} else if (dataTypeStr.find("f8E") != std::string::npos) {
8486
dataTypeStr = "fp8";
85-
if (dataTypeStr.at(0) == 's' || dataTypeStr.at(0) == 'u')
87+
} else if (dataType.isInteger() &&
88+
(dataTypeStr.at(0) == 's' || dataTypeStr.at(0) == 'u')) {
8689
dataTypeStr = dataTypeStr.substr(1);
90+
}
8791

8892
return dataTypeStr;
8993
}

0 commit comments

Comments
 (0)