Proper thread stop in ls

disintar · Apr 11, 2024 · c92384f · c92384f
1 parent 1644d67
commit c92384f
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/tvm-python/PyLiteClient.h b/tvm-python/PyLiteClient.h
@@ -230,8 +230,10 @@ class PyLiteClient {
 
   void stop() {
     scheduler_.run_in_context_external([&] { send_closure(engine, &LiteClientActorEngine::exit); });
+    scheduler_.run_in_context_external([&] { engine.reset(); });
     scheduler_.run_in_context_external([] { td::actor::SchedulerContext::get()->stop(); });
     scheduler_thread_.join();
+    scheduler_.stop();
   }
 
  private: