Skip to content

Commit 2776707

Browse files
[Bugfix][Aclgraph] failed to update graph task (#4282)
### What this PR does / why we need it? bugfix the error of full graph aclgraph Signed-off-by: chenmenglong <[email protected]>
1 parent c87a77e commit 2776707

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

vllm_ascend/compilation/acl_graph.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -278,7 +278,8 @@ def update_mla_attn_params(update_stream, forward_context, runtime_shape,
278278
else:
279279
seq_lens_list = seq_lens_list + [0] * (runtime_shape -
280280
len(seq_lens_list))
281-
torch.npu.graph_task_update_begin(update_stream, handle)
281+
282+
torch.npu.graph_task_update_begin(update_stream, handle)
282283

283284
torch_npu.npu_fused_infer_attention_score.out(
284285
q_nope,

0 commit comments

Comments
 (0)