From 1e70735c21980d3f52eb3184ed712e6faee1627c Mon Sep 17 00:00:00 2001 From: "tzulingk@nvidia.com" Date: Sat, 8 Nov 2025 21:08:45 -0800 Subject: [PATCH] Fix the wrong path of deep_ep kernel Signed-off-by: tzulingk@nvidia.com --- .../fault_tolerance/deploy/container/Dockerfile.local_vllm | 7 ++++--- 1 file changed, 4 insertions(+), 3 deletions(-) diff --git a/tests/fault_tolerance/deploy/container/Dockerfile.local_vllm b/tests/fault_tolerance/deploy/container/Dockerfile.local_vllm index 398ace0f2b..5a4631f9b7 100644 --- a/tests/fault_tolerance/deploy/container/Dockerfile.local_vllm +++ b/tests/fault_tolerance/deploy/container/Dockerfile.local_vllm @@ -105,9 +105,10 @@ COPY --from=vllm_source /usr/local/lib/python3.12/dist-packages ${VIRTUAL_ENV}/l RUN if [ -f ${VIRTUAL_ENV}/lib/python${PYTHON_VERSION}/site-packages/__editable__.pplx_kernels-0.0.1.pth ]; then \ echo "/opt/vllm/ep_kernels_workspace/pplx-kernels/src" > ${VIRTUAL_ENV}/lib/python${PYTHON_VERSION}/site-packages/__editable__.pplx_kernels-0.0.1.pth; \ fi && \ - if [ -f ${VIRTUAL_ENV}/lib/python${PYTHON_VERSION}/site-packages/__editable__.deep_ep-0.0.1.pth ]; then \ - echo "/opt/vllm/ep_kernels_workspace/DeepEP" > ${VIRTUAL_ENV}/lib/python${PYTHON_VERSION}/site-packages/__editable__.deep_ep-0.0.1.pth; \ - fi && \ + # Fix ALL DeepEP .pth files (version-agnostic using find since version includes git hash) + find ${VIRTUAL_ENV}/lib/python${PYTHON_VERSION}/site-packages -name "__editable__.deep_ep*.pth" -exec sed -i 's|/vllm-workspace|/opt/vllm|g' {} \; && \ + # Fix DeepEP finder modules (these contain the actual hardcoded MAPPING paths) + find ${VIRTUAL_ENV}/lib/python${PYTHON_VERSION}/site-packages -name "__editable___deep_ep*_finder.py" -exec sed -i "s|'/vllm-workspace|'/opt/vllm|g" {} \; && \ # Also check for any other .pth files that might reference /vllm-workspace find ${VIRTUAL_ENV}/lib/python${PYTHON_VERSION}/site-packages -name "*.pth" -exec sed -i 's|/vllm-workspace|/opt/vllm|g' {} \;