"vscode:/vscode.git/clone" did not exist on "4a151dd45308f812d125b6ea239b1730b7f647c4"
Unverified Commit 1fc50263 authored by ZhengHongming888's avatar ZhengHongming888 Committed by GitHub
Browse files

feat: enable intel XPU Dockerfile for Dev Target (#7134)


Signed-off-by: default avatarHongming Zheng <hongming.zheng@intel.com>
Co-authored-by: default avatarZhan Xue <zhan.xue@intel.com>
parent bddaaa26
......@@ -206,6 +206,11 @@ RUN --mount=from=wheel_builder,target=/wheel_builder \
fi; \
fi
{% if device == "xpu" %}
ENV NIXL_LIB_DIR=/opt/intel/intel_nixl/lib/x86_64-linux-gnu \
NIXL_PLUGIN_DIR=/opt/intel/intel_nixl/lib/x86_64-linux-gnu/plugins \
NIXL_PREFIX=/opt/intel/intel_nixl
{% else %}
# NIXL is installed under lib64 (manylinux/AlmaLinux convention used by the wheel_builder).
# All frameworks reference NIXL_LIB_DIR=/opt/nvidia/nvda_nixl/lib64.
# 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 \
TRITON_PTXAS_PATH=/usr/local/cuda/bin/ptxas \
TRITON_CUDART_PATH=/usr/local/cuda/include \
NVIDIA_DRIVER_CAPABILITIES=video,compute,utility
{% endif %}
# Base LD_LIBRARY_PATH with universal paths (all frameworks have these)
# 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
chmod 755 /opt/dynamo/.launch_screen && \
echo 'cat /opt/dynamo/.launch_screen' >> /etc/bash.bashrc
{% if device == "xpu" %}
SHELL ["bash", "-c"]
CMD ["bash", "-c", "source /root/.bashrc && exec bash"]
{% else %}
ENTRYPOINT ["/opt/nvidia/nvidia_entrypoint.sh"]
CMD []
{% endif %}
......@@ -78,5 +78,10 @@ RUN mkdir -p /home/$USERNAME/.cache/ \
&& chmod g+w /home/$USERNAME/.cache/ \
&& chmod g+w /home/$USERNAME/.cache/pre-commit
{% if device == "xpu" %}
SHELL ["bash", "-c"]
CMD ["bash", "-c", "source /root/.bashrc && exec bash"]
{% else %}
ENTRYPOINT ["/opt/nvidia/nvidia_entrypoint.sh"]
CMD []
{% endif %}
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment