Use for statement directly as an operand instead of having it pretend to be an induction variable.
PiperOrigin-RevId: 206759180
diff --git a/lib/IR/Statement.cpp b/lib/IR/Statement.cpp
index 6ace8ff..3ac481f 100644
--- a/lib/IR/Statement.cpp
+++ b/lib/IR/Statement.cpp
@@ -198,7 +198,7 @@
ForStmt::ForStmt(AffineConstantExpr *lowerBound, AffineConstantExpr *upperBound,
AffineConstantExpr *step, MLIRContext *context)
: Statement(Kind::For), StmtBlock(StmtBlockKind::For),
- MLValue(MLValueKind::InductionVar, Type::getAffineInt(context)),
+ MLValue(MLValueKind::ForStmt, Type::getAffineInt(context)),
lowerBound(lowerBound), upperBound(upperBound), step(step) {}
//===----------------------------------------------------------------------===//