Skip to content

Commit 1fc5026

Browse files
feat: enable intel XPU Dockerfile for Dev Target (#7134)
Signed-off-by: Hongming Zheng <hongming.zheng@intel.com> Co-authored-by: Zhan Xue <zhan.xue@intel.com>
1 parent bddaaa2 commit 1fc5026

File tree

2 files changed

+16
-0
lines changed

2 files changed

+16
-0
lines changed

container/templates/dev.Dockerfile

Lines changed: 11 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -206,6 +206,11 @@ RUN --mount=from=wheel_builder,target=/wheel_builder \
206206
fi; \
207207
fi
208208

209+
{% if device == "xpu" %}
210+
ENV NIXL_LIB_DIR=/opt/intel/intel_nixl/lib/x86_64-linux-gnu \
211+
NIXL_PLUGIN_DIR=/opt/intel/intel_nixl/lib/x86_64-linux-gnu/plugins \
212+
NIXL_PREFIX=/opt/intel/intel_nixl
213+
{% else %}
209214
# NIXL is installed under lib64 (manylinux/AlmaLinux convention used by the wheel_builder).
210215
# All frameworks reference NIXL_LIB_DIR=/opt/nvidia/nvda_nixl/lib64.
211216
# For vllm/trtllm/none: This resets the same values already set in runtime (no harm).
@@ -227,6 +232,7 @@ ENV CUDA_HOME=/usr/local/cuda \
227232
TRITON_PTXAS_PATH=/usr/local/cuda/bin/ptxas \
228233
TRITON_CUDART_PATH=/usr/local/cuda/include \
229234
NVIDIA_DRIVER_CAPABILITIES=video,compute,utility
235+
{% endif %}
230236

231237
# Base LD_LIBRARY_PATH with universal paths (all frameworks have these)
232238
# Framework-specific paths are conditionally added in /etc/profile.d/50-framework-paths.sh
@@ -362,5 +368,10 @@ RUN --mount=type=bind,source=./container/launch_message/dev.txt,target=/opt/dyna
362368
chmod 755 /opt/dynamo/.launch_screen && \
363369
echo 'cat /opt/dynamo/.launch_screen' >> /etc/bash.bashrc
364370

371+
{% if device == "xpu" %}
372+
SHELL ["bash", "-c"]
373+
CMD ["bash", "-c", "source /root/.bashrc && exec bash"]
374+
{% else %}
365375
ENTRYPOINT ["/opt/nvidia/nvidia_entrypoint.sh"]
366376
CMD []
377+
{% endif %}

container/templates/local_dev.Dockerfile

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -78,5 +78,10 @@ RUN mkdir -p /home/$USERNAME/.cache/ \
7878
&& chmod g+w /home/$USERNAME/.cache/ \
7979
&& chmod g+w /home/$USERNAME/.cache/pre-commit
8080

81+
{% if device == "xpu" %}
82+
SHELL ["bash", "-c"]
83+
CMD ["bash", "-c", "source /root/.bashrc && exec bash"]
84+
{% else %}
8185
ENTRYPOINT ["/opt/nvidia/nvidia_entrypoint.sh"]
8286
CMD []
87+
{% endif %}

0 commit comments

Comments
 (0)