Skip to content

Commit 0f3662f

Browse files
tjtanaaYuqi Zhang
authored andcommitted
[BUG] [ROCm] [MLA] Fix variable name bug due to change in variable name in PR vllm-project#17483 (vllm-project#17961)
Signed-off-by: tjtanaa <[email protected]> Signed-off-by: Yuqi Zhang <[email protected]>
1 parent f2d2b93 commit 0f3662f

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

vllm/v1/attention/backends/mla/rocm_aiter_mla.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -98,17 +98,17 @@ def _get_paged_kv_tensors(
9898
paged_kv_last_page_len,
9999
)
100100

101-
def _build_decode(self, block_table: torch.Tensor,
101+
def _build_decode(self, block_table_tensor: torch.Tensor,
102102
seq_lens: torch.Tensor) -> AiterMLADecodeMetadata:
103103

104104
(
105105
paged_kv_indices,
106106
paged_kv_indptr,
107107
paged_last_page_len,
108-
) = self._get_paged_kv_tensors(block_table, seq_lens)
108+
) = self._get_paged_kv_tensors(block_table_tensor, seq_lens)
109109

110110
attn_metadata = AiterMLADecodeMetadata(
111-
block_table=block_table,
111+
block_table=block_table_tensor,
112112
seq_lens=seq_lens,
113113
paged_kv_indptr=paged_kv_indptr,
114114
paged_kv_indices=paged_kv_indices,

0 commit comments

Comments
 (0)