Skip to content

Commit d96d5fa

Browse files
authored
[Test] quick fix mla ut (#4318)
### What this PR does / why we need it? ### Does this PR introduce _any_ user-facing change? ### How was this patch tested? - vLLM version: v0.11.0 - vLLM main: vllm-project/vllm@2918c1b Signed-off-by: GDzhu01 <809721801@qq.com>
1 parent 5c9f4a4 commit d96d5fa

File tree

1 file changed

+0
-2
lines changed

1 file changed

+0
-2
lines changed

tests/ut/models/test_mla.py

Lines changed: 0 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -211,5 +211,3 @@ def test_forward(self, mock_get_forward_context, mock_tp_size,
211211
output = attn.forward(positions, hidden_states)
212212

213213
self.assertEqual(output.shape, (3, self.hidden_size))
214-
self.assertTrue(
215-
torch.allclose(output, output.view(-1, self.hidden_size)))

0 commit comments

Comments
 (0)