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

Chengji Yao llvmlistbot at llvm.org
Thu Nov 2 21:21:13 PDT 2023


================
@@ -316,6 +316,15 @@ ArrayAttr Builder::getStrArrayAttr(ArrayRef<StringRef> values) {
   return getArrayAttr(attrs);
 }
 
+ArrayAttr
+Builder::getArrayOfI32ArrayAttr(ArrayRef<SmallVector<int32_t>> values) {
+  auto attrs =
+      llvm::map_to_vector<8>(values, [this](ArrayRef<int32_t> v) -> Attribute {
+        return getDenseI32ArrayAttr(v);
+      });
+  return getArrayAttr(attrs);
+}
----------------
yaochengji wrote:

It isn't used currently. Previous this was used by the sharding option attribute logic. The PR description has also been modified to align this change.

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


More information about the Mlir-commits mailing list