apps/FastDeploy
1
0
Fork 0
mirror of https://github.com/PaddlePaddle/FastDeploy.git synced 2025-10-02 15:22:24 +08:00
Code Issues Actions 2 Packages Projects Releases Wiki Activity

[Code Simplification] fix init_distributed_environment() (#2982)

Browse Source
This commit is contained in:
lizexu123
2025-07-24 11:43:28 +08:00
committed by GitHub
parent bfeb664ab8
commit 832d25334a
2 changed files with 5 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
fastdeploy/worker/worker_process.py
View File

@@ -17,7 +17,7 @@
import argparse
import json
import time
from typing import List
from typing import Tuple
import numpy as np
import paddle
@@ -74,7 +74,7 @@ def get_worker(fd_config: FDConfig, local_rank: int, rank: int) -> WorkerBase:
return GcuWorker(fd_config=fd_config, local_rank=local_rank, rank=rank)
def init_distributed_environment(seed: int = 20) -> List[int]:
def init_distributed_environment(seed: int = 20) -> Tuple[int, int]:
"""Initialize Paddle Fleet and get rank of worker"""
# Global rank
ranks = dist.get_world_size()
@@ -122,9 +122,9 @@ def update_fd_config_for_mm(fd_config: FDConfig) -> None:
class PaddleDisWorkerProc:
"""
Paddle Distrubuted wrapper for fastdeploy.worker.Worker,
Paddle Distributed wrapper for fastdeploy.worker.Worker,
for handling single-node multi-GPU tensor parallel.
The wrapper internally executea an event loop that continuously executes requests
The wrapper internally executes an event loop that continuously executes requests
in the task queue. Control flow is transmitted by IPC.
"""