[rocdl] Register some MI210 (gfx90a) supported mfma cases (#16592) MI210 is CDNA2 and also supports the two mfma intrinsics we are targeting right now.
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});