Skip to content

Commit 1494937

Browse files
committed
[NIXL][XPU] pin nixl version
Signed-off-by: zhenwei-intel <[email protected]>
1 parent b2e65cb commit 1494937

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

tools/install_nixl_from_source_ubuntu.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -9,6 +9,7 @@
99

1010
# --- Configuration ---
1111
WHEELS_CACHE_HOME = os.environ.get("WHEELS_CACHE_HOME", "/tmp/wheels_cache")
12+
NIXL_VERSION = os.environ.get("NIXL_VERSION", "0.7.0")
1213
ROOT_DIR = os.path.dirname(os.path.abspath(__file__))
1314
UCX_DIR = os.path.join("/tmp", "ucx_source")
1415
NIXL_DIR = os.path.join("/tmp", "nixl_source")
@@ -37,7 +38,7 @@ def is_pip_package_installed(package_name):
3738
def find_nixl_wheel_in_cache(cache_dir):
3839
"""Finds a nixl wheel file in the specified cache directory."""
3940
# The repaired wheel will have a 'manylinux' tag, but this glob still works.
40-
search_pattern = os.path.join(cache_dir, "nixl*.whl")
41+
search_pattern = os.path.join(cache_dir, f"nixl*{NIXL_VERSION}*.whl")
4142
wheels = glob.glob(search_pattern)
4243
if wheels:
4344
# Sort to get the most recent/highest version if multiple exist
@@ -146,6 +147,7 @@ def build_and_install_prerequisites(args):
146147
print("\n[2/3] Building NIXL wheel from source...", flush=True)
147148
if not os.path.exists(NIXL_DIR):
148149
run_command(["git", "clone", NIXL_REPO_URL, NIXL_DIR])
150+
run_command(["git", "checkout", NIXL_VERSION], cwd=NIXL_DIR)
149151

150152
build_env = os.environ.copy()
151153
build_env["PKG_CONFIG_PATH"] = os.path.join(ucx_install_path, "lib", "pkgconfig")

0 commit comments

Comments
 (0)