fix full graph case

ganyi1996ppo · ganyi1996ppo · commit e68d9a50a36c · 2025-12-18T14:00:55.000Z
Signed-off-by: ganyi &lt;ygan@amd.com&gt;
diff --git a/vllm/v1/attention/backends/mla/rocm_aiter_mla_sparse.py b/vllm/v1/attention/backends/mla/rocm_aiter_mla_sparse.py
@@ -143,7 +143,7 @@ class ROCMAiterMLASparseMetadata:
 class ROCMAiterMLASparseMetadataBuilder(
     AttentionMetadataBuilder[ROCMAiterMLASparseMetadata]
 ):
-    cudagraph_support: ClassVar[AttentionCGSupport] = (
+    _cudagraph_support: ClassVar[AttentionCGSupport] = (
         AttentionCGSupport.UNIFORM_SINGLE_TOKEN_DECODE
     )
 

Original file line number	Diff line number	Diff line change
`@@ -143,7 +143,7 @@ class ROCMAiterMLASparseMetadata:`
`143`	`143`	`class ROCMAiterMLASparseMetadataBuilder(`
`144`	`144`	`AttentionMetadataBuilder[ROCMAiterMLASparseMetadata]`
`145`	`145`	`):`
`146`		`- cudagraph_support: ClassVar[AttentionCGSupport] = (`
	`146`	`+ _cudagraph_support: ClassVar[AttentionCGSupport] = (`
`147`	`147`	`AttentionCGSupport.UNIFORM_SINGLE_TOKEN_DECODE`
`148`	`148`	`)`
`149`	`149`