"lib/runtime/vscode:/vscode.git/clone" did not exist on "8c75ed799170e2c8d7f3df42eab6af7ecbd4b5eb"
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 \ ...@@ -206,6 +206,11 @@ RUN --mount=from=wheel_builder,target=/wheel_builder \
fi; \ fi; \
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). # NIXL is installed under lib64 (manylinux/AlmaLinux convention used by the wheel_builder).
# All frameworks reference NIXL_LIB_DIR=/opt/nvidia/nvda_nixl/lib64. # 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). # 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 \ ...@@ -227,6 +232,7 @@ ENV CUDA_HOME=/usr/local/cuda \
TRITON_PTXAS_PATH=/usr/local/cuda/bin/ptxas \ TRITON_PTXAS_PATH=/usr/local/cuda/bin/ptxas \
TRITON_CUDART_PATH=/usr/local/cuda/include \ TRITON_CUDART_PATH=/usr/local/cuda/include \
NVIDIA_DRIVER_CAPABILITIES=video,compute,utility NVIDIA_DRIVER_CAPABILITIES=video,compute,utility
{% endif %}
# Base LD_LIBRARY_PATH with universal paths (all frameworks have these) # 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 # 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 ...@@ -362,5 +368,10 @@ RUN --mount=type=bind,source=./container/launch_message/dev.txt,target=/opt/dyna
chmod 755 /opt/dynamo/.launch_screen && \ chmod 755 /opt/dynamo/.launch_screen && \
echo 'cat /opt/dynamo/.launch_screen' >> /etc/bash.bashrc 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"] ENTRYPOINT ["/opt/nvidia/nvidia_entrypoint.sh"]
CMD [] CMD []
{% endif %}
...@@ -78,5 +78,10 @@ RUN mkdir -p /home/$USERNAME/.cache/ \ ...@@ -78,5 +78,10 @@ RUN mkdir -p /home/$USERNAME/.cache/ \
&& chmod g+w /home/$USERNAME/.cache/ \ && chmod g+w /home/$USERNAME/.cache/ \
&& chmod g+w /home/$USERNAME/.cache/pre-commit && 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"] ENTRYPOINT ["/opt/nvidia/nvidia_entrypoint.sh"]
CMD [] 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