[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