EngineX/xc-llm-ascend
3
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

[UT] Fix ut test (#4472)

Browse Source 
### 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.2
- vLLM main: https://github.com/vllm-project/vllm/commit/v0.11.2

Signed-off-by: hfadzxy <starmoon_zhang@163.com>
This commit is contained in:
zhangxinyuehfad
2025-11-26 21:37:47 +08:00
committed by GitHub
parent d252e36ae8
commit 84d7f5a10d
1 changed files with 4 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
tests/ut/attention/test_mla_v1.py
View File

@@ -456,6 +456,8 @@ class TestAscendMLAMetadataBuilderBuild(TestBase):
@patch("vllm_ascend.attention.mla_v1.get_ascend_config") @patch("vllm_ascend.attention.mla_v1.get_ascend_config")
def test_build_prefix_no_cache_metadata(self, mock_get_ascend_config, def test_build_prefix_no_cache_metadata(self, mock_get_ascend_config,
mock_dcp_world_size): mock_dcp_world_size):
if not torch.npu.is_available():
self.skipTest("NPU not available, skipping NPU-dependent tests")
mock_dcp_world_size.return_value = 1 mock_dcp_world_size.return_value = 1
common_attn_metadata = AscendCommonAttentionMetadata( common_attn_metadata = AscendCommonAttentionMetadata(
@@ -506,6 +508,8 @@ class TestAscendMLAMetadataBuilderBuild(TestBase):
@patch("vllm_ascend.attention.mla_v1.get_ascend_config") @patch("vllm_ascend.attention.mla_v1.get_ascend_config")
def test_build_chunked_prefix_metadata(self, mock_get_ascend_config, def test_build_chunked_prefix_metadata(self, mock_get_ascend_config,
mock_dcp_world_size): mock_dcp_world_size):
if not torch.npu.is_available():
self.skipTest("NPU not available, skipping NPU-dependent tests")
mock_dcp_world_size.return_value = 1 mock_dcp_world_size.return_value = 1
common_attn_metadata = AscendCommonAttentionMetadata( common_attn_metadata = AscendCommonAttentionMetadata(