From a7bb38ea79634787a28014fca124528a3cd8da55 Mon Sep 17 00:00:00 2001
From: Shucai Xiao <shucai.xiao@amd.com>
Date: Wed, 10 Jan 2024 22:31:59 -0600
Subject: [PATCH] enable layout conversion from mfma to dot_op for mfma16.
 (#453)

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

* backup changes
---
 lib/Analysis/Utility.cpp | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/lib/Analysis/Utility.cpp b/lib/Analysis/Utility.cpp
index d32397ace..ee3be561e 100644
--- a/lib/Analysis/Utility.cpp
+++ b/lib/Analysis/Utility.cpp
@@ -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