Unverified Commit f90aa446 authored by Soila Kavulya's avatar Soila Kavulya Committed by GitHub
Browse files

[NIXL][XPU]Fix nixl import on XPU (#40430)

Signed-off-by: Soila Kavulya <soila.p.kavulya.intel.com>
parent cefa5281
......@@ -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
......
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