Skip to content

Commit d8e5832

Browse files
committed
fix ci
1 parent c043fcc commit d8e5832

File tree

1 file changed

+5
-9
lines changed

1 file changed

+5
-9
lines changed

fastdeploy/worker/worker_process.py

Lines changed: 5 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -192,17 +192,13 @@ def init_health_status(self) -> None:
192192
)
193193
self.worker_ready_signal.value[self.local_rank % self.max_chips_per_node] = 1
194194

195-
if self.parallel_config.local_data_parallel_id == 0:
196-
current_suffix = self.parallel_config.engine_pid
197-
else:
198-
current_suffix = self.parallel_config.engine_worker_queue_port
199195
# init worker_healthy_live_signal
200196
workers_alive = np.zeros(shape=[min(array_size, self.parallel_config.tensor_parallel_size)], dtype=np.int32)
201197
self.worker_healthy_live_signal = IPCSignal(
202198
name="worker_healthy_live_signal",
203199
array=workers_alive,
204200
dtype=np.int32,
205-
suffix=current_suffix,
201+
suffix=self.parallel_config.engine_worker_queue_port,
206202
create=False,
207203
)
208204
local_rank = self.local_rank % self.parallel_config.tensor_parallel_size
@@ -214,7 +210,7 @@ def init_health_status(self) -> None:
214210
name="model_weights_status",
215211
array=workers_model_weights,
216212
dtype=np.int32,
217-
suffix=current_suffix,
213+
suffix=self.parallel_config.engine_worker_queue_port,
218214
create=False,
219215
)
220216

@@ -224,7 +220,7 @@ def init_health_status(self) -> None:
224220
name="exist_task_signal",
225221
array=workers_exist_task,
226222
dtype=np.int32,
227-
suffix=current_suffix,
223+
suffix=self.parallel_config.engine_worker_queue_port,
228224
create=False,
229225
)
230226

@@ -234,7 +230,7 @@ def init_health_status(self) -> None:
234230
name="exist_swapped_task_signal",
235231
array=workers_swapped_task,
236232
dtype=np.int32,
237-
suffix=current_suffix,
233+
suffix=self.parallel_config.engine_worker_queue_port,
238234
create=False,
239235
)
240236

@@ -244,7 +240,7 @@ def init_health_status(self) -> None:
244240
name="exist_prefill_task_signal",
245241
array=exist_prefill_task_signal_data,
246242
dtype=np.int32,
247-
suffix=current_suffix,
243+
suffix=self.parallel_config.engine_worker_queue_port,
248244
create=False,
249245
)
250246

0 commit comments

Comments
 (0)