github/ROCm
1
1
Fork 0
mirror of https://github.com/ROCm/ROCm.git synced 2026-04-05 03:01:17 -04:00
Code Issues Packages Projects Releases Wiki Activity

[Triton-MLIR][BACKEND] Support bfloat16 and clean up some test code (#998)

Browse Source
This commit is contained in:
Keren Zhou
2022-12-20 22:26:51 -08:00
committed by GitHub
parent 899bb0a0e7
commit 50a5128448
15 changed files with 699 additions and 480 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
lib/Analysis/Utility.cpp
View File

@@ -134,6 +134,13 @@ bool supportMMA(Value value, int version) {
(elemTy.isInteger(8) && version >= 2);
}
Type getElementType(Value value) {
auto type = value.getType();
if (auto tensorType = type.dyn_cast<RankedTensorType>())
return tensorType.getElementType();
return type;
}
std::string getValueOperandName(Value value, AsmState &state) {
std::string opName;
llvm::raw_string_ostream ss(opName);