[Mlir-commits] [mlir] [mlir][mesh] Add collective communication operations (PR #71960)
Chengji Yao
llvmlistbot at llvm.org
Mon Nov 13 13:10:51 PST 2023
================
@@ -129,6 +205,347 @@ MeshShardingAttr::verify(function_ref<InFlightDiagnostic()> emitError,
return success();
}
+//===----------------------------------------------------------------------===//
+// collective communication ops
+//===----------------------------------------------------------------------===//
+
+namespace {
+
+std::optional<DenseI16ArrayAttr>
+canonicalizeAxesSetAttribute(DenseI16ArrayAttr attr) {
----------------
yaochengji wrote:
But it seems your reply cannot explain why we have to make `mesh_axes` support duplications like [2, 1, 2]?
https://github.com/llvm/llvm-project/pull/71960
More information about the Mlir-commits
mailing list