Skip to content

Commit d64429b

Browse files
[NIXL][XPU] update install script of NIXL (#28778)
Signed-off-by: zhenwei-intel <[email protected]>
1 parent 561253b commit d64429b

File tree

2 files changed

+3
-1
lines changed

2 files changed

+3
-1
lines changed

docker/Dockerfile.xpu

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -14,6 +14,7 @@ RUN apt clean && apt-get update -y && \
1414
libxext6 \
1515
libgl1 \
1616
lsb-release \
17+
libaio-dev \
1718
numactl \
1819
wget \
1920
vim \
@@ -68,8 +69,8 @@ RUN --mount=type=cache,target=/root/.cache/pip \
6869
RUN python3 -m pip install -e tests/vllm_test_utils
6970

7071
# install nixl from source code
72+
ENV NIXL_VERSION=0.7.0
7173
RUN python3 /workspace/vllm/tools/install_nixl_from_source_ubuntu.py
72-
ENV LD_LIBRARY_PATH="$LD_LIBRARY_PATH:/usr/local/lib/python3.12/dist-packages/.nixl.mesonpy.libs/plugins/"
7374

7475
RUN --mount=type=cache,target=/root/.cache/pip \
7576
pip uninstall oneccl oneccl-devel -y

tools/install_nixl_from_source_ubuntu.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -175,6 +175,7 @@ def build_and_install_prerequisites(args):
175175
build_env["LD_LIBRARY_PATH"] = (
176176
f"{ucx_lib_path}:{ucx_plugin_path}:{existing_ld_path}".strip(":")
177177
)
178+
build_env["LDFLAGS"] = "-Wl,-rpath,$ORIGIN"
178179
print(f"--> Using LD_LIBRARY_PATH: {build_env['LD_LIBRARY_PATH']}", flush=True)
179180

180181
temp_wheel_dir = os.path.join(ROOT_DIR, "temp_wheelhouse")

0 commit comments

Comments
 (0)