Unverified Commit 7de18d54 authored by TJian's avatar TJian Committed by GitHub
Browse files

[BUG] [ROCm] [MLA] Fix variable name bug due to change in variable name in PR #17483 (#17961)


Signed-off-by: default avatartjtanaa <tunjian.tan@embeddedllm.com>
parent a810b5b0
......@@ -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,
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment