Skip to content

Commit 4bb471c

Browse files
committed
fix _cudagraph_support introduced by #28479
Signed-off-by: MengqingCao <[email protected]>
1 parent 588ed71 commit 4bb471c

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

vllm_ascend/worker/model_runner_v1.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -3866,8 +3866,8 @@ def initialize_aclgraph_capture(self) -> None:
38663866
graph_support = builder.aclgraph_support.value
38673867
builder_aclgraph = builder.aclgraph_support
38683868
else:
3869-
graph_support = builder.cudagraph_support.value
3870-
builder_aclgraph = builder.cudagraph_support
3869+
graph_support = builder._cudagraph_support.value
3870+
builder_aclgraph = builder._cudagraph_support
38713871
if graph_support < min_ag_support.value:
38723872
min_ag_support = builder_aclgraph
38733873
min_ag_builder_name = builder.__class__.__name__

0 commit comments

Comments
 (0)