Fix GLM-5.3 forget-gate shape for nvCUTEDSL verify (#40607)

2026-09-21 13:52:26 -07:00
parent e0c2e8dc4d
commit f532ad1f9a
1 changed files with 1 additions and 0 deletions
@@ -1583,6 +1583,7 @@ class KDAAttnBackend(MambaAttnBackendBase):
            onorm_eps = None
            onorm_gate = None

+        a = a.reshape(1, seq_len, h, layer.head_k_dim)
        out = fused_kda_decode_mtp_dspark(
            x_q=x_q,
            x_k=x_k,