[AMD] ROCm7.2: Add /sgl-workspace/aiter to PYTHONPATH (#18972)
This commit is contained in:
@@ -156,7 +156,8 @@ RUN cd aiter \
|
|||||||
sh -c "PREBUILD_KERNELS=1 GPU_ARCHS=$GPU_ARCH_LIST python setup.py develop"; \
|
sh -c "PREBUILD_KERNELS=1 GPU_ARCHS=$GPU_ARCH_LIST python setup.py develop"; \
|
||||||
else \
|
else \
|
||||||
sh -c "GPU_ARCHS=$GPU_ARCH_LIST python setup.py develop"; \
|
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
|
# Build vLLM
|
||||||
|
|||||||
Reference in New Issue
Block a user