From cc4132233217f4ea1f8a13faf96d1523bcc28a77 Mon Sep 17 00:00:00 2001
From: PENGUINLIONG <admin@penguinliong.moe>
Date: Mon, 16 Jan 2023 09:05:53 +0800
Subject: [PATCH] Fixed invalid iterator

---
 taichi/ir/ir.cpp | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/taichi/ir/ir.cpp b/taichi/ir/ir.cpp
index 0d5946ced60c1..a728cdda1a0d2 100644
--- a/taichi/ir/ir.cpp
+++ b/taichi/ir/ir.cpp
@@ -350,7 +350,7 @@ void Block::replace_with(Stmt *old_statement,
     *iter = std::move(new_statements[0]);
     (*iter)->parent = this;
   } else {
-    statements.erase(iter);
+    iter = statements.erase(iter);
     insert_at(std::move(new_statements), iter);
   }
 }