From 2e01bf08e96256f9989fe7fbdc40463810a187bc Mon Sep 17 00:00:00 2001
From: Ilya V <152324710+joviliast@users.noreply.github.com>
Date: Wed, 10 Jan 2024 15:58:57 +0100
Subject: [PATCH] [HotFix] Fix dot op for RDNA3 architecture (#451)

Disabled BlockedToWMMA layout transformation until WMMA is supported completely

Signed-off-by: joviliast <iveselov.nn@gmail.com>
---
 lib/Dialect/TritonGPU/Transforms/AccelerateAMDMatmul.cpp | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/lib/Dialect/TritonGPU/Transforms/AccelerateAMDMatmul.cpp b/lib/Dialect/TritonGPU/Transforms/AccelerateAMDMatmul.cpp
index b4554a73a..d1742fdd4 100644
--- a/lib/Dialect/TritonGPU/Transforms/AccelerateAMDMatmul.cpp
+++ b/lib/Dialect/TritonGPU/Transforms/AccelerateAMDMatmul.cpp
@@ -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()) {