[Mlir-commits] [mlir] [mlir][mesh] Add collective communication	operations (PR #71960)
    Boian Petkantchin 
    llvmlistbot at llvm.org
       
    Tue Nov 14 10:59:58 PST 2023
    
    
  
================
@@ -129,6 +205,347 @@ MeshShardingAttr::verify(function_ref<InFlightDiagnostic()> emitError,
   return success();
 }
 
+//===----------------------------------------------------------------------===//
+// collective communication ops
+//===----------------------------------------------------------------------===//
+
+namespace {
+
+std::optional<DenseI16ArrayAttr>
+canonicalizeAxesSetAttribute(DenseI16ArrayAttr attr) {
----------------
sogartar wrote:
Done.
I also change the meaning of `mesh_axes` to induce the order of devices in the device groups. I added documentation about this.
https://github.com/llvm/llvm-project/pull/71960
    
    
More information about the Mlir-commits
mailing list