[BugFix] Avoid secondary error in ShmRingBuffer destructor (#6530)

vllm-project · Jul 18, 2024 · d25877d · d25877d
1 parent 1c27d25
commit d25877d
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/vllm/distributed/device_communicators/shm_broadcast.py b/vllm/distributed/device_communicators/shm_broadcast.py
@@ -119,9 +119,10 @@ def __reduce__(self):
         )
 
     def __del__(self):
-        self.shared_memory.close()
-        if self.is_creator:
-            self.shared_memory.unlink()
+        if hasattr(self, "shared_memory"):
+            self.shared_memory.close()
+            if self.is_creator:
+                self.shared_memory.unlink()
 
     @contextmanager
     def get_data(self, current_idx: int):
@@ -428,7 +429,6 @@ def enqueue(self, obj):
 
     def dequeue(self):
         if self._is_local_reader:
-            overflow = False
             with self.acquire_read() as buf:
                 overflow = buf[0] == 1
                 if not overflow: