commit 09deadfb8a58d17a4cf136ce916a661836eff2cf
author Lei Zhang <antiagainst@gmail.com> Tue Feb 27 16:39:52 2024 -0800
committer GitHub <noreply@github.com> Wed Feb 28 00:39:52 2024 +0000
tree 2a59885deb5930df674866b0fc5df16c81279e78
parent 4b1a4e275f3face507525f920be83d7aebf3934c

[rocdl] Register some MI210 (gfx90a) supported mfma cases (#16592)

MI210 is CDNA2 and also supports the two mfma
intrinsics we are targeting right now.

compiler/plugins/target/ROCM/ROCMTargetFeatures.cpp

diff --git a/compiler/plugins/target/ROCM/ROCMTargetFeatures.cpp b/compiler/plugins/target/ROCM/ROCMTargetFeatures.cpp
index 3da61cf..2e93989 100644
--- a/compiler/plugins/target/ROCM/ROCMTargetFeatures.cpp
+++ b/compiler/plugins/target/ROCM/ROCMTargetFeatures.cpp
@@ -21,8 +21,11 @@
 }
 
 ArrayAttr getROCMSupportedMmaAttrs(MLIRContext *context, StringRef targetArch) {
-  // MI300a/x
-  if (targetArch == "gfx940" || targetArch == "gfx942") {
+  if (targetArch == "gfx940" || targetArch == "gfx942") { // MI300A/X
+    return getMfmaArrayAttr(context,
+                            {IREE::GPU::MFMAIntrinsic::F16_16x16x16_F32,
+                             IREE::GPU::MFMAIntrinsic::F16_32x32x8_F32});
+  } else if (targetArch == "gfx90a") { // MI210
     return getMfmaArrayAttr(context,
                             {IREE::GPU::MFMAIntrinsic::F16_16x16x16_F32,
                              IREE::GPU::MFMAIntrinsic::F16_32x32x8_F32});