[Mlir-commits] [mlir] [MLIR][Mesh] Add sharding propagation pass (PR #69665)
Mehdi Amini
llvmlistbot at llvm.org
Thu Oct 19 22:25:19 PDT 2023
================
@@ -122,6 +138,24 @@ def MeshSharding : AttrDef<Mesh_Dialect, "MeshSharding"> {
$partial_axes^ `]`)? `>`
}];
+ let builders = [
+ AttrBuilder<(ins "SymbolRefAttr":$cluster,
+ "ArrayRef<SmallVector<int32_t>>":$split_axes,
+ "ArrayRef<int32_t>": $partial_axes,
+ "mesh::Partial": $partial_type), [{
+ SmallVector<DenseI32ArrayAttr> splitAxesAttr = llvm::to_vector(
+ llvm::map_range(split_axes, [&](ArrayRef<int32_t> array) {
----------------
joker-eph wrote:
See `llvm::map_to_vector` which combines `to_vector` and `map_range` I beileve
https://github.com/llvm/llvm-project/pull/69665
More information about the Mlir-commits
mailing list