update

alibaba · Dec 30, 2024 · befe932 · befe932
1 parent cedce15
commit befe932
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tao_compiler/mlir/disc/transforms/lhlo_legalize_roots_to_loops.cc b/tao_compiler/mlir/disc/transforms/lhlo_legalize_roots_to_loops.cc
@@ -1069,7 +1069,7 @@ void emitInitLoops(OpBuilder& b, ArrayRef<Operation*> col_reduction_ops) {
   Location loc = first_root->getLoc();
   Value zero = b.create<arith::ConstantIndexOp>(loc, 0);
   Value one = b.create<arith::ConstantIndexOp>(loc, 1);
-  Value two = b.create<arith::ConstantIndexOp>(loc, 1);
+  Value two = b.create<arith::ConstantIndexOp>(loc, 2);
   Value num_elements = emitNumElementsComputation(b, loc, first_root);
 
   Value var;