[Mlir-commits] [mlir] [MLIR][Mesh] Add sharding propagation pass (PR #69665)

Boian Petkantchin llvmlistbot at llvm.org
Wed Oct 25 11:26:06 PDT 2023


================
@@ -34,6 +34,48 @@ namespace {
 // Utilities
 //===----------------------------------------------------------------------===//
 
+// This method returns all possible sharding attributes. For example,
+// mustShardings = [shard0, None] and optionalShardings = [None, shard1], the
+// result will be [[shard0, shard1], [shard0, None]]
+static SmallVector<SmallVector<MeshShardingAttr>>
+getOrderedPossibleShardingAttrs(ArrayRef<MeshShardingAttr> mustShardings,
----------------
sogartar wrote:

Could you mention in the comments what is the ordering?

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


More information about the Mlir-commits mailing list