Skip to content

Commit 90d1f5c

Browse files
author
lwq
committed
Format Code
Signed-off-by: lwq <liwenquan5@huawei.com>
1 parent 2eeda23 commit 90d1f5c

File tree

1 file changed

+0
-3
lines changed

1 file changed

+0
-3
lines changed

tests/ut/attention/test_mla_v1.py

Lines changed: 0 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -439,9 +439,6 @@ def test_forward_decode_without_graph(self,
439439
self.impl.qk_nope_head_dim)
440440
k_pe = torch.randn(num_tokens, self.impl.num_heads,
441441
self.impl.qk_rope_head_dim)
442-
kv_c_and_k_pe_cache = torch.randn(num_blocks, block_size,
443-
self.impl.num_heads,
444-
self.impl.kv_lora_rank)
445442
metadata = MagicMock()
446443
metadata.decode = MagicMock()
447444
metadata.decode.block_table = MagicMock()

0 commit comments

Comments
 (0)