[XPU][CI] enhance xpu test support (#20652)

Signed-off-by: Ma, Liangliang <liangliang.ma@intel.com> Co-authored-by: zhenwei-intel <zhenweiliu@habana.ai>
2025-07-10 00:53:09 +08:00
parent eb58f5953d
commit a3e4e85ece
5 changed files with 18 additions and 12 deletions
--- a/vllm/distributed/device_communicators/xpu_communicator.py
+++ b/vllm/distributed/device_communicators/xpu_communicator.py
@@ -53,3 +53,6 @@ class XpuCommunicator(DeviceCommunicatorBase):
        else:
            output_tensor = None
        return output_tensor
+
+    def broadcast(self, input_: torch.Tensor, src: int = 0) -> None:
+        dist.broadcast(input_, src=src, group=self.device_group)