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: ...@@ -24,7 +24,7 @@ if "UCX_RCACHE_MAX_UNRELEASED" not in os.environ:
os.environ["UCX_RCACHE_MAX_UNRELEASED"] = "1024" os.environ["UCX_RCACHE_MAX_UNRELEASED"] = "1024"
try: try:
if current_platform.is_cuda(): if not current_platform.is_rocm():
from nixl._api import nixl_agent as NixlWrapper from nixl._api import nixl_agent as NixlWrapper
else: else:
from rixl._api import nixl_agent as NixlWrapper from rixl._api import nixl_agent as NixlWrapper
...@@ -35,7 +35,7 @@ except ImportError: ...@@ -35,7 +35,7 @@ except ImportError:
NixlWrapper = None # type: ignore[assignment, misc] NixlWrapper = None # type: ignore[assignment, misc]
try: try:
if current_platform.is_cuda(): if not current_platform.is_rocm():
from nixl._api import nixl_agent_config from nixl._api import nixl_agent_config
else: else:
from rixl._api import nixl_agent_config from rixl._api import nixl_agent_config
...@@ -44,7 +44,7 @@ except ImportError: ...@@ -44,7 +44,7 @@ except ImportError:
logger.warning_once("NIXL agent config is not available") logger.warning_once("NIXL agent config is not available")
try: try:
if current_platform.is_cuda(): if not current_platform.is_rocm():
from nixl._bindings import nixlXferTelemetry from nixl._bindings import nixlXferTelemetry
else: else:
from rixl._bindings import nixlXferTelemetry 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