[kernel slimming] Move fast_hadamard_transform to jit_kernel (#18475)

2026-02-14 23:06:21 +08:00
parent ae95869292
commit c29394e3c8
17 changed files with 2170 additions and 71 deletions
--- a/python/sglang/srt/layers/attention/nsa/nsa_indexer.py
+++ b/python/sglang/srt/layers/attention/nsa/nsa_indexer.py
@@ -125,7 +125,7 @@ def rotate_activation(x: torch.Tensor) -> torch.Tensor:
    if _is_hip:
        from fast_hadamard_transform import hadamard_transform
    else:
-        from sgl_kernel import hadamard_transform
+        from sglang.jit_kernel.hadamard import hadamard_transform

    hidden_size = x.size(-1)
    assert (