[CI] Fix lint that was not applied in #21458 (#21818)

This commit is contained in:
Alex Nails
2026-03-31 23:58:12 -07:00
committed by GitHub
parent 2861596fc6
commit 912494f596
+3 -1
View File
@@ -198,7 +198,9 @@ class Qwen3Attention(nn.Module):
)
return q, k, v
def forward_prepare_aiter_fused_mrope(self, positions, hidden_states, forward_batch):
def forward_prepare_aiter_fused_mrope(
self, positions, hidden_states, forward_batch
):
"""Fused QK-norm + 3D mRoPE + KV cache write for decode (ROCm/aiter).
The fused HIP kernel replaces split → QK norm → mRoPE → cache write,