[AMD] Fix aiter version in rocm image (#18076)
This commit is contained in:
@@ -98,6 +98,10 @@ RUN if [ "$BUILD_LLVM" = "1" ]; then \
|
|||||||
|
|
||||||
# -----------------------
|
# -----------------------
|
||||||
# AITER
|
# AITER
|
||||||
|
# Unset setuptools_scm override so AITER gets its own version (AITER_COMMIT), not SGLang's
|
||||||
|
# (SETUPTOOLS_SCM_PRETEND_VERSION is set later for SGLang nightly builds and would otherwise
|
||||||
|
# leak into AITER's version when AITER uses setuptools_scm)
|
||||||
|
ENV SETUPTOOLS_SCM_PRETEND_VERSION=
|
||||||
RUN pip uninstall -y aiter
|
RUN pip uninstall -y aiter
|
||||||
RUN git clone ${AITER_REPO} \
|
RUN git clone ${AITER_REPO} \
|
||||||
&& cd aiter \
|
&& cd aiter \
|
||||||
@@ -164,9 +168,9 @@ RUN if [ "$BUILD_MOONCAKE" = "1" ]; then \
|
|||||||
# Build SGLang
|
# Build SGLang
|
||||||
ARG BUILD_TYPE=all
|
ARG BUILD_TYPE=all
|
||||||
|
|
||||||
# Set version for setuptools_scm if provided (for nightly builds)
|
# Set version for setuptools_scm if provided (for nightly builds). Only pass in the SGLang
|
||||||
|
# pip install RUN so it does not affect AITER, sgl-model-gateway, TileLang, FHT, MORI, etc.
|
||||||
ARG SETUPTOOLS_SCM_PRETEND_VERSION
|
ARG SETUPTOOLS_SCM_PRETEND_VERSION
|
||||||
ENV SETUPTOOLS_SCM_PRETEND_VERSION=${SETUPTOOLS_SCM_PRETEND_VERSION}
|
|
||||||
|
|
||||||
RUN pip install IPython \
|
RUN pip install IPython \
|
||||||
&& pip install orjson \
|
&& pip install orjson \
|
||||||
@@ -191,9 +195,9 @@ RUN git clone ${SGL_REPO} \
|
|||||||
&& cd .. \
|
&& cd .. \
|
||||||
&& rm -rf python/pyproject.toml && mv python/pyproject_other.toml python/pyproject.toml \
|
&& rm -rf python/pyproject.toml && mv python/pyproject_other.toml python/pyproject.toml \
|
||||||
&& if [ "$BUILD_TYPE" = "srt" ]; then \
|
&& if [ "$BUILD_TYPE" = "srt" ]; then \
|
||||||
python -m pip --no-cache-dir install -e "python[srt_hip,diffusion_hip]"; \
|
export SETUPTOOLS_SCM_PRETEND_VERSION="${SETUPTOOLS_SCM_PRETEND_VERSION}" && python -m pip --no-cache-dir install -e "python[srt_hip,diffusion_hip]"; \
|
||||||
else \
|
else \
|
||||||
python -m pip --no-cache-dir install -e "python[all_hip]"; \
|
export SETUPTOOLS_SCM_PRETEND_VERSION="${SETUPTOOLS_SCM_PRETEND_VERSION}" && python -m pip --no-cache-dir install -e "python[all_hip]"; \
|
||||||
fi
|
fi
|
||||||
|
|
||||||
RUN python -m pip cache purge
|
RUN python -m pip cache purge
|
||||||
|
|||||||
Reference in New Issue
Block a user