mirror of
https://git.datalinker.icu/vllm-project/vllm.git
synced 2025-12-10 07:04:53 +08:00
[XPU] XCCL support enabled in torch 2.8.0.dev nightly builds (#20705)
Signed-off-by: ratnampa <ratnam.parikh@intel.com>
This commit is contained in:
parent
e2de455c34
commit
35514b682a
@ -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
|
||||
|
||||
Loading…
x
Reference in New Issue
Block a user