Skip to content

Commit 2c50849

Browse files
committed
Remove unused block_tables_for_hamming initialization in KvCompOnDevice class
1 parent cf5e682 commit 2c50849

File tree

1 file changed

+0
-3
lines changed

1 file changed

+0
-3
lines changed

ucm/sparse/kvcomp/kvcomp_hbm.py

Lines changed: 0 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -164,9 +164,6 @@ def __init__(self, vllm_config: VllmConfig, role: UcmSparseRole):
164164
fill_value=self.hash_topk_tokens // self.block_size,
165165
dtype=torch.int32,
166166
device='cpu')
167-
self.block_tables_for_hamming = torch.zeros([self.max_batch_size, self.max_seq_len_for_hamming],
168-
dtype=torch.int32,
169-
device=self.device)
170167
self.seq_lens_for_hamming = torch.zeros([self.max_batch_size],
171168
dtype=torch.int32,
172169
device=self.device)

0 commit comments

Comments
 (0)