[AMD] ROCm7.2: Add /sgl-workspace/aiter to PYTHONPATH (#18972)

2026-02-18 02:21:39 -08:00
parent 90d5e27f79
commit 0215d47007
1 changed files with 2 additions and 1 deletions
@@ -156,7 +156,8 @@ RUN cd aiter \
          sh -c "PREBUILD_KERNELS=1 GPU_ARCHS=$GPU_ARCH_LIST python setup.py develop"; \
        else \
          sh -c "GPU_ARCHS=$GPU_ARCH_LIST python setup.py develop"; \
-        fi
+        fi \
+     && echo "export PYTHONPATH=/sgl-workspace/aiter:\${PYTHONPATH}" >> /etc/bash.bashrc

 # -----------------------
 # Build vLLM