From 2d975e16b0f38cd476f2b572a46d4d40301591fb Mon Sep 17 00:00:00 2001 From: lizhenyun01 <1500424927@qq.com> Date: Fri, 5 Sep 2025 22:29:26 +0800 Subject: [PATCH] [BugFix] fix TaskQueue dp_id in multi node (#3919) --- fastdeploy/engine/common_engine.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/fastdeploy/engine/common_engine.py b/fastdeploy/engine/common_engine.py index 0489e3017..38eae1da7 100644 --- a/fastdeploy/engine/common_engine.py +++ b/fastdeploy/engine/common_engine.py @@ -225,7 +225,8 @@ class EngineSevice: client_id=0, local_data_parallel_size=self.cfg.parallel_config.data_parallel_size, local_data_parallel_id=min( - self.cfg.worker_num_per_node * self.cfg.node_rank + self.cfg.parallel_config.local_data_parallel_id, + self.cfg.worker_num_per_node // self.cfg.parallel_config.tensor_parallel_size * self.cfg.node_rank + + self.cfg.parallel_config.local_data_parallel_id, self.cfg.parallel_config.data_parallel_size - 1, ), )