use self-hosted to build sgl-kernel (#3154)

2025-01-26 23:02:57 +08:00
parent 02431b9ad2
commit f265d15b96
3 changed files with 14 additions and 2 deletions
--- a/sgl-kernel/build.sh
+++ b/sgl-kernel/build.sh
@@ -15,6 +15,7 @@ docker run --rm \
    pytorch/manylinux-builder:cuda${CUDA_VERSION} \
    bash -c "
    ${PYTHON_ROOT_PATH}/bin/pip install --no-cache-dir torch==2.5.1 --index-url https://download.pytorch.org/whl/cu${CUDA_VERSION//.} && \
+    ${PYTHON_ROOT_PATH}/bin/pip install --no-cache-dir ninja && \
    export TORCH_CUDA_ARCH_LIST='7.5 8.0 8.9 9.0+PTX' && \
    export CUDA_VERSION=${CUDA_VERSION} && \
    export SGL_KERNEL_ENABLE_BF16=1 && \