functional name fix

aoti_attention.py

@@ -102,6 +102,7 @@ def sparse_attention_functional(
     # Prefix-exempt top-k mask, assembled from fresh tensors (functional, export-traceable):
     # video query rows keep the top-k VIDEO kv tiles, prefix kv tiles are always selected,
     # prefix query rows are dense over everything.
+    num_q_video = n_tiles - num_prefix_tiles
     video_kv_scores = scores[..., num_prefix_tiles:, num_prefix_tiles:]          # [B, H, nq_video, kv_video]
     indices = video_kv_scores.topk(topk, dim=-1, sorted=False).indices
     mask_video_kv = torch.zeros_like(video_kv_scores, dtype=torch.bool).scatter(-1, indices, True)