From 277670730c9bb1b3b40e0c489b08cd2e695d02ee Mon Sep 17 00:00:00 2001 From: MengLong Chen <71744434+dragondream-chen@users.noreply.github.com> Date: Wed, 19 Nov 2025 21:30:48 +0800 Subject: [PATCH] [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 --- vllm_ascend/compilation/acl_graph.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/vllm_ascend/compilation/acl_graph.py b/vllm_ascend/compilation/acl_graph.py index 889fb34..7522f8b 100644 --- a/vllm_ascend/compilation/acl_graph.py +++ b/vllm_ascend/compilation/acl_graph.py @@ -278,7 +278,8 @@ def update_mla_attn_params(update_stream, forward_context, runtime_shape, else: seq_lens_list = seq_lens_list + [0] * (runtime_shape - len(seq_lens_list)) - torch.npu.graph_task_update_begin(update_stream, handle) + + torch.npu.graph_task_update_begin(update_stream, handle) torch_npu.npu_fused_infer_attention_score.out( q_nope,