[Mlir-commits] [mlir] mlir::mesh::shardingOp adding shard-size control (PR #98145)
Boian Petkantchin
llvmlistbot at llvm.org
Tue Jul 9 18:55:36 PDT 2024
================
@@ -450,53 +496,64 @@ reshardOn1DMesh(ImplicitLocOpBuilder &builder, MeshOp mesh,
handlePartialAxesDuringResharding(builder, sourceSharding, targetSharding,
sourceShard);
- if (reducedSourceSharding == targetSharding) {
+ if (!force && reducedSourceSharding == targetSharding) {
----------------
sogartar wrote:
Why do we need `force`?
https://github.com/llvm/llvm-project/pull/98145
More information about the Mlir-commits
mailing list