We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent f8a6235 commit 71b0b00Copy full SHA for 71b0b00
mlir/lib/Dialect/Rock/Tuning/GridwiseGemmParams.cpp
@@ -80,10 +80,14 @@ class ParamLookupTable {
80
llvm::raw_string_ostream os(dataTypeStr);
81
os << dataType;
82
83
- if (dataTypeStr.find("f8E") != std::string::npos)
+ if (dataTypeStr == "bf16") {
84
+ dataTypeStr = "f16";
85
+ } else if (dataTypeStr.find("f8E") != std::string::npos) {
86
dataTypeStr = "fp8";
- if (dataTypeStr.at(0) == 's' || dataTypeStr.at(0) == 'u')
87
+ } else if (dataType.isInteger() &&
88
+ (dataTypeStr.at(0) == 's' || dataTypeStr.at(0) == 'u')) {
89
dataTypeStr = dataTypeStr.substr(1);
90
+ }
91
92
return dataTypeStr;
93
}
0 commit comments