[Mlir-commits] [mlir] mlir::mesh::shardingOp adding shard-size control (PR #98145)
Boian Petkantchin
llvmlistbot at llvm.org
Tue Jul 9 18:55:34 PDT 2024
================
@@ -427,11 +441,12 @@ getShardingAttribute(OpResult result, const ShardingOption &shardingOption,
}
removeTrailingEmptySubArray(splitAxes);
- return MeshShardingAttr::get(result.getContext(), shardingOption.mesh,
- splitAxes, partialAxes, partialType);
+ return MeshSharding::get(shardingOption.mesh,
+ fromArrayOfVector(result.getContext(), splitAxes),
+ partialAxes, partialType);
}
-static FailureOr<MeshShardingAttr>
+static FailureOr<MeshSharding>
getShardingAttribute(OpOperand &opOperand, const ShardingOption &shardingOption,
----------------
sogartar wrote:
We should drop `Atrribute` from the name.
https://github.com/llvm/llvm-project/pull/98145
More information about the Mlir-commits
mailing list