Skip to content

GitLab

Unverified Commit 35514b68 authored by Ratnam Parikh's avatar Ratnam Parikh Committed by GitHub
Browse files

[XPU] XCCL support enabled in torch 2.8.0.dev nightly builds (#20705) 


Signed-off-by: default avatarratnampa <ratnam.parikh@intel.com>
parent e2de455c
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
vllm/utils/__init__.py
View file @ 35514b68
......@@ -2085,10 +2085,10 @@ def supports_dynamo() -> bool:
return base_torch_version >= Version("2.4.0")
# Supports xccl with PyTorch versions >= 2.8.0 for XPU platform
# Supports xccl with PyTorch versions >= 2.8.0.dev for XPU platform
def supports_xccl() -> bool:
return is_torch_equal_or_newer(
"2.8.0") and torch.distributed.is_xccl_available()
"2.8.0.dev") and torch.distributed.is_xccl_available()
# Some backends use pytorch version < 2.4.0 which doesn't
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment