cleanup code

Signed-off-by: Ceng23333 <441651826@qq.com>

Author: Ceng23333

test/infinicore/ops/gla_attention.py

@@ -40,7 +40,7 @@
 
 
 def _torch_simple_gla_recurrent_ref(q, k, v, g_gamma, scale: float) -> torch.Tensor:
-    """Reference recurrence (same as gla_attention.py / InfiniCore header)."""
+    """Reference recurrence (HF LightningAttention Simple GLA; matches InfiniLM/InfiniCore math)."""
     dtype = q.dtype
     qf = q.transpose(1, 2).float()
     kf = k.transpose(1, 2).float()