Fix flang build after D97804

diff --git a/flang/lib/Optimizer/Dialect/FIROps.cpp b/flang/lib/Optimizer/Dialect/FIROps.cpp
index 2f09de6..cb8ec75 100644 (file)
--- a/flang/lib/Optimizer/Dialect/FIROps.cpp
+++ b/flang/lib/Optimizer/Dialect/FIROps.cpp
@@ -957,9 +957,10 @@ static void print(mlir::OpAsmPrinter &p, fir::IterWhileOp op) {
     llvm::interleaveComma(
         llvm::zip(regionArgs.drop_front(), operands.drop_front()), p,
         [&](auto it) { p << std::get<0>(it) << " = " << std::get<1>(it); });
-    auto resTypes = op.finalValue() ? op.getResultTypes()
-                                    : op.getResultTypes().drop_front();
-    p << ") -> (" << resTypes << ')';
+    p << ") -> (";
+    llvm::interleaveComma(
+        llvm::drop_begin(op.getResultTypes(), op.finalValue() ? 0 : 1), p);
+    p << ")";
   } else if (op.finalValue()) {
     p << " -> (" << op.getResultTypes() << ')';
   }