Fixed invalid iterator

taichi-dev · Jan 16, 2023 · cc41322 · cc41322
1 parent ff1fff1
commit cc41322
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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);
   }
 }