Skip to content

Commit f6f9c12

Browse files
[fix] fix ipc signal suffix for ep (#4324)
1 parent 8d2aaf3 commit f6f9c12

File tree

1 file changed

+4
-4
lines changed

1 file changed

+4
-4
lines changed

fastdeploy/worker/worker_process.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -198,7 +198,7 @@ def init_health_status(self) -> None:
198198
name="launched_expert_service_signal",
199199
array=launched_expert_service_signal_data,
200200
dtype=np.int32,
201-
suffix=self.parallel_config.engine_worker_queue_port,
201+
suffix=self.parallel_config.engine_pid,
202202
create=False,
203203
)
204204
while self.launched_expert_service_signal.value[self.local_rank % self.max_chips_per_node] == 0:
@@ -215,7 +215,7 @@ def init_health_status(self) -> None:
215215
name="worker_ready_signal",
216216
array=workers_ready,
217217
dtype=np.int32,
218-
suffix=self.parallel_config.engine_worker_queue_port,
218+
suffix=self.parallel_config.engine_pid,
219219
create=False,
220220
)
221221
self.worker_ready_signal.value[self.local_rank % self.max_chips_per_node] = 1
@@ -426,7 +426,7 @@ def initialize_kv_cache(self) -> None:
426426
name="get_profile_block_num",
427427
array=get_profile_block_num,
428428
dtype=np.int32,
429-
suffix=self.parallel_config.engine_worker_queue_port,
429+
suffix=self.parallel_config.engine_pid,
430430
create=False,
431431
)
432432
self.get_profile_block_num_signal.value[0] = num_blocks_local
@@ -469,7 +469,7 @@ def load_model(self) -> None:
469469
name="loaded_model_signal",
470470
array=loaded_model_signal_data,
471471
dtype=np.int32,
472-
suffix=self.parallel_config.engine_worker_queue_port,
472+
suffix=self.parallel_config.engine_pid,
473473
create=False,
474474
)
475475
if self.ranks > 1:

0 commit comments

Comments
 (0)