squall/vllm
1
0
Fork 0
Code Issues Pull Requests Actions 1 Packages Projects Releases Wiki Activity

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

Browse Source
This commit is contained in:
Nick Hill 2024-07-17 22:24:43 -07:00 committed by GitHub
parent 1c27d25fb5
commit d25877dd9b
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
vllm/distributed/device_communicators/shm_broadcast.py
View File

@ -119,6 +119,7 @@ class ShmRingBuffer:
) )
def __del__(self): def __del__(self):
if hasattr(self, "shared_memory"):
self.shared_memory.close() self.shared_memory.close()
if self.is_creator: if self.is_creator:
self.shared_memory.unlink() self.shared_memory.unlink()
@ -428,7 +429,6 @@ class MessageQueue:
def dequeue(self): def dequeue(self):
if self._is_local_reader: if self._is_local_reader:
overflow = False
with self.acquire_read() as buf: with self.acquire_read() as buf:
overflow = buf[0] == 1 overflow = buf[0] == 1
if not overflow: if not overflow: