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/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):