EngineX/xc-llm-ascend
3
0
Fork 0
Code Issues Pull Requests Actions Projects Releases Wiki Activity

[Bugfix][Aclgraph] failed to update graph task (#4282)

Browse Source 
### What this PR does / why we need it?
bugfix the error of full graph aclgraph


Signed-off-by: chenmenglong <chenmenglong1@huawei.com>
This commit is contained in:
MengLong Chen
2025-11-19 21:30:48 +08:00
committed by GitHub
parent c87a77e8b4
commit 277670730c
1 changed files with 2 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
vllm_ascend/compilation/acl_graph.py
View File

@@ -278,7 +278,8 @@ def update_mla_attn_params(update_stream, forward_context, runtime_shape,
else: else:
seq_lens_list = seq_lens_list + [0] * (runtime_shape - seq_lens_list = seq_lens_list + [0] * (runtime_shape -
len(seq_lens_list)) 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( torch_npu.npu_fused_infer_attention_score.out(
q_nope, q_nope,