Fix some ci issue and refactor modelrunner (#2445)
### What this PR does / why we need it?
Fix some ci issue and refactor modelrunner
### Does this PR introduce _any_ user-facing change?
N/A
### How was this patch tested?
CI passed with existing test.
- vLLM version: v0.10.0
- vLLM main:
4d9c61993a
---------
Signed-off-by: wangli <wangli858794774@gmail.com>
Signed-off-by: MengqingCao <cmq0113@163.com>
Signed-off-by: weiguihua2 <weiguihua2@huawei.com>
Co-authored-by: wangli <wangli858794774@gmail.com>
Co-authored-by: weiguihua2 <weiguihua2@huawei.com>
This commit is contained in:
@@ -536,10 +536,10 @@ class TestNPUPlatform(TestBase):
|
||||
mock_config = MagicMock(spec=ModelConfig)
|
||||
self.assertTrue(self.platform.supports_v1(mock_config))
|
||||
|
||||
def test_get_piecewise_backend_cls_returns_correct_value(self):
|
||||
def test_get_static_graph_wrapper_cls_returns_correct_value(self):
|
||||
self.assertEqual(
|
||||
self.platform.get_piecewise_backend_cls(),
|
||||
"vllm_ascend.compilation.piecewise_backend.NPUPiecewiseBackend",
|
||||
self.platform.get_static_graph_wrapper_cls(),
|
||||
"vllm_ascend.compilation.acl_graph.ACLGraphWrapper",
|
||||
)
|
||||
|
||||
@patch("torch.distributed.is_hccl_available", return_value=True)
|
||||
|
||||
Reference in New Issue
Block a user