Fix potential bug of lang::Program that could be double finalized (#5550

)
taichi-dev · Jul 28, 2022 · 0aa0a17 · 0aa0a17
1 parent 719a927
commit 0aa0a17
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/taichi/program/program.cpp b/taichi/program/program.cpp
@@ -381,6 +381,9 @@ uint64 Program::fetch_result_uint64(int i) {
 }
 
 void Program::finalize() {
+  if (finalized_) {
+    return;
+  }
   synchronize();
 
   TI_TRACE("Program finalizing...");
@@ -498,8 +501,7 @@ void Program::fill_ndarray_fast(Ndarray *ndarray, uint32_t val) {
 }
 
 Program::~Program() {
-  if (!finalized_)
-    finalize();
+  finalize();
 }
 
 std::unique_ptr<AotModuleBuilder> Program::make_aot_module_builder(Arch arch) {