refactor: Move scalar_types.py to sgl-kernel to avoid circular import (#8720)

2025-08-08 10:22:16 +08:00
parent ed0a3dd54a
commit 39fd178831
3 changed files with 354 additions and 2 deletions
--- a/sgl-kernel/python/sgl_kernel/fused_moe.py
+++ b/sgl-kernel/python/sgl_kernel/fused_moe.py
@@ -5,7 +5,7 @@ import torch


 def get_scalar_type(num_bits: int, has_zp: bool):
-    from sglang.srt.layers.quantization.scalar_type import scalar_types
+    from sgl_kernel.scalar_type import scalar_types

    if has_zp:
        assert num_bits == 4