From 08cb436d4e9271fe6272702fe10da768aad42df9 Mon Sep 17 00:00:00 2001 From: zdoba Date: Wed, 1 Apr 2026 16:41:25 +0800 Subject: [PATCH] Fix: ensure input tensor is contiguous in GroupCoordinator.all_gather (#2367) Signed-off-by: daixinning Co-authored-by: daixinning --- vllm_omni/diffusion/distributed/group_coordinator.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/vllm_omni/diffusion/distributed/group_coordinator.py b/vllm_omni/diffusion/distributed/group_coordinator.py index b722f61c0..8ab38f2a6 100644 --- a/vllm_omni/diffusion/distributed/group_coordinator.py +++ b/vllm_omni/diffusion/distributed/group_coordinator.py @@ -213,7 +213,7 @@ class GroupCoordinator: input_size[0] *= world_size output_tensor = torch.empty(input_size, dtype=input_.dtype, device=input_.device) # All-gather. - torch.distributed.all_gather_into_tensor(output_tensor, input_, group=self.device_group) + torch.distributed.all_gather_into_tensor(output_tensor, input_.contiguous(), group=self.device_group) if dim != 0: input_size[0] //= world_size output_tensor = output_tensor.reshape(