[Mlir-commits] [mlir] Lower allreduce (PR #144716)

Frank Schlimbach llvmlistbot at llvm.org
Mon Jun 23 03:37:30 PDT 2025


================
@@ -584,11 +584,11 @@ def Mesh_AllReduceOp : Mesh_CollectiveCommunicationOpBase<"all_reduce", [
     ```
   }];
   let arguments = !con(commonArgs, (ins
-    AnyRankedTensor:$input,
+    AnyTypeOf<[AnyMemRef, AnyRankedTensor]>:$input,
----------------
fschlimb wrote:

The communication primitives might be added during bufferization, like through implementing the `BufferizableOpInterface`. Hence they must support tensors and memrefs.

https://github.com/llvm/llvm-project/pull/144716


More information about the Mlir-commits mailing list