ROCm: sgl-kernel enablement starting with sgl_moe_align_block (#3287)
This commit is contained in:
@@ -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 \
|
||||
|
||||
Reference in New Issue
Block a user