Skip to content

Commit 1e31b07

Browse files
momo609wangxiaoxin-sherie
andauthored
fix qwen3next full graph break. (#3812)
### What this PR does / why we need it? fix qwen3next full graph break. linearattention doesnot has aclgraph_support attr,so change to cudagraph_support to support vllm. <img width="603" height="120" alt="image" src="https://github.com/user-attachments/assets/d2de53bb-4147-495a-9129-51d9083749be" /> ### Does this PR introduce _any_ user-facing change? ### How was this patch tested? - vLLM version: v0.11.0rc3 - vLLM main: vllm-project/vllm@releases/v0.11.1 Signed-off-by: wangxiaoxin-sherie <[email protected]> Co-authored-by: wangxiaoxin-sherie <[email protected]>
1 parent c76db62 commit 1e31b07

File tree

1 file changed

+6
-1
lines changed

1 file changed

+6
-1
lines changed

vllm_ascend/worker/model_runner_v1.py

Lines changed: 6 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -3775,7 +3775,12 @@ def initialize_aclgraph_capture(self) -> None:
37753775

37763776
for attn_group in self._attn_group_iterator():
37773777
builder = attn_group.get_metadata_builder()
3778-
if builder.aclgraph_support.value < min_ag_support.value:
3778+
graph_support = None
3779+
if hasattr(builder, 'aclgraph_support'):
3780+
graph_support = builder.aclgraph_support.value
3781+
else:
3782+
graph_support = builder.cudagraph_support.value
3783+
if graph_support < min_ag_support.value:
37793784
min_ag_support = builder.aclgraph_support
37803785
min_ag_builder_name = builder.__class__.__name__
37813786

0 commit comments

Comments
 (0)