[MLIR] fix msan uninitialized value error (#71286)

yaochengji · web-flow · commit 7d367bc92bad · 2023-11-04T09:12:24.000-07:00
diff --git a/mlir/lib/Dialect/Mesh/Interfaces/ShardingInterface.cpp b/mlir/lib/Dialect/Mesh/Interfaces/ShardingInterface.cpp
@@ -423,7 +423,8 @@ static LogicalResult addShardOp(OpBuilder &b, OpResult result,
   }
 
   // process the partial axes
-  Partial partialType;
+  // partialType will be ignored if partialAxes is empty
+  Partial partialType = Partial::Sum;
   for (auto it : llvm::zip(loopTypes, shardingOption.shardingArray)) {
     IteratorType iType = std::get<0>(it);
     if (isReductionLoop(iType)) {

Original file line number	Diff line number	Diff line change
`@@ -423,7 +423,8 @@ static LogicalResult addShardOp(OpBuilder &b, OpResult result,`
`423`	`423`	`}`
`424`	`424`
`425`	`425`	`// process the partial axes`
`426`		`- Partial partialType;`
	`426`	`+ // partialType will be ignored if partialAxes is empty`
	`427`	`+ Partial partialType = Partial::Sum;`
`427`	`428`	`for (auto it : llvm::zip(loopTypes, shardingOption.shardingArray)) {`
`428`	`429`	`IteratorType iType = std::get<0>(it);`
`429`	`430`	`if (isReductionLoop(iType)) {`