[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