[Mlir-commits] [llvm] [mlir] [mlir][mesh] Add all-scatter operation (PR #81218)

Chengji Yao llvmlistbot at llvm.org
Tue Feb 13 11:40:20 PST 2024


================
@@ -341,6 +342,68 @@ def Mesh_AllReduceOp : Mesh_CollectiveCommunicationOpBase<"all_reduce", [
   let hasCanonicalizer = 1;
 }
 
+def Mesh_AllScatterOp : Mesh_CollectiveCommunicationOpBase<"all_scatter", [
----------------
yaochengji wrote:

I'm wondering if it's appropriate to include scatter in the operation's name, as scatter typically implies that the operation involves communication.

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


More information about the Mlir-commits mailing list