[Mlir-commits] [mlir] [mlir][mesh] Add TableGen deffinitions of more collective ops (PR #73842)

Chengji Yao llvmlistbot at llvm.org
Fri Dec 1 13:49:25 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"]>,
----------------
yaochengji wrote:

Could we use `AllTypesMatch` here?

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


More information about the Mlir-commits mailing list