diff --git a/python/sglang/srt/layers/attention/nsa/nsa_indexer.py b/python/sglang/srt/layers/attention/nsa/nsa_indexer.py index 087e228796b0..51ba1bcab7fc 100644 --- a/python/sglang/srt/layers/attention/nsa/nsa_indexer.py +++ b/python/sglang/srt/layers/attention/nsa/nsa_indexer.py @@ -123,11 +123,13 @@ def topk_transform( def rotate_activation(x: torch.Tensor) -> torch.Tensor: - # from sgl_kernel import hadamard_transform if _is_hip or _is_sm103: from fast_hadamard_transform import hadamard_transform else: - from sgl_kernel import hadamard_transform + try: + from sgl_kernel import hadamard_transform + except ImportError: + from fast_hadamard_transform import hadamard_transform hidden_size = x.size(-1) assert (