[Mlir-commits] [mlir] [mlir][mesh] Add collective communication operations (PR #71960)

Mehdi Amini llvmlistbot at llvm.org
Wed Nov 15 01:30:47 PST 2023


================
@@ -77,6 +79,15 @@ def Mesh_ClusterOp : Mesh_Op<"cluster", [Symbol]> {
     $sym_name `(` `rank` `=` $rank (`,` `dim_sizes` `=` $dim_sizes^)? `)`
       attr-dict
   }];
+  let extraClassDeclaration = [{
+    ::mlir::SmallVector<int64_t> canonicalDimSizes();
+
+    template <typename OutIt>
+    void canonicalDimSizes(OutIt outIt) {
----------------
joker-eph wrote:

Can you document these methods? What is the "canonical" form?

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


More information about the Mlir-commits mailing list