Skip to content

Commit 5dcf47b

Browse files
author
wangxiaoxin-sherie
committed
xx
1 parent ecfb820 commit 5dcf47b

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

vllm_ascend/compilation/acl_graph.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -314,9 +314,9 @@ def update_attn_dcp_pcp_params(update_stream, forward_context, runtime_shape):
314314
graph_params.handles[runtime_shape],
315315
graph_params.events[runtime_shape],
316316
):
317-
(q_nope, k_nope, value, num_heads, num_kv_heads, scale, block_table,
318-
block_size, actual_seq_lengths_kv, attn_output, softmax_lse, cp_rank,
319-
dcp_rank, dcp_size) = param
317+
(q_nope, k_nope, value, num_heads, num_kv_heads, scale,
318+
block_table, block_size, actual_seq_lengths_kv, attn_output,
319+
softmax_lse, cp_rank, dcp_rank, dcp_size) = param
320320
actual_seq_lengths_kv = forward_context.attn_metadata[
321321
key].decode_meta.num_computed_tokens_of_pcp_dcp[:, cp_rank,
322322
dcp_rank]

0 commit comments

Comments
 (0)