From a0c3b8dd2d66da145edcb6b44c44f1ddf9763622 Mon Sep 17 00:00:00 2001 From: wangxiyuan Date: Wed, 22 Oct 2025 14:14:15 +0800 Subject: [PATCH] [v0.11.0]cherry-pick fix ut (#3608) (#3614) cherry-pick fix ut (#3608) Signed-off-by: wangxiyuan --- tests/ut/kv_connector/test_mooncake_connector.py | 1 - tests/ut/torchair/models/test_torchair_deepseek_v2.py | 5 +---- 2 files changed, 1 insertion(+), 5 deletions(-) diff --git a/tests/ut/kv_connector/test_mooncake_connector.py b/tests/ut/kv_connector/test_mooncake_connector.py index 3eb58a3..7bb2142 100644 --- a/tests/ut/kv_connector/test_mooncake_connector.py +++ b/tests/ut/kv_connector/test_mooncake_connector.py @@ -1010,7 +1010,6 @@ def mock_string_to_int64_hash(s): return hash(s) -@unittest.skip("skip") class TestMooncakeConnectorWorker(unittest.TestCase): def setUp(self): diff --git a/tests/ut/torchair/models/test_torchair_deepseek_v2.py b/tests/ut/torchair/models/test_torchair_deepseek_v2.py index 52a43ce..09d66f2 100644 --- a/tests/ut/torchair/models/test_torchair_deepseek_v2.py +++ b/tests/ut/torchair/models/test_torchair_deepseek_v2.py @@ -303,14 +303,11 @@ def test_torchair_deepseek_v2_mla_attention(mock_rms_norm, mock_distributed, @patch("torch_npu.npu_add_rms_norm") @patch("torch_npu.npu_rms_norm") @patch("torch.ops.vllm.maybe_wait_prefetch_done", side_effect=lambda x: None) -@patch("torch.ops.vllm.maybe_chunk_residual") -def test_torchair_deepseek_v2_decoder_layer(mock_maybe_chunk_residual, - mock_maybe_wait_prefetch_done, +def test_torchair_deepseek_v2_decoder_layer(mock_maybe_wait_prefetch_done, mock_rms_norm, mock_add_norm, mock_distributed, base_config, vllm_config, mock_forward_context, patch_attention_init): - mock_maybe_chunk_residual.return_value = torch.randn(2, 4, 128) mock_rms_norm.return_value = (torch.randn(2, 128), torch.randn(2, 128)) mock_add_norm.return_value = (torch.randn(2, 128), torch.randn(2, 128), torch.randn(2, 128))