pass pre-commit check

leideng · leideng · commit 4361da065db1 · 2025-12-29T17:50:19.000+08:00
diff --git a/ucm/sparse/kvcomp/kvcomp_hbm.py b/ucm/sparse/kvcomp/kvcomp_hbm.py
@@ -224,7 +224,11 @@ def __init__(self, vllm_config: VllmConfig, role: UcmSparseRole):
                         [self.max_batch_size], dtype=torch.int32, device=self.device
                     )
                     self.hamming_output = torch.zeros(
-                        [self.max_batch_size, self.num_key_heads, self.hash_topk_tokens // self.block_size],
+                        [
+                            self.max_batch_size,
+                            self.num_key_heads,
+                            self.hash_topk_tokens // self.block_size,
+                        ],
                         dtype=torch.int32,
                         device=self.device,
                     )

Original file line number	Diff line number	Diff line change
`@@ -224,7 +224,11 @@ def __init__(self, vllm_config: VllmConfig, role: UcmSparseRole):`
`224`	`224`	`[self.max_batch_size], dtype=torch.int32, device=self.device`
`225`	`225`	`)`
`226`	`226`	`self.hamming_output = torch.zeros(`
`227`		`- [self.max_batch_size, self.num_key_heads, self.hash_topk_tokens // self.block_size],`
	`227`	`+ [`
	`228`	`+ self.max_batch_size,`
	`229`	`+ self.num_key_heads,`
	`230`	`+ self.hash_topk_tokens // self.block_size,`
	`231`	`+ ],`
`228`	`232`	`dtype=torch.int32,`
`229`	`233`	`device=self.device,`
`230`	`234`	`)`