misc: update build setup (#2306)

This commit is contained in:
Yineng Zhang
2024-12-02 02:03:49 +08:00
committed by GitHub
parent 7301a39b13
commit 28bc60dcab
5 changed files with 46 additions and 1 deletions

13
sgl-kernel/build.sh Executable file
View File

@@ -0,0 +1,13 @@
#!/bin/bash
set -ex
docker run --rm -it \
-v "$(pwd)":/sgl-kernel \
pytorch/manylinux-builder:cuda12.1 \
bash -c "
pip install --no-cache-dir torch==2.4.0 --index-url https://download.pytorch.org/whl/cu121 && \
export TORCH_CUDA_ARCH_LIST='7.5 8.0 8.9 9.0+PTX' && \
cd /sgl-kernel && \
python setup.py bdist_wheel
"