Add SameOperandsEncoding

triton-lang · Mar 16, 2023 · 34973c1 · 34973c1
1 parent 00539ba
commit 34973c1
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/include/triton/Dialect/Triton/IR/TritonOps.td b/include/triton/Dialect/Triton/IR/TritonOps.td
@@ -343,7 +343,10 @@ def TT_DotOp : TT_Op<"dot", [Pure,
 // Reduce Op
 //
 def TT_ReduceOp: TT_Op<"reduce",
-                       [Pure, DeclareOpInterfaceMethods<InferTypeOpInterface>, SingleBlock]> {
+                       [Pure,
+                        SameOperandsEncoding,
+                        SingleBlock,
+                        DeclareOpInterfaceMethods<InferTypeOpInterface>]> {
     let summary = "Reduction using generic combination algorithm";
     let arguments = (ins Variadic<TT_Tensor>:$operands, I32Attr:$axis);
     let results = (outs Variadic<TT_Type>:$result);