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

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

@ -119,9 +119,10 @@ class ShmRingBuffer:
)
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 @@ class MessageQueue:
def dequeue(self):
if self._is_local_reader:
overflow = False
with self.acquire_read() as buf:
overflow = buf[0] == 1
if not overflow: