From 2d5605e89bd858e4a6e3b631c6cb15aaec2e8e7a Mon Sep 17 00:00:00 2001 From: Lianmin Zheng Date: Fri, 31 Oct 2025 02:22:15 -0700 Subject: [PATCH] Fix ci install to allow prerelease (#12449) --- scripts/ci/ci_install_dependency.sh | 2 +- sgl-kernel/python/sgl_kernel/fused_moe.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/scripts/ci/ci_install_dependency.sh b/scripts/ci/ci_install_dependency.sh index b40854dce..e8eb785d7 100755 --- a/scripts/ci/ci_install_dependency.sh +++ b/scripts/ci/ci_install_dependency.sh @@ -86,7 +86,7 @@ else export UV_SYSTEM_PYTHON=true PIP_CMD="uv pip" - PIP_INSTALL_SUFFIX="--index-strategy unsafe-best-match" + PIP_INSTALL_SUFFIX="--index-strategy unsafe-best-match --prerelease allow" # Clean up existing installations $PIP_CMD uninstall sgl-kernel sglang || true diff --git a/sgl-kernel/python/sgl_kernel/fused_moe.py b/sgl-kernel/python/sgl_kernel/fused_moe.py index ab31d73b9..1f0bc7263 100644 --- a/sgl-kernel/python/sgl_kernel/fused_moe.py +++ b/sgl-kernel/python/sgl_kernel/fused_moe.py @@ -2,7 +2,7 @@ import functools from typing import Optional import torch -from sgl_kernel import silu_and_mul +from sgl_kernel.elementwise import silu_and_mul def get_scalar_type(num_bits: int, has_zp: bool):