cleanup

lucidrains · lucidrains · commit 6d9f41b1a72f · 2025-02-26T16:16:07.000Z
diff --git a/native_sparse_attention_pytorch/triton_native_sparse_attention.py b/native_sparse_attention_pytorch/triton_native_sparse_attention.py
@@ -140,8 +140,6 @@ def forward_kernel(
 
     # lse
 
-    offs_lse_qh = tl.arange(0, QUERY_HEAD_GROUPS)
-
     lse_ptrs = (
         Lse +
         offs_qh[:, None] * seqlen_q_rounded +
