Revert "Remove redundant destructor calls"

This reverts commit 6b4514b.
HabanaAI · Sep 25, 2024 · cf48997 · cf48997
1 parent 6b4514b
commit cf48997
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 0 deletions.
diff --git a/vllm/executor/habana_executor.py b/vllm/executor/habana_executor.py
@@ -195,6 +195,9 @@ def check_health(self) -> None:
     def shutdown(self) -> None:
         self.driver_worker.shutdown_inc()
 
+    def __del__(self):
+        self.shutdown()
+
 
 class HabanaExecutorAsync(HabanaExecutor, ExecutorAsyncBase):
 

diff --git a/vllm/worker/habana_worker.py b/vllm/worker/habana_worker.py
@@ -320,6 +320,9 @@ def list_prompt_adapters(self) -> Set[int]:
     def shutdown_inc(self):
         self.model_runner.shutdown_inc()
 
+    def __del__(self):
+        self.shutdown_inc()
+
     @property
     def max_model_len(self) -> int:
         return self.model_config.max_model_len