diag fix: None gate

app.py

@@ -480,7 +480,7 @@ def aoti_diag() -> str:
              f"tiles {geometry.n_tiles}, topk {geometry.topk}", ""]
 
     def eager(gate_h):
-        return vsa_h3.sparse_attention(q, k, v, gate_h.transpose(1, 2), geometry)
+        return vsa_h3.sparse_attention(q, k, v, None if gate_h is None else gate_h.transpose(1, 2), geometry)
 
     def functional(gate_h):
         return aoti_attention.sparse_attention_functional(