xinyun/vllm
1
0
Fork 0
mirror of https://git.datalinker.icu/vllm-project/vllm.git synced 2026-05-26 14:17:10 +08:00
Code Issues Packages Projects Releases Wiki Activity

barrier

Browse Source 
Signed-off-by: Tyler Michael Smith <tyler@neuralmagic.com>
This commit is contained in:
Tyler Michael Smith 2025-07-10 18:20:25 -04:00
parent 7b86860ff5
commit 29596317b0
1 changed files with 1 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
vllm/distributed/parallel_state.py
View File

@ -782,6 +782,7 @@ class GroupCoordinator:
def prepare_communication_buffer_for_model(self, model: torch.nn.Module): def prepare_communication_buffer_for_model(self, model: torch.nn.Module):
if self.device_communicator is not None: if self.device_communicator is not None:
torch.distributed.barrier(self.device_communicator)
self.device_communicator.prepare_communication_buffer_for_model( self.device_communicator.prepare_communication_buffer_for_model(
model) model)