[Mlir-commits] [mlir] [MLIR][Mesh] Add sharding propagation pass (PR #69665)
Chengji Yao
llvmlistbot at llvm.org
Wed Oct 25 19:46:45 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,
----------------
yaochengji wrote:
Done.
https://github.com/llvm/llvm-project/pull/69665
More information about the Mlir-commits
mailing list