[CI][NIXL] Fix NIXL EP import canary for the nixl 1.3.0 wheel and pin nixl==1.3.0 (#45166)

Signed-off-by: Ovidiu Mara <ovidium@nvidia.com>
Signed-off-by: ovidiusm <ovidium@nvidia.com>
Co-authored-by: Wentao Ye <44945378+yewentao256@users.noreply.github.com>
This commit is contained in:
ovidiusm
2026-06-25 19:33:42 -07:00
committed by GitHub
co-authored by Wentao Ye
parent 552a9dbe59
commit dbc49b6b99
2 changed files with 4 additions and 24 deletions
+1 -1
View File
@@ -2,5 +2,5 @@ lmcache >= 0.3.9
# CuPy 14.1.0 imports pytest from cupy.testing._random. Use <14.1.0
# until a fixed newer release is verified for runtime images.
cupy-cuda13x < 14.1.0
nixl == 1.2.0 # Required for disaggregated prefill
nixl == 1.3.0
mooncake-transfer-engine >= 0.3.8
@@ -4,8 +4,6 @@
import importlib
import importlib.metadata as metadata
import subprocess
import sys
import types
import pytest
@@ -65,25 +63,7 @@ def test_nixl_and_nixl_ep_imports() -> None:
# Exercise the NIXL EP extension used by fused MoE expert parallelism.
nixl_ep = importlib.import_module("nixl_ep")
print(f"nixl_ep: {nixl_ep.__file__}")
assert nixl_ep.__file__ is not None
nixl_ep_cpp = _import_nixl_ep_cpp(nixl_ep)
assert nixl_ep_cpp.__file__ is not None
extension_file = nixl_ep_cpp.__file__
print(f"nixl_ep_cpp: {extension_file}")
completed = subprocess.run(
["ldd", extension_file],
capture_output=True,
check=False,
text=True,
)
print(completed.stdout)
if completed.stderr:
print(completed.stderr, file=sys.stderr)
assert completed.returncode == 0
if torch.version.cuda is not None:
cuda_major = torch.version.cuda.split(".", maxsplit=1)[0]
expected_cudart = f"libcudart.so.{cuda_major}"
assert expected_cudart in completed.stdout
assert f"{expected_cudart} => not found" not in completed.stdout
# Check that the NIXL EP extension is loaded.
assert nixl_ep.Config is not None