apps/FastDeploy
1
0
Fork 0
mirror of https://github.com/PaddlePaddle/FastDeploy.git synced 2026-04-23 00:17:25 +08:00
Code Issues Actions 21 Packages Projects Releases Wiki Activity

[Bugfix]fix model weight signal tensor num (#5900)

Browse Source
This commit is contained in:
gaoziyuan
2026-01-06 14:36:59 +08:00
committed by GitHub
parent 9445fbe054
commit e99ec4c9d5
1 changed files with 2 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
fastdeploy/worker/worker_process.py
+2 -1
View File
@@ -282,7 +282,8 @@ class PaddleDisWorkerProc:
def _broadcast_model_weights_signal(self, src: int, group) -> int:
model_weights_signal_tensor = paddle.full(shape=[1], fill_value=self.model_weights_signal[0], dtype="int32")
paddle.distributed.broadcast(model_weights_signal_tensor, src=src, group=group)
return model_weights_signal_tensor.item()
value = model_weights_signal_tensor.numpy()[0]
return int(value)
def _tp_barrier_wait(self):
if current_platform.is_xpu():