[bug] fix router installation to include additional dependency (#12348)

This commit is contained in:
Simo Lin
2025-10-29 12:45:18 -07:00
committed by GitHub
parent ed1044ac1b
commit 5259becd3c
8 changed files with 47 additions and 9 deletions

View File

@@ -24,6 +24,29 @@ rm -rf /root/.cache/flashinfer
# Install apt packages
apt install -y git libnuma-dev
# Install protoc for router build (gRPC protobuf compilation)
if ! command -v protoc &> /dev/null; then
echo "Installing protoc..."
if command -v apt-get &> /dev/null; then
# Ubuntu/Debian
apt-get update
apt-get install -y wget unzip gcc g++ perl make
elif command -v yum &> /dev/null; then
# RHEL/CentOS
yum update -y
yum install -y wget unzip gcc gcc-c++ perl-core make
fi
cd /tmp
wget https://github.com/protocolbuffers/protobuf/releases/download/v32.0/protoc-32.0-linux-x86_64.zip
unzip protoc-32.0-linux-x86_64.zip -d /usr/local
rm protoc-32.0-linux-x86_64.zip
protoc --version
cd -
else
echo "protoc already installed: $(protoc --version)"
fi
# Install uv
if [ "$IS_BLACKWELL" = "1" ]; then
# The blackwell CI runner has some issues with pip and uv,
@@ -58,8 +81,11 @@ else
$PIP_CMD install -e "python[dev]" --extra-index-url https://download.pytorch.org/whl/${CU_VERSION} $PIP_INSTALL_SUFFIX --upgrade
fi
# Install OpenSSL development libraries for router build
apt install -y libssl-dev pkg-config
# Install router for pd-disagg test
SGLANG_ROUTER_BUILD_NO_RUST=1 $PIP_CMD install -e "sgl-router" $PIP_INSTALL_SUFFIX
$PIP_CMD install -e "sgl-router" $PIP_INSTALL_SUFFIX
# Install sgl-kernel
SGL_KERNEL_VERSION_FROM_KERNEL=$(grep -Po '(?<=^version = ")[^"]*' sgl-kernel/pyproject.toml)