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

enable layout conversion from mfma to dot_op for mfma16. (#453)

Browse Source 
* enable the layout conversion from mfma layout to dot_operand layout for mfma16

* backup changes
This commit is contained in:
Shucai Xiao
2024-01-10 22:31:59 -06:00
committed by GitHub
parent b5ed97873c
commit a7bb38ea79
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
lib/Analysis/Utility.cpp
View File

@@ -612,7 +612,7 @@ bool isMfmaToDotShortcut(RankedTensorType &srcTy, RankedTensorType &dstTy) {
dotOperandLayout.getOpIdx() == 0 &&
dotOperandLayout.getKWidth() == 4 &&
dotOperandLayout.getParent() == mfmaLayout &&
mfmaLayout.getNonKDim() == 32 && mfmaLayout.getIsTransposed() &&
(mfmaLayout.getNonKDim() == 32 || mfmaLayout.getNonKDim() == 16) && mfmaLayout.getIsTransposed() &&
(srcTy.getElementType().isF16() || srcTy.getElementType().isBF16());
}
#endif