mla cleanup

SageMoore · SageMoore · commit c1c003fde1c7 · 2025-09-03T03:11:22.000Z
Signed-off-by: Sage Moore &lt;sage@neuralmagic.com&gt;
diff --git a/vllm/v1/attention/backends/mla/common.py b/vllm/v1/attention/backends/mla/common.py
@@ -586,7 +586,7 @@ def _build_decode(self, block_table_tensor: torch.Tensor,
         )
 
     def build_for_cudagraph_capture(
-                self, common_attn_metadata: CommonAttentionMetadata) -> M:
+            self, common_attn_metadata: CommonAttentionMetadata) -> M:
         """
         This method builds the metadata for full cudagraph capture.
         Currently, only decode is supported for full cudagraphs with MLA.

Original file line number	Diff line number	Diff line change
`@@ -586,7 +586,7 @@ def _build_decode(self, block_table_tensor: torch.Tensor,`
`586`	`586`	`)`
`587`	`587`
`588`	`588`	`def build_for_cudagraph_capture(`
`589`		`- self, common_attn_metadata: CommonAttentionMetadata) -> M:`
	`589`	`+ self, common_attn_metadata: CommonAttentionMetadata) -> M:`
`590`	`590`	`"""`
`591`	`591`	`This method builds the metadata for full cudagraph capture.`
`592`	`592`	`Currently, only decode is supported for full cudagraphs with MLA.`