minke.yu/sglang
1
0
Fork 0
Code Issues Pull Requests Actions 13 Packages Projects Releases Wiki Activity

[AMD][Diffusion] support timestep embedding kernel for AMD GPUs (#16766)

Browse Source
This commit is contained in:
Hubert Lu
2026-01-12 22:17:07 -08:00
committed by GitHub
parent ff3ddb9d9b
commit 8716589826
9 changed files with 40 additions and 19 deletions
Download Patch File Download Diff File Expand all files Collapse all files
docker/rocm.Dockerfile
+2 -5
View File
@@ -21,7 +21,6 @@ ENV BUILD_LLVM="0"
ENV BUILD_AITER_ALL="1"
ENV BUILD_MOONCAKE="1"
ENV AITER_COMMIT="v0.1.4"
ENV NO_DEPS_FLAG=""
# ===============================
# Base image 942 and args
@@ -32,7 +31,6 @@ ENV BUILD_LLVM="0"
ENV BUILD_AITER_ALL="1"
ENV BUILD_MOONCAKE="1"
ENV AITER_COMMIT="v0.1.9.post1"
ENV NO_DEPS_FLAG=""
# ===============================
# Base image 950 and args
@@ -43,7 +41,6 @@ ENV BUILD_LLVM="0"
ENV BUILD_AITER_ALL="0"
ENV BUILD_MOONCAKE="1"
ENV AITER_COMMIT="v0.1.9.post1"
ENV NO_DEPS_FLAG=""
# ===============================
# Chosen arch and args
FROM ${GPU_ARCH}
@@ -187,9 +184,9 @@ RUN git clone ${SGL_REPO} \
&& cd .. \
&& rm -rf python/pyproject.toml && mv python/pyproject_other.toml python/pyproject.toml \
&& if [ "$BUILD_TYPE" = "srt" ]; then \
python -m pip --no-cache-dir install -e "python[srt_hip,diffusion]" ${NO_DEPS_FLAG}; \
python -m pip --no-cache-dir install -e "python[srt_hip,diffusion]"; \
else \
python -m pip --no-cache-dir install -e "python[all_hip,diffusion]" ${NO_DEPS_FLAG}; \
python -m pip --no-cache-dir install -e "python[all_hip,diffusion]"; \
fi
RUN python -m pip cache purge