[Mlir-commits] [llvm] [mlir] [mlir][mesh] Add all-scatter operation (PR #81218)
Boian Petkantchin
llvmlistbot at llvm.org
Tue Feb 13 14:26:04 PST 2024
================
@@ -341,6 +342,68 @@ def Mesh_AllReduceOp : Mesh_CollectiveCommunicationOpBase<"all_reduce", [
let hasCanonicalizer = 1;
}
+def Mesh_AllScatterOp : Mesh_CollectiveCommunicationOpBase<"all_scatter", [
----------------
sogartar wrote:
I rename it to `all_slice`.
https://github.com/llvm/llvm-project/pull/81218
More information about the Mlir-commits
mailing list