Fix ci install to allow prerelease (#12449)

2025-10-31 02:22:15 -07:00
parent 300b4c212e
commit 2d5605e89b
2 changed files with 2 additions and 2 deletions
--- 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
--- 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):