Unverified Commit 482c3db2 authored by fzyzcjy's avatar fzyzcjy Committed by GitHub
Browse files

Fix sgl-kernel arch and missing package in CI (#8869)

parent 47824c14
...@@ -13,9 +13,8 @@ fi ...@@ -13,9 +13,8 @@ fi
SCRIPT_DIR="$( cd "$( dirname "${BASH_SOURCE[0]}" )" && pwd )" SCRIPT_DIR="$( cd "$( dirname "${BASH_SOURCE[0]}" )" && pwd )"
bash "${SCRIPT_DIR}/killall_sglang.sh" bash "${SCRIPT_DIR}/killall_sglang.sh"
if ! command -v git >/dev/null 2>&1; then if [ "$MODE_BLACKWELL" = "1" ]; then
apt update apt-get install -y git libnuma-dev
apt install -y git
fi fi
# Update pip # Update pip
...@@ -34,6 +33,12 @@ rm -rf /usr/local/lib/python3.10/dist-packages/sgl_kernel* ...@@ -34,6 +33,12 @@ rm -rf /usr/local/lib/python3.10/dist-packages/sgl_kernel*
# Install the main package # Install the main package
pip install -e "python[dev]" --extra-index-url https://download.pytorch.org/whl/${CU_VERSION} --break-system-packages pip install -e "python[dev]" --extra-index-url https://download.pytorch.org/whl/${CU_VERSION} --break-system-packages
if [ "$MODE_BLACKWELL" = "1" ]; then
# TODO auto determine sgl-kernel version
SGL_KERNEL_VERSION=0.3.2
pip3 install https://github.com/sgl-project/whl/releases/download/v${SGL_KERNEL_VERSION}/sgl_kernel-${SGL_KERNEL_VERSION}-cp39-abi3-manylinux2014_x86_64.whl --break-system-packages
fi
# Show current packages # Show current packages
pip list pip list
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment