[AMD] Skip mfma layout in maybeDuplicate

The workaround introduced in triton-lang#4048 "forgot" to skip mfma layout.
zhanglx13 · Jun 19, 2024 · c79231d · c79231d
1 parent f04df24
commit c79231d
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/include/triton/Conversion/TritonGPUToLLVM/ElementwiseOpToLLVMBase.h b/include/triton/Conversion/TritonGPUToLLVM/ElementwiseOpToLLVMBase.h
@@ -88,6 +88,8 @@ class ElementwiseOpConversionBase : public ConvertOpToLLVMPattern<SourceOp> {
       // encoding not available
       return resultVals;
     Attribute baseEncoding = encoding;
+    if (isa<AMDMfmaEncodingAttr>(baseEncoding))
+      return resultVals;
     while (auto sliced = dyn_cast<SliceEncodingAttr>(baseEncoding))
       baseEncoding = sliced.getParent();
     if (isa<NvidiaMmaEncodingAttr, DotOperandEncodingAttr>(baseEncoding)) {