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

[HotFix] Fix dot op for RDNA3 architecture (#451)

Browse Source 
Disabled BlockedToWMMA layout transformation until WMMA is supported completely

Signed-off-by: joviliast <iveselov.nn@gmail.com>
This commit is contained in:
Ilya V
2024-01-10 15:58:57 +01:00
committed by GitHub
parent ce9dacec72
commit 2e01bf08e9
1 changed files with 2 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
lib/Dialect/TritonGPU/Transforms/AccelerateAMDMatmul.cpp
View File

@@ -418,7 +418,8 @@ public:
MatrixCoreVersion::CDNA_MFMA3 == matrixCoreVer) {
patterns.add<::BlockedToMFMA>(context, getMfmaVersion(matrixCoreVer),
matrixInstructionSize);
} else if (matrixCoreVer == MatrixCoreVersion::RDNA_WMMA) {
} else if (MatrixCoreVersion::RDNA_WMMA == matrixCoreVer &&
enableWmmaTransform) {
patterns.add<::BlockedToWMMA>(context);
}
if (applyPatternsAndFoldGreedily(m, std::move(patterns)).failed()) {