Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in
Toggle navigation
Menu
Open sidebar
OpenDAS
vllm_cscc
Commits
d93d2d74
Unverified
Commit
d93d2d74
authored
Jan 07, 2025
by
YiSheng5
Committed by
GitHub
Jan 07, 2025
Browse files
[XPU] Make pp group initilized for pipeline-parallelism (#11648)
Signed-off-by:
yisheng
<
yi.sheng@intel.com
>
parent
d0169e1b
Changes
1
Show whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
6 additions
and
0 deletions
+6
-0
vllm/worker/xpu_worker.py
vllm/worker/xpu_worker.py
+6
-0
No files found.
vllm/worker/xpu_worker.py
View file @
d93d2d74
...
@@ -11,6 +11,7 @@ import torch.distributed
...
@@ -11,6 +11,7 @@ import torch.distributed
from
vllm.config
import
VllmConfig
from
vllm.config
import
VllmConfig
from
vllm.distributed
import
(
ensure_model_parallel_initialized
,
from
vllm.distributed
import
(
ensure_model_parallel_initialized
,
init_distributed_environment
)
init_distributed_environment
)
from
vllm.distributed.parallel_state
import
get_pp_group
from
vllm.logger
import
init_logger
from
vllm.logger
import
init_logger
from
vllm.model_executor
import
set_random_seed
from
vllm.model_executor
import
set_random_seed
from
vllm.platforms
import
current_platform
from
vllm.platforms
import
current_platform
...
@@ -176,3 +177,8 @@ class XPUWorker(LoraNotSupportedWorkerBase, Worker):
...
@@ -176,3 +177,8 @@ class XPUWorker(LoraNotSupportedWorkerBase, Worker):
parallel_config
.
pipeline_parallel_size
)
parallel_config
.
pipeline_parallel_size
)
# global all_reduce needed for overall oneccl warm up
# global all_reduce needed for overall oneccl warm up
torch
.
distributed
.
all_reduce
(
torch
.
zeros
(
1
).
xpu
())
torch
.
distributed
.
all_reduce
(
torch
.
zeros
(
1
).
xpu
())
if
parallel_config
.
pipeline_parallel_size
>
1
:
# Add pp group init to avoid
# p2p communication as the first call
get_pp_group
().
all_reduce
(
torch
.
zeros
(
1
).
xpu
())
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment