[Mlir-commits] [mlir] [mlir][mesh] Add TableGen deffinitions of more collective ops (PR #73842)
Boian Petkantchin
llvmlistbot at llvm.org
Fri Dec 1 17:11:55 PST 2023
================
@@ -339,6 +339,185 @@ def Mesh_AllToAllOp : Mesh_CollectiveCommunicationOpBase<"all_to_all", [
let hasCanonicalizer = 1;
}
+def Mesh_BroadcastOp : Mesh_CollectiveCommunicationOpBase<"broadcast", [
+ AllShapesMatch<["input", "result"]>,
----------------
sogartar wrote:
I was thinking about that, but can't there be another tensor type? Or maybe the result has something in the `encoding` field, then the program would be considered incorrect.
https://github.com/llvm/llvm-project/pull/73842
More information about the Mlir-commits
mailing list