|
9 | 9 |
|
10 | 10 | # --- Configuration --- |
11 | 11 | WHEELS_CACHE_HOME = os.environ.get("WHEELS_CACHE_HOME", "/tmp/wheels_cache") |
| 12 | +NIXL_VERSION = os.environ.get("NIXL_VERSION", "0.7.0") |
12 | 13 | ROOT_DIR = os.path.dirname(os.path.abspath(__file__)) |
13 | 14 | UCX_DIR = os.path.join("/tmp", "ucx_source") |
14 | 15 | NIXL_DIR = os.path.join("/tmp", "nixl_source") |
@@ -37,7 +38,7 @@ def is_pip_package_installed(package_name): |
37 | 38 | def find_nixl_wheel_in_cache(cache_dir): |
38 | 39 | """Finds a nixl wheel file in the specified cache directory.""" |
39 | 40 | # 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") |
41 | 42 | wheels = glob.glob(search_pattern) |
42 | 43 | if wheels: |
43 | 44 | # Sort to get the most recent/highest version if multiple exist |
@@ -146,6 +147,7 @@ def build_and_install_prerequisites(args): |
146 | 147 | print("\n[2/3] Building NIXL wheel from source...", flush=True) |
147 | 148 | if not os.path.exists(NIXL_DIR): |
148 | 149 | run_command(["git", "clone", NIXL_REPO_URL, NIXL_DIR]) |
| 150 | + run_command(["git", "checkout", NIXL_VERSION], cwd=NIXL_DIR) |
149 | 151 |
|
150 | 152 | build_env = os.environ.copy() |
151 | 153 | build_env["PKG_CONFIG_PATH"] = os.path.join(ucx_install_path, "lib", "pkgconfig") |
|
0 commit comments