diag eager gate fix

app.py

@@ -480,7 +480,8 @@ def aoti_diag() -> str:
              f"tiles {geometry.n_tiles}, topk {geometry.topk}", ""]
 
     def eager(gate_h):
-        return vsa_h3.sparse_attention(q, k, v, None if gate_h is None else gate_h.transpose(1, 2), geometry)
+        # eager sparse_attention takes the gate as [B, S, H, D] and transposes internally
+        return vsa_h3.sparse_attention(q, k, v, gate_h, geometry)
 
     def functional(gate_h):
         gate_t = None if gate_h is None else gate_h.transpose(1, 2)