From 7de18d541b0da661685d481d7306cbe5e9f7960b Mon Sep 17 00:00:00 2001 From: TJian Date: Mon, 12 May 2025 00:14:30 +0800 Subject: [PATCH] [BUG] [ROCm] [MLA] Fix variable name bug due to change in variable name in PR #17483 (#17961) Signed-off-by: tjtanaa --- vllm/v1/attention/backends/mla/rocm_aiter_mla.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/vllm/v1/attention/backends/mla/rocm_aiter_mla.py b/vllm/v1/attention/backends/mla/rocm_aiter_mla.py index f46010d757af6..3abb185c5b8fe 100644 --- a/vllm/v1/attention/backends/mla/rocm_aiter_mla.py +++ b/vllm/v1/attention/backends/mla/rocm_aiter_mla.py @@ -98,17 +98,17 @@ class AiterMLAMetadataBuilder(MLACommonMetadataBuilder[AiterMLAMetadata]): paged_kv_last_page_len, ) - def _build_decode(self, block_table: torch.Tensor, + def _build_decode(self, block_table_tensor: torch.Tensor, seq_lens: torch.Tensor) -> AiterMLADecodeMetadata: ( paged_kv_indices, paged_kv_indptr, paged_last_page_len, - ) = self._get_paged_kv_tensors(block_table, seq_lens) + ) = self._get_paged_kv_tensors(block_table_tensor, seq_lens) attn_metadata = AiterMLADecodeMetadata( - block_table=block_table, + block_table=block_table_tensor, seq_lens=seq_lens, paged_kv_indptr=paged_kv_indptr, paged_kv_indices=paged_kv_indices,