From 983056e456a418662601f2f4414a27b2c576bf67 Mon Sep 17 00:00:00 2001 From: Nick Hill Date: Thu, 25 Sep 2025 20:11:44 -0700 Subject: [PATCH] [Misc] Remove unnecessary memoryviews in shm_broadcast.py (#25721) Signed-off-by: Nick Hill --- vllm/distributed/device_communicators/shm_broadcast.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/vllm/distributed/device_communicators/shm_broadcast.py b/vllm/distributed/device_communicators/shm_broadcast.py index 0ee432fea15e..0fc9d1cf4f51 100644 --- a/vllm/distributed/device_communicators/shm_broadcast.py +++ b/vllm/distributed/device_communicators/shm_broadcast.py @@ -136,8 +136,8 @@ class ShmRingBuffer: self.shared_memory = shared_memory.SharedMemory( create=True, size=self.total_bytes_of_buffer) # initialize the metadata section to 0 - with memoryview(self.shared_memory.buf[self.metadata_offset:] - ) as metadata_buffer: + with self.shared_memory.buf[self. + metadata_offset:] as metadata_buffer: torch.frombuffer(metadata_buffer, dtype=torch.uint8).fill_(0) else: # we are opening an existing buffer @@ -182,14 +182,14 @@ class ShmRingBuffer: def get_data(self, current_idx: int): start = self.data_offset + current_idx * self.max_chunk_bytes end = start + self.max_chunk_bytes - with memoryview(self.shared_memory.buf[start:end]) as buf: + with self.shared_memory.buf[start:end] as buf: yield buf @contextmanager def get_metadata(self, current_idx: int): start = self.metadata_offset + current_idx * self.metadata_size end = start + self.metadata_size - with memoryview(self.shared_memory.buf[start:end]) as buf: + with self.shared_memory.buf[start:end] as buf: yield buf