ROCm: sgl-kernel enablement starting with sgl_moe_align_block (#3287)

This commit is contained in:
HAI
2025-02-04 05:44:44 -08:00
committed by GitHub
parent d39899e85c
commit 2c1a695ff1
6 changed files with 131 additions and 13 deletions

View File

@@ -28,6 +28,9 @@ RUN git clone ${SGL_REPO} \
echo "Using ${SGL_BRANCH} branch."; \
git checkout ${SGL_BRANCH}; \
fi \
&& cd sgl-kernel \
&& python setup_rocm.py install \
&& cd .. \
&& if [ "$BUILD_TYPE" = "srt" ]; then \
python -m pip --no-cache-dir install -e "python[srt_hip]"; \
else \