Use for statement directly as an operand instead of having it pretend to be an induction variable.
PiperOrigin-RevId: 206759180
diff --git a/lib/Parser/Parser.cpp b/lib/Parser/Parser.cpp
index 3bcec37..b20a40a 100644
--- a/lib/Parser/Parser.cpp
+++ b/lib/Parser/Parser.cpp
@@ -2117,7 +2117,7 @@
ForStmt *forStmt = builder.createFor(lowerBound, upperBound, step);
// Create SSA value definition for the induction variable.
- addDefinition({inductionVariableName, 0, loc}, forStmt->getInductionVar());
+ addDefinition({inductionVariableName, 0, loc}, forStmt);
// If parsing of the for statement body fails,
// MLIR contains for statement with those nested statements that have been