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

Boian Petkantchin llvmlistbot at llvm.org
Mon Nov 13 13:40:24 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:

I wanted to use a set but there is no such attribute, so I went for an array. In math the set `{2, 1, 2}` is the same as `{2, 1}`. I can see why you want to forbid duplication as it may result from a programming error where the intention was something else.

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


More information about the Mlir-commits mailing list