Skip to content

Commit 6c3b6ff

Browse files
committed
lint fix
Signed-off-by: ganyi <[email protected]>
1 parent 5e51501 commit 6c3b6ff

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

vllm_ascend/attention/mla_v1.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -999,8 +999,8 @@ def _forward_decode(
999999
else:
10001000
assert len(kv_c_and_k_pe_cache) > 1
10011001
attn_output = torch_npu.atb.npu_multi_head_latent_attention(
1002-
q_nope, q_pe, kv_c_and_k_pe_cache[0],
1003-
kv_c_and_k_pe_cache[1], attn_metadata.decode.block_table,
1002+
q_nope, q_pe, kv_c_and_k_pe_cache[0], kv_c_and_k_pe_cache[1],
1003+
attn_metadata.decode.block_table,
10041004
attn_metadata.decode.seq_lens, self.num_heads, self.scale,
10051005
self.num_kv_heads)
10061006
current_ms_metadata = get_multistream_comm_context()

0 commit comments

Comments
 (0)