Skip to content

Commit 791b101

Browse files
revert worker process ipc signal suffix (#4323)
1 parent af38722 commit 791b101

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
@@ -183,7 +183,7 @@ def init_health_status(self) -> None:
183183
name="launched_expert_service_signal",
184184
array=launched_expert_service_signal_data,
185185
dtype=np.int32,
186-
suffix=self.parallel_config.engine_worker_queue_port,
186+
suffix=self.parallel_config.engine_pid,
187187
create=False,
188188
)
189189
while self.launched_expert_service_signal.value[self.local_rank % self.max_chips_per_node] == 0:
@@ -200,7 +200,7 @@ def init_health_status(self) -> None:
200200
name="worker_ready_signal",
201201
array=workers_ready,
202202
dtype=np.int32,
203-
suffix=self.parallel_config.engine_worker_queue_port,
203+
suffix=self.parallel_config.engine_pid,
204204
create=False,
205205
)
206206
self.worker_ready_signal.value[self.local_rank % self.max_chips_per_node] = 1
@@ -418,7 +418,7 @@ def initialize_kv_cache(self) -> None:
418418
name="get_profile_block_num",
419419
array=get_profile_block_num,
420420
dtype=np.int32,
421-
suffix=self.parallel_config.engine_worker_queue_port,
421+
suffix=self.parallel_config.engine_pid,
422422
create=False,
423423
)
424424
self.get_profile_block_num_signal.value[0] = num_blocks_local
@@ -473,7 +473,7 @@ def load_model(self) -> None:
473473
name="loaded_model_signal",
474474
array=loaded_model_signal_data,
475475
dtype=np.int32,
476-
suffix=self.parallel_config.engine_worker_queue_port,
476+
suffix=self.parallel_config.engine_pid,
477477
create=False,
478478
)
479479
if self.ranks > 1:

0 commit comments

Comments
 (0)