From f90aa446629f74d4d7049771fe938e04b5c2d3ef Mon Sep 17 00:00:00 2001 From: Soila Kavulya Date: Tue, 21 Apr 2026 18:26:33 -0700 Subject: [PATCH] [NIXL][XPU]Fix nixl import on XPU (#40430) Signed-off-by: Soila Kavulya --- vllm/distributed/nixl_utils.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/vllm/distributed/nixl_utils.py b/vllm/distributed/nixl_utils.py index b2b433339b..2da37017a3 100644 --- a/vllm/distributed/nixl_utils.py +++ b/vllm/distributed/nixl_utils.py @@ -24,7 +24,7 @@ if "UCX_RCACHE_MAX_UNRELEASED" not in os.environ: os.environ["UCX_RCACHE_MAX_UNRELEASED"] = "1024" try: - if current_platform.is_cuda(): + if not current_platform.is_rocm(): from nixl._api import nixl_agent as NixlWrapper else: from rixl._api import nixl_agent as NixlWrapper @@ -35,7 +35,7 @@ except ImportError: NixlWrapper = None # type: ignore[assignment, misc] try: - if current_platform.is_cuda(): + if not current_platform.is_rocm(): from nixl._api import nixl_agent_config else: from rixl._api import nixl_agent_config @@ -44,7 +44,7 @@ except ImportError: logger.warning_once("NIXL agent config is not available") try: - if current_platform.is_cuda(): + if not current_platform.is_rocm(): from nixl._bindings import nixlXferTelemetry else: from rixl._bindings import nixlXferTelemetry