fix indentation bug after merging

leideng · leideng · commit 7d7b2a25f257 · 2025-12-29T17:43:02.000+08:00
diff --git a/ucm/sparse/kvcomp/kvcomp_hbm.py b/ucm/sparse/kvcomp/kvcomp_hbm.py
@@ -328,7 +328,7 @@ def attention_begin(
                         torch.bfloat16
                     )
                     valid_k_hash_token = attn_metadata.slot_mapping.flatten().numel()
-                reshape_and_cache_khash_triton(
+                    reshape_and_cache_khash_triton(
                         k_hash_compute[:valid_k_hash_token],
                         attn_metadata.slot_mapping.flatten(),
                         k_hash,

Original file line number	Diff line number	Diff line change
`@@ -328,7 +328,7 @@ def attention_begin(`
`328`	`328`	`torch.bfloat16`
`329`	`329`	`)`
`330`	`330`	`valid_k_hash_token = attn_metadata.slot_mapping.flatten().numel()`
`331`		`- reshape_and_cache_khash_triton(`
	`331`	`+ reshape_and_cache_khash_triton(`
`332`	`332`	`k_hash_compute[:valid_k_hash_token],`
`333`	`333`	`attn_metadata.slot_mapping.flatten(),`
`334`	`334`	`k_hash,`