Skip to content

Commit 240859f

Browse files
committed
fix
1 parent 409f804 commit 240859f

File tree

2 files changed

+3
-3
lines changed

2 files changed

+3
-3
lines changed

lightllm/server/core/objs/shm_array.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -21,7 +21,7 @@ def create_shm(self):
2121
self.arr = np.ndarray(self.shape, dtype=self.dtype, buffer=self.shm.buf)
2222

2323
def link_shm(self):
24-
self.shm = shared_memory.SharedMemory(name=self.name, create=False, size=self.dest_size)
24+
self.shm = create_or_link_shm(self.name, self.dest_size)
2525
assert self.shm.size == self.dest_size
2626
self.arr = np.ndarray(self.shape, dtype=self.dtype, buffer=self.shm.buf)
2727
return

lightllm/utils/shm_utils.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -26,7 +26,7 @@ def create_or_link_shm(name, expected_size):
2626
try:
2727
try:
2828
shm = shared_memory.SharedMemory(name=safe_name, create=True, size=expected_size)
29-
logger.info(f"Created new shared memory: {name} ({expected_size=})")
29+
logger.info(f"Created new shared memory: {name} (size={expected_size})")
3030
return shm
3131
except FileExistsError:
3232
try:
@@ -37,7 +37,7 @@ def create_or_link_shm(name, expected_size):
3737
except Exception as e:
3838
logger.error(f"Unexpected error attaching to shared memory {name}: {e}")
3939
raise
40-
logger.info(f"Attached to existing shared memory: {name} ({expected_size=})")
40+
logger.info(f"Attached to existing shared memory: {name} (size={expected_size})")
4141
return shm
4242
except Exception as e:
4343
logger.error(f"Unexpected error creating shared memory {safe_name}: {e}")

0 commit comments

Comments
 (0)