[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