@@ -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 %}
365375ENTRYPOINT ["/opt/nvidia/nvidia_entrypoint.sh" ]
366376CMD []
377+ {% endif %}
0 commit comments