Skip to content

Commit 3ee4b47

Browse files
committed
rebase main and fix
Signed-off-by: zzhx1 <zzh_201018@outlook.com>
1 parent 6b73624 commit 3ee4b47

File tree

2 files changed

+10
-4
lines changed

2 files changed

+10
-4
lines changed

tests/ut/quantization/test_w4a8_dynamic.py

Lines changed: 9 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -11,10 +11,16 @@
1111
class TestAscendW4A8DynamicLinearMethod(TestBase):
1212

1313
def setUp(self):
14-
with patch('vllm_ascend.quantization.w4a8_dynamic.get_current_vllm_config') as mock_get_current_vllm_config:
14+
with patch(
15+
'vllm_ascend.quantization.w4a8_dynamic.get_current_vllm_config'
16+
) as mock_get_current_vllm_config:
1517
mock_vllm_config = Mock()
16-
mock_vllm_config.quant_config = Mock(quant_description={"group_size": 256})
17-
mock_vllm_config.scheduler_config = Mock(max_num_batched_tokens=2048, max_model_len=2048, enable_chunked_prefill=False)
18+
mock_vllm_config.quant_config = Mock(
19+
quant_description={"group_size": 256})
20+
mock_vllm_config.scheduler_config = Mock(
21+
max_num_batched_tokens=2048,
22+
max_model_len=2048,
23+
enable_chunked_prefill=False)
1824
mock_get_current_vllm_config.return_value = mock_vllm_config
1925
self.method = AscendW4A8DynamicLinearMethod()
2026
self.method.group_size = 8

tests/ut/torchair/models/test_torchair_deepseek_v2.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -104,7 +104,7 @@ def mock_distributed():
104104
mlp_tp_group.rank_in_group = 0
105105
mlp_tp_group.world_size = 1
106106
mlp_tp_group.all_gather = Mock(return_value=torch.randn(2, 4, 128))
107-
107+
108108
mock_vllm_config = Mock()
109109
mock_vllm_config.scheduler_config = Mock(max_num_seqs=256)
110110
mock_vllm_config.model_config = Mock(max_model_len=2048, quant_config=None)

0 commit comments

Comments
 (0)