Skip to content

Commit c79231d

Browse files
committed
[AMD] Skip mfma layout in maybeDuplicate
The workaround introduced in triton-lang#4048 "forgot" to skip mfma layout.
1 parent f04df24 commit c79231d

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

include/triton/Conversion/TritonGPUToLLVM/ElementwiseOpToLLVMBase.h

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -88,6 +88,8 @@ class ElementwiseOpConversionBase : public ConvertOpToLLVMPattern<SourceOp> {
8888
// encoding not available
8989
return resultVals;
9090
Attribute baseEncoding = encoding;
91+
if (isa<AMDMfmaEncodingAttr>(baseEncoding))
92+
return resultVals;
9193
while (auto sliced = dyn_cast<SliceEncodingAttr>(baseEncoding))
9294
baseEncoding = sliced.getParent();
9395
if (isa<NvidiaMmaEncodingAttr, DotOperandEncodingAttr>(baseEncoding)) {

0 commit comments

Comments
 (0)