[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